Merge pull request #45 from rzashakeri/develop

✅ test: Add VisitModelTest and test_visit_count

blog/tests/test_models.py

@@ -1,6 +1,8 @@
 from django.test import TestCase
 from django.core.files.uploadedfile import SimpleUploadedFile
-from blog.models import Post, Tag
+from blog.models import Post, Tag, Visit
+
+
 class PostModelTest(TestCase):
     def setUp(self):
         self.tag1 = Tag.objects.create(name='tag1')
@@ -53,7 +55,6 @@ def test_post_tags_blank(self):
         self.assertEqual(post.tags.count(), 0)
 
 
-
 class TagModelTest(TestCase):
 
     def setUp(self):
@@ -80,4 +81,20 @@ def test_ordering_by_name(self):
         tags = Tag.objects.all()
         self.assertEqual(tags[0], tag1)
         self.assertEqual(tags[1], self.tag)
-        self.assertEqual(tags[2], tag2)
+        self.assertEqual(tags[2], tag2)
+
+
+class VisitModelTest(TestCase):
+    def setUp(self):
+        self.post = Post.objects.create(
+            title='Test Post',
+            icon=SimpleUploadedFile('icon.png', b''),
+            content='Test content'
+        )
+        self.visit = Visit.objects.create(post=self.post, ip_address='127.0.0.1')
+
+    def test_visit_str(self):
+        self.assertEqual(str(self.visit), 'Test Post - 127.0.0.1')
+
+    def test_visit_count(self):
+        self.assertEqual(self.post.visit_count, 1)