def test_user_creation(): try: del_user = User.get('user_id', 'fake-id') del_user.delete_instance() except: pass user = User(user_id='fake-id', password='******') user.save() test_user = User.get('user_id', 'fake-id') assert user == test_user user.delete_instance()
def test_unique_user(): try: del_user = User.get('user_id', 'fake-id') del_user.delete_instance() except: pass user = User(user_id='fake-id', password='******') user.save() with pytest.raises(IntegrityError): duplicate_user = User(user_id='fake-id', password='******') duplicate_user.save() user.delete_instance()
def create_user(user_id, password='******', third_party_info=None, **user_attributes): """Create user, depending on global settings this will be in production or info DB.""" user = User( **{ 'user_id': user_id.lower(), 'password': password, 'third_party_info': third_party_info, **user_attributes }) user.save() info("User %s created successfully, with token: %s", user_id, user.token) return user
def test_delete_user(): try: del_user = User.get('user_id', 'fake-id') del_user.delete_instance() except: pass user = User(user_id='fake-id', password='******') user.save() print(user.__dict__) user.delete_instance() test_user = User.get('user_id', 'fake-id') assert test_user is None