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