def test_save_by_id_same(self): generated_data = self._generate_data(year=self.year, week=self.week) test_data = generated_data[0] game_id = test_data['game_id'] spread = Spread() spread.save_by_id(year=self.year, week=self.week, game_id=game_id, data=test_data) test_data['game_line'] += randint(7, 14) spread.save_by_id(year=self.year, week=self.week, game_id=game_id, data=test_data) # Check datastore data = SpreadModel.query(SpreadModel.game_id == game_id).fetch(2) self.assertEqual(len(data), 1) self.assertEqual(data[0].to_dict(), test_data)
def put(self, year, week, game): year = int(year) week = int(week) game = int(game) spread_data = json.loads(self.request.POST['spread']) spread = Spread() result = spread.save_by_id(year=year, week=week, game_id=game, data=spread_data) self._send_response()
def test_save_by_id(self): generated_data = self._generate_data(year=self.year, week=self.week) test_data = generated_data[0] game_id = test_data['game_id'] spread = Spread() count = spread.save_by_id(year=self.year, week=self.week, game_id=game_id, data=test_data) self.assertEqual(count, 1) # Check datastore data = SpreadModel.query(SpreadModel.game_id == game_id).fetch(1) self.assertEqual(data[0].to_dict(), test_data)