def test_update_user(self):
""" Test the user/update API call.
We attempt to update a user's details.
"""
# Create a random user and open a session for that user.
user = self.create_random_user()
err_msg = self.open_session(user.user_id, user.password)
self.assertEqual(err_msg, None)
# Now try changing the user's name, password and PIN number.
new_name = utils.random_letters()
new_password = utils.random_letters()
new_pin_number = utils.random_digits(min_length=4, max_length=4)
changes = {"name" : new_name,
"password" : new_password,
"pin_number" : new_pin_number}
success,response = self.call_api("tahua.api.user.update",
include_session_key=True,
pin_number=user.pin_number,
changes=changes)
if not success: self.fail(response)
# Finally, check that the user has been updated.
updated_user = User.objects.get(user_id=user.user_id)
self.assertEqual(updated_user.name, new_name)
self.assertEqual(updated_user.password, new_password)
self.assertEqual(updated_user.pin_number, new_pin_number)
def test_new_user(self):
""" Test the user/new API call.
We attempt to create a new unique user, and then try creating a
second user with the same user ID.
"""
# Call the API to create a user.
user_id = self.random_user_id()
name = utils.random_letters()
password = utils.random_letters()
pin_number = utils.random_digits(min_length=4, max_length=4)
success,response = self.call_api("tahua.api.user.new",
user_id=user_id,
name=name,
password=password,
pin_number=pin_number)
if not success: self.fail(response)
# Check that the newly-created user actually exists.
user = User.objects.get(user_id=user_id)
self.assertEqual(user.user_id, user_id)
self.assertEqual(user.name, name)
self.assertEqual(user.password, password)
self.assertEqual(user.pin_number, pin_number)
def random_user_id(self):
""" Return a random but unique ID for a User.
"""
while True:
prefix = utils.random_letters()
suffix = utils.random_digits()
user_id = prefix + ":" + suffix
if not User.objects.filter(user_id=user_id).exists():
return user_id
def create_random_user(self):
""" Create and return a new randomly-named User.
Note that we create the user directly, rather than going via the
API.
"""
user = User()
user.user_id = self.random_user_id()
user.name = utils.random_letters()
user.password = utils.random_letters()
user.pin_number = utils.random_digits(min_length=4, max_length=4)
user.save()
return user
