def test_add_user_already_existing(self, _):
rah = RemoteAccessHandler(Mock())
rah.os_util = MockOSUtil()
tstpassword = "******"
tstuser = "******"
expiration_date = datetime.utcnow() + timedelta(days=1)
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
users = get_user_dictionary(rah.os_util.get_users())
self.assertTrue(tstuser in users,
"{0} missing from users".format(tstuser))
self.assertEqual(1, len(users.keys()))
actual_user = users[tstuser]
self.assertEqual(actual_user[7],
(expiration_date +
timedelta(days=1)).strftime("%Y-%m-%d"))
# add the new duplicate user, ensure it's not created and does not overwrite the existing user.
# this does not test the user add function as that's mocked, it tests processing skips the remaining
# calls after the initial failure
new_user_expiration = datetime.utcnow() + timedelta(days=5)
self.assertRaises(RemoteAccessError, rah.add_user, tstuser, pwd,
new_user_expiration)
# refresh users
users = get_user_dictionary(rah.os_util.get_users())
self.assertTrue(
tstuser in users,
"{0} missing from users after dup user attempted".format(tstuser))
self.assertEqual(1, len(users.keys()))
actual_user = users[tstuser]
self.assertEqual(actual_user[7],
(expiration_date +
timedelta(days=1)).strftime("%Y-%m-%d"))
def test_remote_access_handler_run_bad_data(self, _1, _2, _3, _4, _5):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstpassword = "******"
tstuser = "******"
expiration_date = datetime.utcnow() + timedelta(days=1)
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
users = get_user_dictionary(rah.os_util.get_users())
self.assertTrue(tstuser in users, "{0} missing from users".format(tstuser))
rah.run()
self.assertTrue(tstuser in users, "{0} missing from users".format(tstuser))
def test_remote_access_handler_run_bad_data(self, _1, _2, _3, _4, _5):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstpassword = "******"
tstuser = "******"
expiration_date = datetime.utcnow() + timedelta(days=1)
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
users = get_user_dictionary(rah.os_util.get_users())
self.assertTrue(tstuser in users, "{0} missing from users".format(tstuser))
rah.run()
self.assertTrue(tstuser in users, "{0} missing from users".format(tstuser))
def test_add_user_bad_creation_data(self, _1, _2, _3):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstpassword = "******"
tstuser = ""
expiration_date = datetime.utcnow() + timedelta(days=1)
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
self.assertEqual(0, len(rah.os_util.get_users()))
self.assertEqual(1, len(error_messages))
self.assertEqual(0, len(info_messages))
error = "Error adding user {0}. test exception for bad username".format(tstuser)
self.assertEqual(error, error_messages[0])
def test_add_user_bad_password_data(self, _1, _2, _3):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstpassword = ""
tstuser = "******"
expiration_date = datetime.utcnow() + timedelta(days=1)
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
self.assertEqual(0, len(rah.os_util.get_users()))
self.assertEqual(1, len(error_messages))
self.assertEqual(1, len(info_messages))
error = "Error creating user {0}. test exception for bad password".format(tstuser)
self.assertEqual(error, error_messages[0])
self.assertEqual("User deleted {0}".format(tstuser), info_messages[0])
def test_add_user_bad_creation_data(self, _1, _2, _3):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstpassword = "******"
tstuser = ""
expiration_date = datetime.utcnow() + timedelta(days=1)
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
self.assertEqual(0, len(rah.os_util.get_users()))
self.assertEqual(1, len(error_messages))
self.assertEqual(0, len(info_messages))
error = "Error adding user {0}. test exception for bad username".format(
tstuser)
self.assertEqual(error, error_messages[0])
def test_error_add_user(self, _1, _2):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstuser = "******"
expiration = datetime.utcnow() + timedelta(days=1)
pwd = "bad password"
rah.add_user(tstuser, pwd, expiration)
users = get_user_dictionary(rah.os_util.get_users())
self.assertEqual(0, len(users))
self.assertEqual(1, len(error_messages))
self.assertEqual(1, len(info_messages))
error = "Error creating user {0}. [CryptError] Error decoding secret\nInner error: Incorrect padding".\
format(tstuser)
self.assertEqual(error, error_messages[0])
self.assertEqual("User deleted {0}".format(tstuser), info_messages[0])
def test_add_user_bad_password_data(self, _1, _2, _3):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstpassword = ""
tstuser = "******"
expiration_date = datetime.utcnow() + timedelta(days=1)
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
self.assertEqual(0, len(rah.os_util.get_users()))
self.assertEqual(1, len(error_messages))
self.assertEqual(1, len(info_messages))
error = "Error creating user {0}. test exception for bad password".format(
tstuser)
self.assertEqual(error, error_messages[0])
self.assertEqual("User deleted {0}".format(tstuser), info_messages[0])
def test_error_add_user(self, _1, _2):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstuser = "******"
expiration = datetime.utcnow() + timedelta(days=1)
pwd = "bad password"
rah.add_user(tstuser, pwd, expiration)
users = get_user_dictionary(rah.os_util.get_users())
self.assertEqual(0, len(users))
self.assertEqual(1, len(error_messages))
self.assertEqual(1, len(info_messages))
error = "Error creating user {0}. [CryptError] Error decoding secret\nInner error: Incorrect padding".\
format(tstuser)
self.assertEqual(error, error_messages[0])
self.assertEqual("User deleted {0}".format(tstuser), info_messages[0])
def test_add_user(self, _1, _2, _3):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstpassword = "******"
tstuser = "******"
expiration_date = datetime.utcnow() + timedelta(days=1)
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
users = get_user_dictionary(rah.os_util.get_users())
self.assertTrue(tstuser in users, "{0} missing from users".format(tstuser))
actual_user = users[tstuser]
expected_expiration = (expiration_date + timedelta(days=1)).strftime("%Y-%m-%d")
self.assertEqual(actual_user[7], expected_expiration)
self.assertEqual(actual_user[4], "JIT_Account")
self.assertEqual(0, len(error_messages))
self.assertEqual(1, len(info_messages))
self.assertEqual(info_messages[0], "User '{0}' added successfully with expiration in {1}"
.format(tstuser, expected_expiration))
def test_add_user(self, _1, _2, _3):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstpassword = "******"
tstuser = "******"
expiration_date = datetime.utcnow() + timedelta(days=1)
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
users = get_user_dictionary(rah.os_util.get_users())
self.assertTrue(tstuser in users, "{0} missing from users".format(tstuser))
actual_user = users[tstuser]
expected_expiration = (expiration_date + timedelta(days=1)).strftime("%Y-%m-%d")
self.assertEqual(actual_user[7], expected_expiration)
self.assertEqual(actual_user[4], "JIT_Account")
self.assertEqual(0, len(error_messages))
self.assertEqual(1, len(info_messages))
self.assertEqual(info_messages[0], "User '{0}' added successfully with expiration in {1}"
.format(tstuser, expected_expiration))
def test_delete_user(self, _1, _2, _3):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstpassword = "******"
tstuser = "******"
expiration_date = datetime.utcnow() + timedelta(days=1)
expected_expiration = (expiration_date + timedelta(days=1)).strftime("%Y-%m-%d")
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
users = get_user_dictionary(rah.os_util.get_users())
self.assertTrue(tstuser in users, "{0} missing from users".format(tstuser))
rah.delete_user(tstuser)
# refresh users
users = get_user_dictionary(rah.os_util.get_users())
self.assertFalse(tstuser in users)
self.assertEqual(0, len(error_messages))
self.assertEqual(2, len(info_messages))
self.assertEqual("User '{0}' added successfully with expiration in {1}".format(tstuser, expected_expiration),
info_messages[0])
self.assertEqual("User deleted {0}".format(tstuser), info_messages[1])
def test_delete_user(self, _1, _2, _3):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstpassword = "******"
tstuser = "******"
expiration_date = datetime.utcnow() + timedelta(days=1)
expected_expiration = (expiration_date + timedelta(days=1)).strftime("%Y-%m-%d")
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
users = get_user_dictionary(rah.os_util.get_users())
self.assertTrue(tstuser in users, "{0} missing from users".format(tstuser))
rah.delete_user(tstuser)
# refresh users
users = get_user_dictionary(rah.os_util.get_users())
self.assertFalse(tstuser in users)
self.assertEqual(0, len(error_messages))
self.assertEqual(2, len(info_messages))
self.assertEqual("User '{0}' added successfully with expiration in {1}".format(tstuser, expected_expiration),
info_messages[0])
self.assertEqual("User deleted {0}".format(tstuser), info_messages[1])
def test_add_user_already_existing(self, _):
rah = RemoteAccessHandler()
rah.os_util = MockOSUtil()
tstpassword = "******"
tstuser = "******"
expiration_date = datetime.utcnow() + timedelta(days=1)
pwd = tstpassword
rah.add_user(tstuser, pwd, expiration_date)
users = get_user_dictionary(rah.os_util.get_users())
self.assertTrue(tstuser in users, "{0} missing from users".format(tstuser))
self.assertEqual(1, len(users.keys()))
actual_user = users[tstuser]
self.assertEqual(actual_user[7], (expiration_date + timedelta(days=1)).strftime("%Y-%m-%d"))
# add the new duplicate user, ensure it's not created and does not overwrite the existing user.
# this does not test the user add function as that's mocked, it tests processing skips the remaining
# calls after the initial failure
new_user_expiration = datetime.utcnow() + timedelta(days=5)
self.assertRaises(RemoteAccessError, rah.add_user, tstuser, pwd, new_user_expiration)
# refresh users
users = get_user_dictionary(rah.os_util.get_users())
self.assertTrue(tstuser in users, "{0} missing from users after dup user attempted".format(tstuser))
self.assertEqual(1, len(users.keys()))
actual_user = users[tstuser]
self.assertEqual(actual_user[7], (expiration_date + timedelta(days=1)).strftime("%Y-%m-%d"))
