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 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)