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()