def tearDown(self): RedisLink.factory().flushdb()
def setUp(self): RedisLink.factory().flushdb()
import unittest from models import User, Post, RedisLink from time import time RedisLink.factory(db=15) class TestUser(unittest.TestCase): def setUp(self): RedisLink.factory().flushdb() def tearDown(self): RedisLink.factory().flushdb() def test_is_authenticated(self): user1 = User(1, 'john') self.assertTrue(user1.is_authenticated()) user2 = User(0, 'peter') self.assertFalse(user2.is_authenticated()) def _create_test_users(self): data = [['john', '111111'], ['peter', '222222'], ['mark', '333333'], ['donald', '44444']] return [User.create_new(username, pwd) for (username, pwd) in data ] def _mass_test_follow(self): users = self._create_test_users() users[0].follow(users[1].id) users[0].follow(users[2].id) users[2].follow(users[1].id)