コード例 #1
0
ファイル: test_models.py プロジェクト: ptouati/flask-blog
class TestUserModelDb(unittest.TestCase):
    def setUp(self):
        self.db_fd, blog.app.config['DATABASE'] = tempfile.mkstemp()
        self.u = User("Test", "*****@*****.**", "password")
        #self.admin = User.query.get(1)
        db.session.add(self.u)

    def test_id_type(self):
        expected = unicode
        uid = self.u.get_id()
        assert type(uid) == expected

    @raises(RuntimeError)
    def test_auth_token(self):
        expected = "something"
        auth_token = self.u.get_auth_token()
        assert auth_token == expected

    #def tedst_repr(self):
    #expected = r"<User /d*?: '.*'>"
    #expected_type = str
    #representation = repr(self.admin)
    #assert False, representation
    #assert re.match(expected, representation)
    #assert type(representation) == expected_type

    def tearDown(self):
        db.session.rollback()
コード例 #2
0
ファイル: test_models.py プロジェクト: ant1441/flask-blog
class TestUserModelDb(unittest.TestCase):
    def setUp(self):
        self.db_fd, blog.app.config['DATABASE'] = tempfile.mkstemp()
        self.u = User("Test", "*****@*****.**", "password")
        #self.admin = User.query.get(1)
        db.session.add(self.u)

    def test_id_type(self):
        expected = unicode
        uid = self.u.get_id()
        assert type(uid) == expected

    @raises(RuntimeError)
    def test_auth_token(self):
        expected = "something"
        auth_token = self.u.get_auth_token()
        assert auth_token == expected

    #def tedst_repr(self):
        #expected = r"<User /d*?: '.*'>"
        #expected_type = str
        #representation = repr(self.admin)
        #assert False, representation
        #assert re.match(expected, representation)
        #assert type(representation) == expected_type

    def tearDown(self):
        db.session.rollback()