From 00ac92b9f930fc1a9cdeb8be0e6971ce46948f6d Mon Sep 17 00:00:00 2001
From: Eric Asante <Eric.Asante@uconn.edu>
Date: Thu, 31 Oct 2024 17:48:01 -0400
Subject: [PATCH] API complete

---
 Backend/main.py      |   5 --
 Backend/test_main.py | 132 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 132 insertions(+), 5 deletions(-)
 create mode 100644 Backend/test_main.py

diff --git a/Backend/main.py b/Backend/main.py
index bcb9f9f..bb429fa 100644
--- a/Backend/main.py
+++ b/Backend/main.py
@@ -7,15 +7,11 @@
 
 app = Flask(__name__)
 
-
-UserAccInfo = {"Name":"eric","Location":"connecticut","Occupation":"Student"} # example user information
-
 #Get Random Pet information - For homepage
 @app.route("/Get_Random_pet", methods=["GET"])
 def get_random_pet():
     return Get_Random_Pet(), 200 # Return dog information
 
-
 # Faviortes route. add user faviorte pet to database
 @app.route("/Add_User_Faviorte", methods=["POST"])
 def add_user_faviorte():
@@ -25,7 +21,6 @@ def add_user_faviorte():
     #Retrun database with succses code
     return jsonify(Add_User_Faviorte(json_data)), 200
 
-
 # Un Faviote Route. remove user faviorte pet from there database 
 @app.route("/Remove_User_Faviorte", methods=["DELETE"])
 def remove_user_faviorte():
diff --git a/Backend/test_main.py b/Backend/test_main.py
new file mode 100644
index 0000000..57becdf
--- /dev/null
+++ b/Backend/test_main.py
@@ -0,0 +1,132 @@
+import requests
+from flask import Flask, json, request, jsonify
+
+def test_Get_Random_pet():
+    """
+    Test for http://localhost:5000/Get_Random_pet
+    """
+
+    url = "http://localhost:5000"
+
+    response = requests.get(url + "/Get_Random_pet")
+
+    expected_response = {
+        "id": "9",
+        "name": "sam",
+        "sex": "male",
+        "Age": "8",
+        "location": "Connecticwut",
+        "Breed": "Pug"
+    }
+
+    assert response.json() == expected_response
+
+# -----------------------------------------------------------------------------
+
+def test_Add_User_Faviorte():
+    """
+    Test for http://localhost:5000/Add_User_Faviorte
+    """
+
+    url = "http://localhost:5000"
+
+    FaviortePet1 = {
+        "id": "9",
+        "name": "sam",
+        "sex": "male",
+        "Age": "8",
+        "location": "Connecticut",
+        "Breed": "Pug"
+    }
+
+
+    response = requests.post((url + "/Add_User_Faviorte"), json=(FaviortePet1))
+
+    assert response.json() == [{'Age': '8', 'Breed': 'Pug', 'id': '9', 'location': 'Connecticut', 'name': 'sam', 'sex': 'male'}]
+
+    FaviortePet2 = {
+        "id": "90",
+        "name": "ben",
+        "sex": "male",
+        "Age": "2",
+        "location": "New York",
+        "Breed": "Husky"
+    }
+
+    response = requests.post((url + "/Add_User_Faviorte"), json=(FaviortePet2))
+
+    assert response.json() == [{'Age': '8', 'Breed': 'Pug', 'id': '9', 'location': 'Connecticut', 'name': 'sam', 'sex': 'male'}, {'Age': '2', 'Breed': 'Husky', 'id': '90'
+, 'location': 'New York', 'name': 'ben', 'sex': 'male'}]
+
+# ----------------------------------------------------------------------------------
+
+def test_Remove_User_Faviorte():
+    """
+    Test for http://localhost:5000/Remove_User_Faviorte
+    """
+
+    url = "http://localhost:5000"
+
+    FaviortePet1 = {
+        "id": "9",
+        "name": "sam",
+        "sex": "male",
+        "Age": "8",
+        "location": "Connecticut",
+        "Breed": "Pug"
+    }
+
+
+    response = requests.delete((url + "/Remove_User_Faviorte"), json=(FaviortePet1))
+
+    assert response.json() == [{'Age': '2', 'Breed': 'Husky', 'id': '90', 'location': 'New York', 'name': 'ben', 'sex': 'male'}]
+
+    FaviortePet2 = {
+        "id": "90",
+        "name": "ben",
+        "sex": "male",
+        "Age": "2",
+        "location": "New York",
+        "Breed": "Husky"
+    }
+
+    response = requests.delete((url + "/Remove_User_Faviorte"), json=(FaviortePet2))
+
+    assert response.json() == []
+# ----------------------------------------------------------------------------------
+
+
+
+def test_Change_User_Location():
+    """
+    Test for http://localhost:5000/Change_User_Location
+    """
+
+    url = "http://localhost:5000"
+
+    response = requests.put((url + "/Change_User_Location"), json={"Location":"New York"})
+
+    assert response.json() == {"Location": "New York","Name": "eric","Occupation": "Student","Password": "2020"}
+
+
+    response = requests.put((url + "/Change_User_Location"), json={"Location":"Chicago"})
+
+    assert response.json() == {"Location": "Chicago","Name": "eric","Occupation": "Student","Password": "2020"}
+
+    response = requests.put((url + "/Change_User_Location"), json={"Location":"Massachusetts"})
+
+    assert response.json() == {"Location": "Massachusetts","Name": "eric","Occupation": "Student","Password": "2020"}
+
+
+
+
+
+
+
+
+
+test_Get_Random_pet()
+test_Add_User_Faviorte()
+test_Remove_User_Faviorte()
+test_Change_User_Location()
+print("Test check Complete: Passed")