def test_add_bucketlist(self): """ Test whether a user can add a bucketlist """ user = User('kennlebu', 'password', 'Kenneth', 'Lebu', 24) user.create_bucketlist("Go to China", datetime.now().strftime('%Y-%m-%d')) user.create_bucketlist("Buy an island", datetime.now().strftime('%Y-%m-%d')) # Check whether the bucket lists have been created self.assertEqual(2, len(user.bucketlists), msg='User should have 2 bucketlists') # Check whether the created objects are of BucketList type for item in user.bucketlists: assert isinstance(item, BucketList)
def test_remove_bucketlist(self): """ Test whether a user can remove a bucketlist """ user = User('kennlebu', 'password', 'Kenneth', 'Lebu', 24) # First add the bucket lists and verify that they exist user.create_bucketlist("Go to China", datetime.now().strftime('%Y-%m-%d')) user.create_bucketlist("Buy an island", datetime.now().strftime('%Y-%m-%d')) self.assertEqual(2, len(user.bucketlists), msg='User should have 2 bucketlists') # Remove the bucketlists and test whether they have been removed user.delete_bucketlist("Go to China") self.assertEqual(1, len(user.bucketlists), msg='User should have 1 bucketlist') user.delete_bucketlist("Buy an island") self.assertEqual(0, len(user.bucketlists), msg='User should have 0 bucketlists') # Test removing a bucketlist that does not exist user.create_bucketlist("Eat a shark", datetime.now().strftime('%Y-%m-%d')) self.assertEqual(user.delete_bucketlist("Eat a snake"), "Bucket list not found", msg="Should return \'Bucket list not found\'")