async def test_get_nlu_data(Faker: asynctest.MagicMock, load_data: asynctest.MagicMock) -> None:
faker_ = Faker()
faker_.name.return_value = "Nikola Tesla"
training_data = TrainingData(
training_examples=[
Message.build("hello", "intent_test"),
Message.build("hello @name", "intent_test"),
Message.build("hello"),
]
)
load_data.return_value = training_data
importer = PlaceholderImporter()
importer.config = {"importers": [{"name": "rasam.PlaceholderImporter"}]}
importer._nlu_files = ["test"]
new_training_data = await importer.get_nlu_data()
faker_.seed_instance.assert_called_once_with(importer.DEFAULT_FAKE_DATA_COUNT)
load_data.assert_called_once_with("test", "en")
message: Message
expected_messages = [
Message.build("hello", "intent_test"),
Message.build("hello Nikola Tesla", "intent_test"),
Message.build("hello"),
]
for message, expected in zip(new_training_data.training_examples, expected_messages):
assert message.get("intent") == expected.get("intent")
assert message.get("text") == expected.get("text")
def test_post(self, ensure_future_mock: MagicMock):
response = self.fetch(
"/api/v1/email",
method="POST",
body=json.dumps({
"jobs": [
{
"id": 13,
"email": "*****@*****.**"
},
{
"id": 14,
"email": "*****@*****.**"
},
],
"template":
"Hello!",
"subject":
"Hi!",
}),
)
self.assertEqual(response.code, 202)
ensure_future_mock.assert_called_once_with(
self.email_service.send_emails.return_value)
self.email_service.send_emails.assert_called_once_with(
jobs=[{
"id": 13,
"email": "*****@*****.**"
}, {
"id": 14,
"email": "*****@*****.**"
}],
template="Hello!",
subject="Hi!",
)
async def test_consume(rpc: AsyncAmqpRpc, management, exchange, queue):
sent_data = 'Test. Тест. 実験。'
request_handler = MagicMock()
await rpc.subscribe(request_handler=request_handler,
exchange=exchange,
queue=queue)
management.publish(exchange=exchange, routing_key=queue, data=sent_data)
await asyncio.sleep(0.1) # give the event loop a chance to process it
request_handler.assert_called_once_with(sent_data)
