def test_retrieve_planet_not_found(self): planet_id = 'yyyyyyyyyyy' planet = { "planet_id": planet_id, "planet_name": "Marcela", "terrain": "Arenoso", "climate": "Deserto" } self.create_table() db = DBConnection() response = db.retrieve_planet('planet_id', planet_id) self.assertEquals(response, {})
def test_retrieve_planet(self): planet_id = 'xxxxxxxxxxxx' planet = { "planet_id": planet_id, "planet_name": "Renata", "terrain": "Arenoso", "climate": "Deserto" } table = self.create_table() table.put_item(Item=planet) db = DBConnection() response = db.retrieve_planet('planet_id', planet_id) self.assertEquals(response, planet)
def get(self): try: search_by = self.get_argument("search_by") value = self.get_argument("value") db = DBConnection() planet = db.retrieve_planet(search_by, value) if not planet: self.set_status(404) self.finish("Planet not found") return planet['films'] = fetch_films(planet['planet_name']) self.write({'planet': planet}) except tornado.web.MissingArgumentError as argError: logger.exception("SearchPlanet::get") raise argError except Exception as e: logger.exception("SearchPlanet::get") raise Exception('Error to search planet')