def test_save_person_add_person_in_db(self): data = {'facebook_id': '123', 'username': '******', 'name': 'name_test', 'gender': 'male'} self.assertEqual(0, Person.query.count()) Person.save_person(**data) self.assertEqual(1, Person.query.count())
def test_save_person_twice_should_raises_exception(self): data = {'facebook_id': '123', 'username': '******', 'name': 'name_test', 'gender': 'male'} Person.save_person(**data) self.assertEqual(1, Person.query.count()) with self.assertRaises(UserFound): Person.save_person(**data) self.assertEqual(1, Person.query.count())
def post(self, facebook_id): try: user_data = Facebook.get_user_data(facebook_id=facebook_id) except UserNotFound: logger.error('POST /person/ facebook_id={} - Invalid facebook_id'.format(facebook_id)) return {'error': 'Invalid facebook_id.'}, 404 try: Person.save_person(**user_data) logger.info('POST /person/ facebook_id={} - Successul inserted'.format(facebook_id)) except UserFound: logger.error('POST /person/ facebook_id={} - Already inserted'.format(facebook_id)) return {'error': 'User already inserted.'}, 400 return {'message': 'ok'}, 201