37 lines
935 B
Python
37 lines
935 B
Python
import pytest
|
|
from control_backend.schemas.ri_message import RIMessage, RIEndpoint, SpeechCommand
|
|
from pydantic import ValidationError
|
|
|
|
|
|
def valid_command_1():
|
|
return SpeechCommand(data="Hallo?")
|
|
|
|
|
|
def invalid_command_1():
|
|
return RIMessage(endpoint=RIEndpoint.PING, data="Hello again.")
|
|
|
|
|
|
def test_valid_speech_command_1():
|
|
command = valid_command_1()
|
|
try:
|
|
RIMessage.model_validate(command)
|
|
SpeechCommand.model_validate(command)
|
|
assert True
|
|
except ValidationError:
|
|
assert False
|
|
|
|
|
|
def test_invalid_speech_command_1():
|
|
command = invalid_command_1()
|
|
passed_ri_message_validation = False
|
|
try:
|
|
# Should succeed, still.
|
|
RIMessage.model_validate(command)
|
|
passed_ri_message_validation = True
|
|
|
|
# Should fail.
|
|
SpeechCommand.model_validate(command)
|
|
assert False
|
|
except ValidationError:
|
|
assert passed_ri_message_validation
|