def test_find_existing_with_all_data(self):
test_data = self._prepopulate_datastore(year=self.year, week=self.week, count=self.expected_count)
spread = Spread()
key = spread._generate_key(year=self.year, week=self.week)
data = spread._find_existing_entries(key, test_data)
self.assertEqual(len(data), self.expected_count)
for game in test_data:
self.assertTrue(game['game_id'] in data)
def test_find_existing_with_some_data(self):
test_data = self._generate_data(year=self.year, week=self.week, count=self.expected_count)
spread = Spread()
key = spread._generate_key(year=self.year, week=self.week)
# Preload half
preload = []
preload_count = self.expected_count/2
for i in range(preload_count):
spread_data = {'parent': key}
spread_data.update(test_data[i])
preload.append(SpreadModel(**spread_data))
ndb.put_multi(preload)
data = spread._find_existing_entries(key, test_data)
self.assertEqual(len(data), preload_count)
for game in preload:
self.assertTrue(game.game_id in data)
