def testUserCreation(self):
'''Create user'''
self.user = models.Users(username='******', email='*****@*****.**', password=generate_password_hash('welcome1'))
session.add(self.user)
session.commit()
utest = session.query(models.Users).filter_by(email='*****@*****.**').first()
self.assertEqual(utest.username, 'todd')
def testUserCreation(self):
'''Create user'''
self.user = models.Users(username='******',
email='*****@*****.**',
password=generate_password_hash('welcome1'))
session.add(self.user)
session.commit()
utest = session.query(
models.Users).filter_by(email='*****@*****.**').first()
self.assertEqual(utest.username, 'todd')
def setUp(self):
'''Database setUp'''
Base.metadata.create_all(engine)
# Create an example user
self.user = models.Users(username="******", email="*****@*****.**",
password=generate_password_hash("test"))
session.add(self.user)
session.commit()
user = session.query(models.Users).filter_by(email='*****@*****.**').first()
print(user.email)
self.assertTrue(user.email, '*****@*****.**')
self.assertTrue(user.email, '*****@*****.**')
def adduser():
name = input("Name: ")
email = input("Email: ")
if session.query(Users).filter_by(email=email).first():
print('User with that email address already exists')
return
password = ''
password_2 = ''
while not (password and password_2) or password != password_2:
password = getpass('Password: '******'Re-enter password: ')
user = Users(username=name, email=email,
password=generate_password_hash(password))
session.add(user)
session.commit()
def setUp(self):
'''Database setUp'''
Base.metadata.create_all(engine)
# Create an example user
self.user = models.Users(username="******",
email="*****@*****.**",
password=generate_password_hash("test"))
session.add(self.user)
session.commit()
user = session.query(
models.Users).filter_by(email='*****@*****.**').first()
print(user.email)
self.assertTrue(user.email, '*****@*****.**')
self.assertTrue(user.email, '*****@*****.**')
def testGoal(self):
'''Create goal entries'''
self.user = models.Users(username='******', email='*****@*****.**', password=generate_password_hash('welcome1'))
session.add(self.user)
session.commit()
goal = models.Goals(user_id=self.user.id, weight_goal=150, health_goal=.5)
session.add(goal)
session.commit()
gtest = session.query(models.Goals) \
.filter_by(user_id=self.user.id) \
.first()
self.assertEqual(gtest.weight_goal, 150)
def testRank(self):
'''Create ranks'''
try:
self.rank1 = models.Ranks(rank=1, rankdesc='good')
self.rank2 = models.Ranks(rank=2, rankdesc='ok')
self.rank3 = models.Ranks(rank=3, rankdesc='bad')
session.add_all([self.rank1, self.rank2, self.rank3])
session.commit()
r1test = session.query(models.Ranks).filter_by(rank=1).first()
r2test = session.query(models.Ranks).filter_by(rank=2).first()
r3test = session.query(models.Ranks).filter_by(rank=3).first()
self.assertEqual(r1test.rankdesc, 'good')
self.assertEqual(r2test.rankdesc, 'ok')
self.assertEqual(r3test.rankdesc, 'bad')
except sqlalchemy.exc.DataError as error:
print(error)
def testRank(self):
'''Create ranks'''
try:
self.rank1 = models.Ranks(rank=1, rankdesc='good')
self.rank2 = models.Ranks(rank=2, rankdesc='ok')
self.rank3 = models.Ranks(rank=3, rankdesc='bad')
session.add_all([self.rank1, self.rank2, self.rank3])
session.commit()
r1test = session.query(models.Ranks).filter_by(rank=1).first()
r2test = session.query(models.Ranks).filter_by(rank=2).first()
r3test = session.query(models.Ranks).filter_by(rank=3).first()
self.assertEqual(r1test.rankdesc, 'good')
self.assertEqual(r2test.rankdesc, 'ok')
self.assertEqual(r3test.rankdesc, 'bad')
except sqlalchemy.exc.DataError as error:
print(error)
def testWeightEntries(self):
'''Create weight entries'''
now = datetime.datetime.now().strftime("%Y-%m-%d")
self.user = models.Users(username='******', email='*****@*****.**', password=generate_password_hash('welcome1'))
session.add(self.user)
session.commit()
weight = models.Weight(weight=200, weight_date=now, user_id=self.user.id)
session.add(weight)
session.commit()
testweight = session.query(models.Weight) \
.filter_by(user_id = self.user.id) \
.filter_by(weight=200) \
.first()
self.assertEqual(testweight.weight, 200)
def testGoal(self):
'''Create goal entries'''
self.user = models.Users(username='******',
email='*****@*****.**',
password=generate_password_hash('welcome1'))
session.add(self.user)
session.commit()
goal = models.Goals(user_id=self.user.id,
weight_goal=150,
health_goal=.5)
session.add(goal)
session.commit()
gtest = session.query(models.Goals) \
.filter_by(user_id=self.user.id) \
.first()
self.assertEqual(gtest.weight_goal, 150)
def testFoodDiary(self):
'''Create food entries'''
now = datetime.datetime.now().strftime("%Y-%m-%d")
self.user = models.Users(username='******', email='*****@*****.**', password=generate_password_hash('welcome1'))
session.add(self.user)
session.commit()
rank1 = models.Ranks(rank=1, rankdesc='Good')
rank2 = models.Ranks(rank=2, rankdesc='Ok')
rank3 = models.Ranks(rank=3, rankdesc='Bad')
session.add_all([rank1, rank2, rank3])
session.commit()
food1 = models.Food(food='TEST FOOD 1', food_date=now, rank_id=1, user_id=self.user.id)
food2 = models.Food(food='TEST FOOD 1', food_date=now, rank_id=2, user_id=self.user.id)
food3 = models.Food(food='TEST FOOD 1', food_date=now, rank_id=3, user_id=self.user.id)
session.add_all([food1, food2, food3])
session.commit()
foodcount = session.query(models.Food).count()
self.assertEqual(foodcount, 3)
def testWeightEntries(self):
'''Create weight entries'''
now = datetime.datetime.now().strftime("%Y-%m-%d")
self.user = models.Users(username='******',
email='*****@*****.**',
password=generate_password_hash('welcome1'))
session.add(self.user)
session.commit()
weight = models.Weight(weight=200,
weight_date=now,
user_id=self.user.id)
session.add(weight)
session.commit()
testweight = session.query(models.Weight) \
.filter_by(user_id = self.user.id) \
.filter_by(weight=200) \
.first()
self.assertEqual(testweight.weight, 200)
def testFoodDiary(self):
'''Create food entries'''
now = datetime.datetime.now().strftime("%Y-%m-%d")
self.user = models.Users(username='******',
email='*****@*****.**',
password=generate_password_hash('welcome1'))
session.add(self.user)
session.commit()
rank1 = models.Ranks(rank=1, rankdesc='Good')
rank2 = models.Ranks(rank=2, rankdesc='Ok')
rank3 = models.Ranks(rank=3, rankdesc='Bad')
session.add_all([rank1, rank2, rank3])
session.commit()
food1 = models.Food(food='TEST FOOD 1',
food_date=now,
rank_id=1,
user_id=self.user.id)
food2 = models.Food(food='TEST FOOD 1',
food_date=now,
rank_id=2,
user_id=self.user.id)
food3 = models.Food(food='TEST FOOD 1',
food_date=now,
rank_id=3,
user_id=self.user.id)
session.add_all([food1, food2, food3])
session.commit()
foodcount = session.query(models.Food).count()
self.assertEqual(foodcount, 3)
def setranks():
rank1 = Ranks(rank=1, rankdesc='Good')
rank2 = Ranks(rank=2, rankdesc='Ok')
rank3 = Ranks(rank=3, rankdesc='Bad')
session.add_all([rank1, rank2, rank3])
session.commit()
