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')
