Add tests for redirect, duplicate URLs, stats, and 404 handling

app/test_main.py

@@ -28,3 +28,35 @@ def test_shorten_url():
     assert "short_url" in data
     short_url = data["short_url"]
     assert short_url.startswith("http://localhost:8000/")
+
+def test_redirect():
+    response = client.post("/shorten", json={"url": "https://google.com"})
+    short_url = response.json()["short_url"]
+    code = short_url.split("/")[-1]
+
+    redirect = client.get(f"/{code}", allow_redirects=False)
+    assert redirect.status_code == 307
+
+def test_duplicate_url_returns_same_code():
+    r1 = client.post("/shorten", json={"url": "https://example.com"})
+    r2 = client.post("/shorten", json={"url": "https://example.com"})
+
+    assert r1.status_code == 200
+    assert r2.status_code == 200
+    assert r1.json()["short_url"] == r2.json()["short_url"]
+
+def test_stats_endpoint():
+    response = client.post("/shorten", json={"url": "https://stats-test.com"})
+    code = response.json()["short_url"].split("/")[-1]
+
+    # trigger one redirect
+    client.get(f"/{code}", allow_redirects=False)
+
+    stats = client.get(f"/stats/{code}")
+    assert stats.status_code == 200
+    data = stats.json()
+    assert data["clicks"] == 1
+
+def test_redirect_404():
+    response = client.get("/nonexistent", allow_redirects=False)
+    assert response.status_code == 404