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)