def test_can_add_legislator(self, logging_mock): date_of_birth = date_to_timestamp(datetime.utcnow().date()) data = {'name': 'Marcelo Jorge Vieira', 'date_of_birth': date_of_birth} legislator = Legislator.add_legislator(self.db, data) expect(legislator.name).to_equal('Marcelo Jorge Vieira') expect(logging_mock.mock_calls).to_include( call.debug('Added legislator: "%s"', 'Marcelo Jorge Vieira') )
def post(self): post_data = loads(self.request.body) name = post_data.get('name') if not name: self.set_status(400, 'Invalid legislator') return data = { 'name': name, 'picture': post_data.get('picture'), 'website': post_data.get('website'), 'email': post_data.get('email'), 'gender': post_data.get('gender'), 'date_of_birth': post_data.get('date_of_birth'), 'about': post_data.get('about'), } legislator = Legislator.add_legislator(self.db, data) self.write_json(legislator.to_dict())