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(
