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