Add more tests

tests/test_main.py

@@ -6,6 +6,10 @@ from httpx import Response
 from discord_rss_bot.main import app, encode_url
 
 client: TestClient = TestClient(app)
+webhook_name: str = "Hello, I am a webhook!"
+webhook_url: str = "https://discord.com/api/webhooks/1234567890/abcdefghijklmnopqrstuvwxyz"
+feed_url: str = "https://lovinator.space/rss_test.xml"
+encoded_feed_url: str = encode_url(webhook_url)
 
 
 def test_read_main() -> None:
@@ -14,14 +18,31 @@ def test_read_main() -> None:
     assert response.status_code == 200
 
 
-def test_add() -> None:
-    """Test the /add page."""
-    response: Response = client.get("/add")
-    assert response.status_code == 200
-
-
 def test_search() -> None:
     """Test the /search page."""
+    # Remove the feed if it already exists before we run the test.
+    feeds: Response = client.get("/")
+    if feed_url in feeds.text:
+        client.post("/remove", data={"feed_url": feed_url})
+        client.post("/remove", data={"feed_url": encoded_feed_url})
+
+    # Delete the webhook if it already exists before we run the test.
+    response: Response = client.post("/delete_webhook", data={"webhook_url": webhook_url})
+
+    # Add the webhook.
+    response: Response = client.post("/add_webhook", data={"webhook_name": webhook_name, "webhook_url": webhook_url})
+    assert response.status_code == 200
+
+    # Add the feed.
+    response: Response = client.post("/add", data={"feed_url": feed_url, "webhook_dropdown": webhook_name})
+    assert response.status_code == 200
+
+    # Check that the feed was added.
+    response = client.get("/")
+    assert response.status_code == 200
+    assert feed_url in response.text
+
+    # Search for an entry.
     response: Response = client.get("/search/?query=a")
     assert response.status_code == 200
 
@@ -31,3 +52,131 @@ def test_encode_url() -> None:
     before: Literal["https://www.google.com/"] = "https://www.google.com/"
     after: Literal["https%3A//www.google.com/"] = "https%3A//www.google.com/"
     assert encode_url(url_to_quote=before) == after
+
+
+def test_add_webhook() -> None:
+    """Test the /add_webhook page."""
+    # Delete the webhook if it already exists before we run the test.
+    response: Response = client.post("/delete_webhook", data={"webhook_url": webhook_url})
+
+    # Add the webhook.
+    response: Response = client.post("/add_webhook", data={"webhook_name": webhook_name, "webhook_url": webhook_url})
+    assert response.status_code == 200
+
+    # Check that the webhook was added.
+    response = client.get("/webhooks")
+    assert response.status_code == 200
+    assert webhook_name in response.text
+
+
+def test_create_feed() -> None:
+    """Test the /create_feed page."""
+    # Remove the feed if it already exists before we run the test.
+    feeds: Response = client.get("/")
+    if feed_url in feeds.text:
+        client.post("/remove", data={"feed_url": feed_url})
+        client.post("/remove", data={"feed_url": encoded_feed_url})
+
+    # Add the feed.
+    response: Response = client.post("/add", data={"feed_url": feed_url, "webhook_dropdown": webhook_name})
+    assert response.status_code == 200
+
+    # Check that the feed was added.
+    response = client.get("/")
+    assert response.status_code == 200
+    assert feed_url in response.text
+
+
+def test_pause_feed() -> None:
+    """Test the /pause_feed page."""
+    # Remove the feed if it already exists before we run the test.
+    feeds: Response = client.get("/")
+    if feed_url in feeds.text:
+        client.post("/remove", data={"feed_url": feed_url})
+        client.post("/remove", data={"feed_url": encoded_feed_url})
+
+    # Add the feed.
+    response: Response = client.post("/add", data={"feed_url": feed_url, "webhook_dropdown": webhook_name})
+
+    # Unpause the feed if it is paused.
+    feeds: Response = client.get("/")
+    if "Paused" in feeds.text:
+        response: Response = client.post("/unpause", data={"feed_url": feed_url})
+        assert response.status_code == 200
+
+    # Pause the feed.
+    response: Response = client.post("/pause", data={"feed_url": feed_url})
+    assert response.status_code == 200
+
+    # Check that the feed was paused.
+    response = client.get("/")
+    assert response.status_code == 200
+    assert feed_url in response.text
+
+
+def test_unpause_feed() -> None:
+    """Test the /unpause_feed page."""
+    # Remove the feed if it already exists before we run the test.
+    feeds: Response = client.get("/")
+    if feed_url in feeds.text:
+        client.post("/remove", data={"feed_url": feed_url})
+        client.post("/remove", data={"feed_url": encoded_feed_url})
+
+    # Add the feed.
+    response: Response = client.post("/add", data={"feed_url": feed_url, "webhook_dropdown": webhook_name})
+
+    # Pause the feed if it is unpaused.
+    feeds: Response = client.get("/")
+    if "Paused" not in feeds.text:
+        response: Response = client.post("/pause", data={"feed_url": feed_url})
+        assert response.status_code == 200
+
+    # Unpause the feed.
+    response: Response = client.post("/unpause", data={"feed_url": feed_url})
+    assert response.status_code == 200
+
+    # Check that the feed was unpaused.
+    response = client.get("/")
+    assert response.status_code == 200
+    assert feed_url in response.text
+
+
+def test_delete_webhook():
+    """Test the /delete_webhook page."""
+    # Remove the feed if it already exists before we run the test.
+    feeds: Response = client.get("/webhooks")
+    if webhook_url in feeds.text:
+        client.post("/delete_webhook", data={"webhook_url": webhook_url})
+
+    # Add the webhook.
+    response: Response = client.post("/add_webhook", data={"webhook_name": webhook_name, "webhook_url": webhook_url})
+
+    # Delete the webhook.
+    response: Response = client.post("/delete_webhook", data={"webhook_url": webhook_url})
+    assert response.status_code == 200
+
+    # Check that the webhook was added.
+    response = client.get("/webhooks")
+    assert response.status_code == 200
+    assert webhook_name not in response.text
+
+
+def test_remove_feed():
+    """Test the /remove page."""
+    # Remove the feed if it already exists before we run the test.
+    feeds: Response = client.get("/")
+    if feed_url in feeds.text:
+        client.post("/remove", data={"feed_url": feed_url})
+        client.post("/remove", data={"feed_url": encoded_feed_url})
+
+    # Add the feed.
+    response: Response = client.post("/add", data={"feed_url": feed_url, "webhook_dropdown": webhook_name})
+
+    # Remove the feed.
+    response: Response = client.post("/remove", data={"feed_url": feed_url})
+    assert response.status_code == 200
+
+    # Check that the feed was removed.
+    response = client.get("/")
+    assert response.status_code == 200
+    assert feed_url not in response.text