def test_delete_does_nothing_when_aircraft_does_not_exist(self):
seen_datetime = datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S')
client_existing = Client([None, '127.0.0.1', seen_datetime])
client_id = self.mapper.insert(client_existing)
client_non_existing = Client([0, '127.0.0.1', seen_datetime])
self.mapper.delete(client_non_existing)
# assert that existing client was not affected
client = self.mapper.fetch(client_id)
self.assertIsNotNone(client)
def test_update_updates_client_when_exists(self):
seen_datetime = datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S')
client = Client([None, '127.0.0.1', seen_datetime])
client_id = self.mapper.insert(client)
client = self.mapper.fetch(client_id)
client.ip = '127.0.0.2'
self.mapper.update(client)
# assert that ip column was updated
client = self.mapper.fetch(client_id)
self.assertEqual('127.0.0.2', client.ip)
def test_map_returns_client_when_all_fields_exist(self):
datetime_expected = datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S')
client_expected = Client([None, self.data['ip'], datetime_expected])
client_actual = self.mapper.map(self.data['ip'])
self.assertEqual(str(client_expected), str(client_actual))
def map(ip):
client = Client([
None, ip,
datetime.strftime(datetime.utcnow(), '%Y-%m-%d %H:%M:%S')
])
return client
def test_fetch_identical_returns_client_when_exists(self):
seen_datetime = datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S')
client = Client([None, '127.0.0.1', seen_datetime])
self.mapper.insert(client)
client = self.repository.fetch_identical(client)
self.assertNotEqual(None, client)
def test_init_sets_is_online_to_false_when_last_seen_is_more_than_threshold(
self):
self.data[2] = (
datetime.datetime.utcnow() -
datetime.timedelta(minutes=45)).strftime('%Y-%m-%d %H:%M:%S')
client = Client(self.data)
self.assertFalse(client.is_online)
def test_init_fills_properties(self):
client = Client(self.data)
self.assertEqual(self.data[0], client.id)
self.assertEqual(self.data[1], client.ip)
self.assertEqual(
datetime.datetime.strptime(self.data[2], '%Y-%m-%d %H:%M:%S'),
client.last_seen)
def setUp(self):
timestamp = int(datetime.strftime(datetime.utcnow(), '%s'))
self.aircraft = Aircraft([1, 'tail', None, timestamp, timestamp])
self.client = Client(
[1, '127.0.0.1',
datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S')])
self.data = {'flight': 'foo', 'text': 'bar', 'timestamp': timestamp}
self.mapper = MessageInputMapper()
def fetch(self, client_id):
self.adapter.execute('SELECT id, ip, last_seen FROM clients WHERE id = ?', (client_id,))
result = self.adapter.fetchone()
client = None
if result:
client = Client(result)
return client
def test_delete_all_deletes_clients_when_exist(self):
seen_datetime = datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S')
client = Client([None, '127.0.0.1', seen_datetime])
client_id = self.mapper.insert(client)
self.mapper.delete_all()
client = self.mapper.fetch(client_id)
self.assertIsNone(client)
def test_fetch_returns_client_when_exists(self):
seen_datetime = datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S')
client = Client([None, '127.0.0.1', seen_datetime])
client_id = self.mapper.insert(client)
client = self.mapper.fetch(client_id)
self.assertIsNotNone(client)
self.assertIsNotNone(client_id)
def test_fetch_identical_returns_none_when_client_does_not_exist(self):
seen_datetime = datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S')
client = Client([None, '127.0.0.1', seen_datetime])
client = self.repository.fetch_identical(client)
self.assertEqual(None, client)
def test_str_returns_expected_string(self):
str_expected = 'ID: {}, IP: {}, Last Seen: {}, Is Online: {}'.format(
self.data[0], self.data[1], self.data[2], True)
client = Client(self.data)
self.assertEqual(str_expected, str(client))
def test_init_sets_is_online_to_true_when_last_seen_is_less_than_threshold(
self):
client = Client(self.data)
self.assertTrue(client.is_online)
def setUp(self):
timestamp = datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S')
self.client = Client(
[1, '127.0.0.1',
datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S')])
self.data = [1, 2, 'foo', 'bar', timestamp]
