Fix CI

.github/workflows/test.yml

@@ -40,4 +40,37 @@ jobs:
           AURA_INSTANCEID: ${{secrets.AURA_INSTANCEID}}
           AURA_INSTANCENAME: ${{secrets.AURA_INSTANCENAME}}
           FIREBASE_API_KEY: ${{secrets.FIREBASE_API_KEY}}
+  build-linux:
+    runs-on: linux-latest
+    strategy:
+      matrix:
+        python-version: [ "3.8", "3.10", "3.9"]
+    steps:
+      - uses: actions/checkout@v4
+      - name: Set up Python 3.8 and install dependencies
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python-version }}
+      - run: pip install -r app/requirements.txt
+      - name: install apt dependencies
+        uses: awalsh128/cache-apt-pkgs-action@latest
+        with:
+          packages: wget libgl1
+      - name: download model
+        run: |
+          wget -O ./model/end2end.onnx $MODEL_URL
+        env:
+          MODEL_URL: ${{secrets.MODEL_URL}}
+      - name: Run test
+        run: pytest
+        env:
+          SUPABASE_URL: ${{secrets.SUPABASE_URL}}
+          SUPABASE_KEY: ${{secrets.SUPABASE_KEY}}
+          FIREBASE_CREDENTIALS: ${{secrets.FIREBASE_CREDENTIALS}}
+          NEO4J_URI: ${{secrets.NEO4J_URI}}
+          NEO4J_USERNAME: ${{secrets.NEO4J_USERNAME}}
+          NEO4J_PASSWORD: ${{secrets.NEO4J_PASSWORD}}
+          AURA_INSTANCEID: ${{secrets.AURA_INSTANCEID}}
+          AURA_INSTANCENAME: ${{secrets.AURA_INSTANCENAME}}
+          FIREBASE_API_KEY: ${{secrets.FIREBASE_API_KEY}}
 

tests/test_friend_request.py

@@ -129,6 +129,8 @@ class TestFriendRequest:
             if response.status_code == 200:
                 flag = True
                 break
+            else:
+                time.sleep(2)
         assert flag == True
         # Create request and re-send
         user_ref.document(inviter["id"]).set({"deviceId": deviceId})

tests/test_video.py

@@ -11,6 +11,7 @@ import requests
 import json
 import cv2
 import shutil
+import time
 from google.cloud.firestore_v1.base_query import FieldFilter
 
 
@@ -120,10 +121,14 @@ class TestVideoAPI:
             test_artifact.update({"status": "testing", "path": "", "thumbnailURL": ""})
         # Testing update on each field
         updateArtifact(test_artifact.id, {"status": "test_done"})
-        assert (
-            db.collection("artifacts").document("test").get().to_dict()["status"]
-            == "test_done"
-        )
+        flag = False
+        for i in range(5):
+            if db.collection("artifacts").document("test").get().to_dict()["status"] == "test_done":
+                flag = True
+                break
+            else:
+                time.sleep(2)
+        assert flag == True
         # Delete data for next time test
         test_artifact.delete()