def test_tried_by_relationship(self):
user = helper.create_test_user()
strain = helper.populate_test_strain()
db.session.add_all([user, strain])
self.assertEqual(strain.tried_by.all(), [])
strain.tried_by.append(user)
self.assertEqual(strain.tried_by.all(), [user])
def test_has_tried(self):
user = helper.create_test_user()
strain = helper.populate_test_strain()
db.session.add_all([user, strain])
self.assertFalse(user.has_tried(strain))
user.tried.append(strain)
self.assertTrue(user.has_tried(strain))
def test_has_not_tried(self):
strains = helper.populate_test_strains(strain_dict=donezo.done)
db.session.add_all(strains)
user = helper.create_test_user()
db.session.add(user)
s = Strain.query.all()
user.tried.extend(s[:5])
self.assertEqual(set(user.tried.all()), set(s[:5]))
self.assertEqual(set(user.has_not_tried()), set(s[5:]))
def test_untry_strain(self):
strain = helper.populate_test_strain()
user = helper.create_test_user()
db.session.add_all([strain, user])
user.tried.append(strain)
self.assertEqual(user.tried.all(), [strain])
user.untry_strain(strain)
self.assertEqual(user.tried.all(), [])
self.assertEqual(user.tried.count(), 0)
def test_repr(self):
user = helper.create_test_user()
self.assertEqual(str(user), '<User Test>')
def test_load_user(self):
user = helper.create_test_user()
db.session.add(user)
self.assertEqual(load_user(1), user)
self.assertIsNone(load_user(2))
