From 784e46010ef4155dcede3ce0bd43bc04037962b2 Mon Sep 17 00:00:00 2001
From: Zhuoyun Wei <wzyboy@wzyboy.org>
Date: Sun, 9 Jul 2023 01:48:56 -0700
Subject: [PATCH] test: index and search view

---
 tests/test_views.py | 38 ++++++++++++++++++++++++++------------
 1 file changed, 26 insertions(+), 12 deletions(-)

diff --git a/tests/test_views.py b/tests/test_views.py
index b01c5ed..4a5ebbe 100644
--- a/tests/test_views.py
+++ b/tests/test_views.py
@@ -1,3 +1,29 @@
+class TestIndexView:
+    def test_index(self, client):
+        resp = client.get('/tweet/')
+        assert '<p>Number of Tweets: <code>2</code>' in resp.text
+        assert 'wzyboy' in resp.text
+        assert 'Uucky_Lee' in resp.text
+
+    def test_index_with_default_user(self, client):
+        client.application.config['T_DEFAULT_USER'] = 'wzyboy'
+        resp = client.get('/tweet/')
+        assert 'wzyboy' in resp.text
+        assert 'Uucky_Lee' not in resp.text
+
+
+class TestSearchView:
+    keywords = ('please connect a keyboard', 'This guy found a starving dog')
+
+    def test_search(self, client):
+        resp = client.get(
+            '/tweet/search.html',
+            query_string={'q': '*'}
+        )
+        for kw in self.keywords:
+            assert kw in resp.text
+
+
 class TestMediaReplacement:
     tweet_id = '1615425412921987074'
     media_filename = 'Fmsk2gHacAAJGL0.jpg'
@@ -22,15 +48,3 @@ def test_fs_media(self, client):
         client.application.config['T_MEDIA_FS_PATH'] = './media'
         resp = client.get(f'/tweet/{self.tweet_id}.html')
         assert f'/tweet/media/pbs.twimg.com/media/{self.media_filename}' in resp.text
-
-
-class TestSearchView:
-    keywords = ('please connect a keyboard', 'This guy found a starving dog')
-
-    def test_search(self, client):
-        resp = client.get(
-            '/tweet/search.html',
-            query_string={'q': '*'}
-        )
-        for kw in self.keywords:
-            assert kw in resp.text