Convert test to python unit test

Author: alan412

backend/requirements.txt

@@ -1,14 +1,19 @@
 aniso8601==10.0.1
 blinker==1.9.0
+certifi==2025.8.3
+charset-normalizer==3.4.3
 click==8.2.1
 Flask==3.1.1
 Flask-RESTful==0.3.10
 Flask-SQLAlchemy==3.1.1
+idna==3.10
 itsdangerous==2.2.0
 Jinja2==3.1.6
 MarkupSafe==3.0.2
 pytz==2025.2
+requests==2.32.5
 six==1.17.0
 SQLAlchemy==2.0.42
 typing_extensions==4.14.1
+urllib3==2.5.0
 Werkzeug==3.1.3

backend/test_api.py

@@ -1,116 +1,84 @@
 #!/usr/bin/env python3
 """
-Simple test script to verify the Storage API endpoints work correctly.
+Unit tests to verify the Storage API endpoints work correctly.
 """
 
+import unittest
 import requests
 import json
 
-BASE_URL = "http://localhost:5001"
-
-def test_root_endpoint():
-    """Test the root endpoint"""
-    print("Testing root endpoint...")
-    try:
-        response = requests.get(f"{BASE_URL}/")
-        print(f"Status: {response.status_code}")
-        print(f"Response: {response.json()}")
-        return response.status_code == 200
-    except Exception as e:
-        print(f"Error: {e}")
-        return False
-
-def test_storage_entry():
-    """Test storage entry endpoints"""
-    print("\nTesting storage entry endpoints...")
+class TestStorageAPI(unittest.TestCase):
+    """Test cases for Storage API endpoints"""
+    
+    def setUp(self):
+        """Set up test configuration"""
+        self.base_url = "http://localhost:5001"
 
-    # Save an entry
-    try:
-        response = requests.post(f"{BASE_URL}/entries/test_key", 
+    def test_root_endpoint(self):
+        """Test the root endpoint"""
+        response = requests.get(f"{self.base_url}/")
+        self.assertEqual(response.status_code, 200)
+        self.assertIsInstance(response.json(), dict)
+    
+    def test_storage_entry_operations(self):
+        """Test storage entry endpoints"""
+        # Save an entry
+        response = requests.post(f"{self.base_url}/entries/test_key", 
                                json={"value": "test_value"})
-        print(f"Save entry status: {response.status_code}")
+        self.assertEqual(response.status_code, 200)
 
         # Fetch the entry
-        response = requests.get(f"{BASE_URL}/entries/test_key")
-        print(f"Fetch entry status: {response.status_code}")
-        print(f"Fetch entry response: {response.json()}")
-        
-        # Test default value
-        response = requests.get(f"{BASE_URL}/entries/missing_key?default=default_val")
-        print(f"Default value response: {response.json()}")
+        response = requests.get(f"{self.base_url}/entries/test_key")
+        self.assertEqual(response.status_code, 200)
+        data = response.json()
+        self.assertIn("value", data)
 
-        return True
-    except Exception as e:
-        print(f"Error: {e}")
-        return False
-
-def test_storage_file():
-    """Test storage file endpoints"""
-    print("\nTesting storage file endpoints...")
+        # Test default value for missing key
+        response = requests.get(f"{self.base_url}/entries/missing_key?default=default_val")
+        self.assertEqual(response.status_code, 200)
+        data = response.json()
+        self.assertIn("value", data)
 
-    try:
+    def test_storage_file_operations(self):
+        """Test storage file endpoints"""
         # Save a file
-        response = requests.post(f"{BASE_URL}/storage/test/example.txt", 
+        response = requests.post(f"{self.base_url}/storage/test/example.txt", 
                                json={"content": "Hello, World!"})
-        print(f"Save file status: {response.status_code}")
+        self.assertEqual(response.status_code, 200)
 
         # Fetch the file
-        response = requests.get(f"{BASE_URL}/storage/test/example.txt")
-        print(f"Fetch file status: {response.status_code}")
-        print(f"Fetch file response: {response.json()}")
+        response = requests.get(f"{self.base_url}/storage/test/example.txt")
+        self.assertEqual(response.status_code, 200)
+        data = response.json()
+        self.assertIn("content", data)
 
         # List files
-        response = requests.get(f"{BASE_URL}/storage/test/")
-        print(f"List files status: {response.status_code}")
-        print(f"List files response: {response.json()}")
-        
-        return True
-    except Exception as e:
-        print(f"Error: {e}")
-        return False
-
-def test_storage_rename():
-    """Test storage rename endpoint"""
-    print("\nTesting storage rename endpoint...")
+        response = requests.get(f"{self.base_url}/storage/test/")
+        self.assertEqual(response.status_code, 200)
+        data = response.json()
+        self.assertIsInstance(data, (list, dict))
 
-    try:
+    def test_storage_rename_operation(self):
+        """Test storage rename endpoint"""
         # Create a test file
-        response = requests.post(f"{BASE_URL}/storage/test_rename_file.txt", 
+        response = requests.post(f"{self.base_url}/storage/test_rename_file.txt", 
                                json={"content": "Content to rename"})
-        print(f"Create test file status: {response.status_code}")
+        self.assertEqual(response.status_code, 200)
 
         # Rename the file
-        response = requests.post(f"{BASE_URL}/storage/rename", 
+        response = requests.post(f"{self.base_url}/storage/rename", 
                                json={"old_path": "test_rename_file.txt", "new_path": "renamed_test_file.txt"})
-        print(f"Rename file status: {response.status_code}")
-        print(f"Rename response: {response.json()}")
+        self.assertEqual(response.status_code, 200)
 
         # Verify old file is gone
-        response = requests.get(f"{BASE_URL}/storage/test_rename_file.txt")
-        print(f"Old file fetch status: {response.status_code} (should be 404)")
+        response = requests.get(f"{self.base_url}/storage/test_rename_file.txt")
+        self.assertEqual(response.status_code, 404)
 
         # Verify new file exists
-        response = requests.get(f"{BASE_URL}/storage/renamed_test_file.txt")
-        print(f"New file fetch status: {response.status_code}")
-        print(f"New file content: {response.json()}")
-        
-        return True
-    except Exception as e:
-        print(f"Error: {e}")
-        return False
+        response = requests.get(f"{self.base_url}/storage/renamed_test_file.txt")
+        self.assertEqual(response.status_code, 200)
+        data = response.json()
+        self.assertIn("content", data)
 
 if __name__ == "__main__":
-    print("Testing Storage API endpoints...")
-    
-    tests = [
-        test_root_endpoint,
-        test_storage_entry,
-        test_storage_file,
-        test_storage_rename
-    ]
-    
-    results = []
-    for test in tests:
-        results.append(test())
-    
-    print(f"\nTest results: {sum(results)}/{len(results)} passed")
+    unittest.main()