def post(self): planet = json.loads(self.request.body) if 'planet_name' and 'climate' and 'terrain' not in planet: raise Exception('Missing body arguments') db = DBConnection() db.insert_planet(planet) # returnar uma mensagem de sucesso self.write({'planet': planet})
def test_insert_planet(self): planet = { "planet_name": "Marino", "terrain": "blah", "climate": "Seco" } table = self.create_table() db = DBConnection() db.insert_planet(planet) response = table.scan(FilterExpression=Attr('planet_name').eq('Marino')) if 'Items' in response: item = response['Items'][0] self.assertTrue('planet_id' in item) self.assertEquals(item['planet_name'], 'Marino') self.assertEquals(item['terrain'], 'blah') self.assertEquals(item['climate'], 'Seco')