<Test>: add test video route

tests/test_friend_request.py

@@ -6,11 +6,10 @@ import mmcv
 import firebase_admin
 import requests
 from fastapi.testclient import TestClient
-from firebase_admin import credentials
 from app.main import app
+from app import db
 from app.constants import deviceId
 from fastapi.routing import APIRoute
-from firebase_admin import firestore
 from google.cloud.firestore_v1.base_query import FieldFilter
 def endpoints():
     endpoints = []
@@ -40,7 +39,7 @@ def client():
     yield client
 @pytest.fixture
 def inviter():
-    url = "https://identitytoolkit.googleapis.com/v1/accounts:signInWithPassword?key=AIzaSyArSoK9Wx9Hpe1R9ZywuLEIMVjCtHjO8Os"
+    url = "https://identitytoolkit.googleapis.com/v1/accounts:signInWithPassword?key=" + os.environ.get("FIREBASE_API_KEY")
 
     payload = json.dumps({
     "email": "test@gmail.com",
@@ -54,20 +53,10 @@ def inviter():
     data = response.json()
     inviter = {"id": data['localId'], "token": data["idToken"]}
     yield inviter
-@pytest.fixture()
-def firebase_app():
-    if "test" in firebase_admin._apps:
-        firebase_app = firebase_admin.get_app('test')
-        yield firebase_app
-    else:
-        firebase_app = firebase_admin.initialize_app(        credential= credentials.Certificate(
-            json.loads(os.environ.get("FIREBASE_CREDENTIALS"))
-        ),name='test')
-        yield firebase_app
     
 @pytest.fixture()
 def invitee():
-    url = "https://identitytoolkit.googleapis.com/v1/accounts:signInWithPassword?key=AIzaSyArSoK9Wx9Hpe1R9ZywuLEIMVjCtHjO8Os"
+    url = "https://identitytoolkit.googleapis.com/v1/accounts:signInWithPassword?key=" + os.environ.get("FIREBASE_API_KEY")
 
     payload = json.dumps({
     "email": "test2@gmail.com",
@@ -83,9 +72,8 @@ def invitee():
     yield invitee
 class TestFriendRequest():
     @pytest.mark.skipif("/friend_request" not in endpoints(),reason="Route not defined")
-    def test_post_friend(self, firebase_app, client, inviter, invitee):
+    def test_post_friend(self, client, inviter, invitee):
         # Call the firebase database
-        db = firestore.client(app=firebase_app)
         friend_request_ref = db.collection('friend_request')
         # Remove all the friend_request use for testing in the past
         query = friend_request_ref.where(filter=FieldFilter("inviter", "==", inviter['id']))

tests/test_video.py

@@ -0,0 +1,88 @@
+from fastapi.testclient import TestClient
+from fastapi.routing import APIRoute
+from app.main import app
+from app import db
+import os
+import pytest
+import requests
+import json
+from google.cloud.firestore_v1.base_query import FieldFilter
+
+def endpoints():
+    endpoints = []
+    for route in app.routes:
+        if isinstance(route, APIRoute):
+            endpoints.append(route.path)
+    return endpoints
+@pytest.fixture
+def client():
+    client = TestClient(app, "http://0.0.0.0:3000")
+    yield client
+@pytest.fixture
+def user():
+    url = "https://identitytoolkit.googleapis.com/v1/accounts:signInWithPassword?key=" + os.environ.get("FIREBASE_API_KEY")
+    payload = json.dumps({
+    "email": "test@gmail.com",
+    "password": "testing",
+    "returnSecureToken": True
+    })
+    headers = {
+    'Content-Type': 'application/json'
+    }
+    response = requests.request("POST", url, headers=headers, data=payload)
+    data = response.json()
+    user = {"id": data['localId'], "token": data["idToken"]}
+    yield user
+class TestVideoAPI:
+    @pytest.mark.skipif("/video" not in endpoints(),reason="Route not defined")
+    def test_video_API(self, user, client):
+        # Test when sent file is not a video
+        payload = {}
+        files=[
+        ('file',('demo.jpg',open('demo.jpg','rb'),'application/octet-stream'))
+        ]
+        headers = {
+        'Content-Type': 'application/json',
+        }
+        response = client.request("POST", 'video', headers=headers, data=payload, files=files)
+        assert response.status_code == 400
+        assert response.text == "File must be video"
+        # Test when no token is pass to route
+        payload = {}
+        files=[
+        ('file',('demo.mp4',open('demo.mp4','rb'),'application/octet-stream'))
+        ]
+        headers = {
+        'Content-Type': 'application/json',
+        }
+        response = client.request("POST", 'video', headers=headers, data=payload, files=files)
+        assert response.status_code == 400
+        assert response.text == "User not found"
+        # Test when all requirements have been fulfilled
+        payload = {}
+        files=[
+        ('file',('demo.mp4',open('demo.mp4','rb'),'application/octet-stream'))
+        ]
+        headers = {
+        'Content-Type': 'application/json',
+        'Authorization': "Bearer " + user['token']
+        }
+        response = client.request("POST", 'video', headers=headers, data=payload, files=files)
+        assert response.status_code == 200
+        artifactName = response.text
+        docs = db.collection("artifacts").where(filter = FieldFilter("name", '==', artifactName))
+        index = 0
+        for doc in docs:
+            # For each document in docs. Verify name and status of the artifact
+            index += 1
+            data = doc.get().to_dict()
+            assert data['name'] == artifactName
+            assert data['status'] == 'pending'
+            assert index == 1
+
+
+        
+
+
+
+