def _insert_user(): """Insert starting default entries into the User table. Args: None Returns: None """ # Insert into User if user.idx_exists(1) is False: password = ''.join( random.SystemRandom().choice(string.ascii_uppercase + string.digits) for _ in range(50)) user.insert_row( DbRowUser(username='******', password=data.hashstring(password), first_name='pattoo', last_name='pattoo', user_type=1, change_password=1, enabled=0)) # Insert admin into User table if user.idx_exists(2) is False: password = '******' user.insert_row( DbRowUser(username='******', password=data.hashstring(password), first_name='admin', last_name='admin', user_type=0, change_password=0, enabled=1))
def _insert_user(): """Insert starting default entries into the User table. Args: None Returns: default_users: dictionary containing the default user credentials """ default_users = [] # Insert into User if user.idx_exists(1) is False: # Creating initial password password = data.hashstring(''.join(random.SystemRandom().choice( string.ascii_uppercase + string.digits) for _ in range(50))) # Inserting default user user.insert_row( DbRowUser( username='******', password=password, first_name='pattoo', last_name='pattoo', role=1, password_expired=1, enabled=0) ) default_users.append(('pattoo', password, 1)) # Insert admin into User table if user.idx_exists(2) is False: # Creating initial password password = data.hashstring(''.join(random.SystemRandom().choice( string.ascii_uppercase + string.digits) for _ in range(50))) user.insert_row( DbRowUser( username='******', password=password, first_name='admin', last_name='admin', role=0, password_expired=1, enabled=1) ) default_users.append(('admin', password, 0)) return default_users
def test_idx_exists(self): """Testing method or function named "idx_exists".""" # Add entry to database uname = data.hashstring(str(random())) passwrd = data.hashstring(str(random())) f_name = data.hashstring(str(random())) l_name = data.hashstring(str(random())) user.insert_row( DbRowUser( username=uname, password=passwrd, first_name=f_name, last_name=l_name, role=1, password_expired=1, enabled=0 ) ) # Make sure it exists idx_user = user.exists(uname) # Verify that the index exists result = user.idx_exists(idx_user) self.assertTrue(result)
def test__insert_user(self): """Testing method or function named "_insert_user".""" _insert_user() # Make sure user exists idx_user = user.exists('pattoo') self.assertEqual(idx_user, 1) # Check if index exists result = user.idx_exists(1) self.assertTrue(result)
def test_insert_row(self): """Testing method or function named "insert_row".""" # Add an entry to the database uname = data.hashstring(str(random())) password = data.hashstring(str(random())) expected = DbRowUser( username=uname, password=password, first_name=data.hashstring(str(random())), last_name=data.hashstring(str(random())), role=1, password_expired=1, enabled=0 ) user.insert_row(expected) # Make sure it exists idx_user = user.exists(uname) # Verify the index exists result = user.idx_exists(idx_user) self.assertTrue(result) # Verify that all the parameters match with db.db_query(20091) as session: row = session.query( User).filter(User.username == uname.encode()).one() self.assertEqual(expected.username, row.username.decode()) self.assertEqual(expected.first_name, row.first_name.decode()) self.assertEqual(expected.last_name, row.last_name.decode()) self.assertEqual(expected.role, row.role) self.assertEqual(expected.password_expired, row.password_expired) self.assertEqual(expected.enabled, row.enabled) # Test password salt = row.password.decode().split('$')[2] password_hash = crypt.crypt(password, '$6${}'.format(salt)) self.assertEqual(row.password.decode(), password_hash)