def test_build_history_no_existing_project(self): """Testing build history when build history object is missing.""" with ndb.Client().context(): update_build_history('test-project', '1', 'fuzzing') expected_build_ids = ['1'] self.assertEqual(BuildsHistory.query().get().build_ids, expected_build_ids)
def test_build_history(self): """Testing build history.""" with ndb.Client().context(): BuildsHistory(id='test-project-fuzzing', build_tag='fuzzing', project='test-project', build_ids=[str(i) for i in range(1, 65)]).put() update_build_history('test-project', '65', 'fuzzing') expected_build_ids = [str(i) for i in range(2, 66)] self.assertEqual(BuildsHistory.query().get().build_ids, expected_build_ids)