def _retrieve_entries(is_active):
""" Handy function which avoid code duplication """
result = spamhaus_service.get_spamhaus_entries(is_active)
if not result:
return {"error": "Error while fetching active spamhaus entries."}, 500
return result, 200
def test_spamhaus_active_entries(self):
now = datetime.now()
last_year = now.replace(year=now.year - 1)
with patch('db.postgres.Postgres') as mock:
instance = mock.return_value
enter = instance.__enter__.return_value
enter.find_spamhaus_entries.return_value = [{
'active': True,
'sbl_number': 1,
'first_seen': last_year,
'last_seen': now,
'cidr': '1.2.3.4/32'
}]
expected = [{
'sblNumber': 1,
'firstSeen': int(mktime(last_year.timetuple())),
'lastSeen': int(mktime(now.timetuple())),
'cidr': '1.2.3.4/32'
}]
result = spamhaus.get_spamhaus_entries(True)
enter.find_spamhaus_entries.assert_called_with(True)
self.assertEqual(1, len(result))
self.assertEqual(expected, result)
def _retrieve_entries(is_active):
""" Handy function which avoid code duplication """
result = spamhaus_service.get_spamhaus_entries(is_active)
if not result:
return {"error": "Error while fetching active spamhaus entries."}, 500
return result, 200
def test_spamhaus_active_entries(self):
now = datetime.now()
last_year = now.replace(year=now.year - 1)
with patch('mongo.mongo.Mongo') as mock:
instance = mock.return_value
enter = instance.__enter__.return_value
enter.find_spamhaus_entries.return_value = [{
'_id': 'xxx',
'active': True,
'sbl_number': 1,
'first_seen': last_year,
'last_seen': now,
'cidr': '1.2.3.4/32'
}]
expected = [{
'sblNumber': 1,
'firstSeen': int(mktime(last_year.timetuple())),
'lastSeen': int(mktime(now.timetuple())),
'cidr': '1.2.3.4/32'
}]
result = spamhaus.get_spamhaus_entries(True)
enter.find_spamhaus_entries.assert_called_with(True)
self.assertEquals(1, len(result))
self.assertEquals(expected, result)
