add-unitTest

app/models.py

@@ -462,6 +462,7 @@ class Like(db.Model):
     liked_post = db.relationship('Post', back_populates='liker', lazy='joined')
 
 
+
 class Collect(db.Model):
     __tablename__ = 'collect'
     collecter_id = db.Column(db.Integer, db.ForeignKey('users.id'), primary_key=True)

test/ModelsTest.py

@@ -0,0 +1,24 @@
+import unittest
+from app import models
+
+
+class TestModels(unittest.TestCase):
+
+    def testExist(self):
+        self.assertIsNotNone(models.db)
+        self.assertIsNotNone(models.User)
+        self.assertIsNotNone(models.Role)
+        self.assertIsNotNone(models.Permission)
+        self.assertIsNotNone(models.Post)
+        self.assertIsNotNone(models.AnonymousUser)
+        self.assertIsNotNone(models.Collect)
+        self.assertIsNotNone(models.Comment)
+        self.assertIsNotNone(models.Follow)
+        self.assertIsNotNone(models.Like)
+        self.assertIsNotNone(models.Activity)
+        self.assertIsNotNone(models.Notification)
+        self.assertIsNotNone(models.Want)
+
+
+if __name__ == '__main__':
+    unittest.main()

test/test1.py

@@ -0,0 +1,22 @@
+import unittest
+
+
+class TestStringMethods(unittest.TestCase):
+
+    def test_upper(self):
+        self.assertEqual('foo'.upper(), 'FOO')
+
+    def test_isupper(self):
+        self.assertTrue('FOO'.isupper())
+        self.assertFalse('Foo'.isupper())
+
+    def test_split(self):
+        s = 'hello world'
+        self.assertEqual(s.split(), ['hello', 'world'])
+        # check that s.split fails when the separator is not a string
+        with self.assertRaises(TypeError):
+            s.split(2)
+
+
+if __name__ == '__main__':
+    unittest.main()