def test_exists(self):
"""Testing method or function named "exists"."""
# Create a translation
uname = data.hashstring(str(random()))
passwrd = data.hashstring(str(random()))
f_name = data.hashstring(str(random()))
l_name = data.hashstring(str(random()))
# Make sure it does not exist
result = user.exists(uname)
self.assertFalse(bool(result))
# Add database row
user.insert_row(
DbRowUser(
username=uname,
password=passwrd,
first_name=f_name,
last_name=l_name,
role=1,
password_expired=0,
enabled=0
)
)
# Make sure it exists
result = user.exists(uname)
self.assertTrue(bool(result))
def main():
"""Test all the pattoo modules with unittests.
Args:
None
Returns:
None
"""
# Set up parser
parser = argparse.ArgumentParser()
parser.add_argument(
'--username', '-u', help='Username to create',
type=str, default='*****@*****.**')
args = parser.parse_args()
# Determine whether user already exists
username = args.username
if bool(user.exists(username)) is False:
row = DbRowUser(
username=username,
password=data.hashstring(str(random())),
first_name='Pattoo: {}'.format(username),
last_name='Test: {}'.format(username),
enabled=True,
)
user.insert_row(row)
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_exists(self):
"""Testing method or function named "exists"."""
# Add user entry to database
uname = data.hashstring(str(random()))
passwrd = data.hashstring(str(random()))
salt_ = 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,
role=0,
password_expired=1,
last_name=l_name,
enabled=0,
))
# Make sure user entry exists
idx_user = user.exists(uname)
# Add chart entry to database
chart_name = data.hashstring(str(random()))
chart_checksum = data.hashstring(str(random()))
chart.insert_row(
DbRowChart(name=chart_name, checksum=chart_checksum, enabled=0))
# Make sure chart entry exists
idx_chart = chart.exists(chart_checksum)
# Make sure favorite does not exist
result = favorite.exists(idx_chart, idx_user)
self.assertFalse(bool(result))
# Add favorite to database
favorite.insert_row(
DbRowFavorite(idx_chart=idx_chart,
idx_user=idx_user,
order=0,
enabled=0))
# Make sure favorite exists
result = favorite.exists(idx_chart, idx_user)
self.assertTrue(bool(result))
def test_idx_exists(self):
"""Testing method or function named "idx_exists"."""
# Add user 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,
user_type=1,
change_password=1,
enabled=0))
# Make sure user entry exists
idx_user = user.exists(uname)
self.assertTrue(bool(idx_user))
# Add chart entry to database
chart_name = data.hashstring(str(random()))
chart_checksum = data.hashstring(str(random()))
chart.insert_row(
DbRowChart(name=chart_name, checksum=chart_checksum, enabled=0))
# Make sure chart entry exists
idx_chart = chart.exists(chart_checksum)
self.assertTrue(bool(idx_chart))
# Add favorite to database
favorite.insert_row(
DbRowFavorite(idx_chart=idx_chart,
idx_user=idx_user,
order=0,
enabled=0))
# Make sure the favorite exists
idx_favorite = favorite.exists(idx_chart, idx_user)
# Verify that the index exists
result = favorite.idx_exists(idx_favorite)
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)