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)