def test_observation_variables(self, patch_query): query_return = MagicMock() filter_return = MagicMock() order_return = MagicMock() patch_query.return_value = query_return query_return.filter = MagicMock(return_value=filter_return) filter_return.order_by = MagicMock(return_value=order_return) variable0 = PropertyMock() variable0.name = 'variable0' variable1 = PropertyMock() variable1.name = 'variable1' order_return.all = MagicMock(return_value=[variable0, variable1]) res = self.app.get(self.apiLinks["observation_variables"]) self.assertEqual(res.status_code, 200) data = self.__get_response_data(res) self.assertEqual(len(data), 2) self.assertDictEqual(data[0], {'id': 0, 'value': 'variable0'}) self.assertDictEqual(data[1], {'id': 1, 'value': 'variable1'})
def test_observation_track(self, patch_get_stations, patch_session): platform_type = PropertyMock() platform_type.name = "platform_type" station = PropertyMock( platform=PropertyMock(type=platform_type), latitude=0, longitude=0, id=0, ) station.name = 'myname' patch_get_stations.return_value = [station] res = self.app.get(self.apiLinks["observation_point"]) self.assertEqual(res.status_code, 200) patch_get_stations.assert_called_with(session=patch_session, platform_types=['none']) data = self.__get_response_data(res) self.assertEqual(len(data['features']), 1) self.assertEqual([0, 0], data['features'][0]['geometry']['coordinates'])
def test_observation_track(self, patch_get_platform_tracks, patch_session): typ = PropertyMock() typ.name = 'none' patch_get_platform_tracks.return_value = [ [0, typ, 0, 0], [0, typ, 1, 1], [1, typ, 0, 0], ] res = self.app.get(self.apiLinks["observation_track"]) self.assertEqual(res.status_code, 200) patch_get_platform_tracks.assert_called_with(patch_session, "day", platform_types=['none']) data = self.__get_response_data(res) self.assertEqual(len(data['features']), 1) self.assertIn([0, 0], data['features'][0]['geometry']['coordinates'])
import ai.amplifier as amp from unittest.mock import patch, PropertyMock, Mock from channels import REPETITIONS from resources import DRONE_AVATAR test_guild = Mock() mocked_member_1 = Mock() mocked_member_1.display_name = "⬡-Drone #0077" mocked_member_2 = Mock() mocked_member_2.display_name = "⬡-Drone #5890" test_channel = PropertyMock() test_channel.guild = test_guild test_channel.name = REPETITIONS class AmplifierTest(unittest.TestCase): def setUp(self): # Reset the get_member return values (prevents dirtying context between tests). test_guild.get_member.side_effect = [mocked_member_1, mocked_member_2] @patch("ai.amplifier.convert_id_to_member", side_effect=[mocked_member_1, mocked_member_2]) def test_amplifier_generator_returns_profile_dictionaries( self, mocked_id_converter): self.assertEqual( next(amp.generate_amplification_information( test_channel, ["0077"]))["username"], "⬡-Drone #0077") self.assertEqual(