def testGetExpiredHistograms(self):
histograms = {
"FirstHistogram": {
"expiry_date": "2000/10/01"
},
"SecondHistogram": {
"expiry_date": "2002/10/01"
},
"ThirdHistogram": {
"expiry_date": "2001/10/01"
},
"FourthHistogram": {},
"FifthHistogram": {
"obsolete": "Has expired.",
"expiry_date": "2000/10/01"
}
}
base_date = datetime.date(2001, 10, 1)
expired_histograms_names = (
generate_expired_histograms_array._GetExpiredHistograms(
histograms, base_date))
self.assertEqual(expired_histograms_names, ["FirstHistogram"])
def testBadExpiryDate(self):
histograms = {
"FirstHistogram": {
"expires_after": "2000-10-01"
},
"SecondHistogram": {
"expires_after": "2000/10/01"
}
}
base_date = datetime.date(2000, 10, 1)
current_milestone = 60
with self.assertRaises(generate_expired_histograms_array.Error) as error:
generate_expired_histograms_array._GetExpiredHistograms(
histograms, base_date, current_milestone)
self.assertEqual(
generate_expired_histograms_array._DATE_FORMAT_ERROR.format(
date="2000/10/01", name="SecondHistogram"), str(error.exception))
def testBadExpiryDate(self):
histograms = {
"FirstHistogram": {
"expiry_date": "2000/10/01"
},
"SecondHistogram": {
"expiry_date": "2000-10-01"
},
}
base_date = datetime.date(2000, 10, 01)
with self.assertRaises(
generate_expired_histograms_array.Error) as error:
generate_expired_histograms_array._GetExpiredHistograms(
histograms, base_date)
self.assertEqual(
"Unable to parse expiry date 2000-10-01 in histogram SecondHistogram.",
str(error.exception))
def testGetExpiredHistograms(self):
histograms = {
"FirstHistogram": {
"expires_after": "2000-10-01"
},
"SecondHistogram": {
"expires_after": "2002-10-01"
},
"ThirdHistogram": {
"expires_after": "2001-10-01"
},
"FourthHistogram": {},
"FifthHistogram": {
"obsolete": "Has expired.",
"expires_after": "2000-10-01"
},
"SixthHistogram": {
"expires_after": "M22"
},
"SeventhHistogram": {
"expires_after": "M60"
},
"EigthHistogram": {
"expires_after": "M65"
},
}
base_date = datetime.date(2001, 10, 1)
current_milestone = 60
expired_histograms_names = (
generate_expired_histograms_array._GetExpiredHistograms(
histograms, base_date, current_milestone))
self.assertEqual(2, len(expired_histograms_names))
self.assertIn("FirstHistogram", expired_histograms_names)
self.assertIn("SixthHistogram", expired_histograms_names)