tests/test_friend_request.py

import os
import pytest
import json 
import cv2
import mmcv
import firebase_admin
import requests
from fastapi.testclient import TestClient
from firebase_admin import credentials
from app.main import app
from app.constants import deviceId
from fastapi.routing import APIRoute
from firebase_admin import firestore
from app import db
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
def read_qr_code(filename):
 """Read an image and read the QR code.
 
 Args:
 filename (string): Path to file
 
 Returns:
 qr (string): Value from QR code
 """
 try:
 img = cv2.imread(filename)
 detect = cv2.QRCodeDetector()
 value, points, straight_qrcode = detect.detectAndDecode(img)
 return value
 except:
 return
@pytest.fixture
def client():
 client = TestClient(app)
 yield client
@pytest.fixture
def inviter():
 url = "https://identitytoolkit.googleapis.com/v1/accounts:signInWithPassword?key=AIzaSyArSoK9Wx9Hpe1R9ZywuLEIMVjCtHjO8Os"

 payload = json.dumps({
 "email": "[email protected]",
 "password": "testing",
 "returnSecureToken": True
 })
 headers = {
 'Content-Type': 'application/json'
 }
 response = requests.request("POST", url, headers=headers, data=payload)
 data = response.json()
 inviter = {"id": data['localId'], "token": data["idToken"]}
 yield inviter
 
@pytest.fixture()
def invitee():
 url = "https://identitytoolkit.googleapis.com/v1/accounts:signInWithPassword?key=AIzaSyArSoK9Wx9Hpe1R9ZywuLEIMVjCtHjO8Os"

 payload = json.dumps({
 "email": "[email protected]",
 "password": "testing2",
 "returnSecureToken": True
 })
 headers = {
 'Content-Type': 'application/json'
 }
 response = requests.request("POST", url, headers=headers, data=payload)
 data = response.json()
 invitee = {"id": data['localId'], "token": data["idToken"]}
 yield invitee
class TestFriendRequest():
 @pytest.mark.skipif("/friend_request" not in endpoints(),reason="Route not defined")
 def test_post_friend(self, client, inviter, invitee):
 # Call the firebase database
 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']))
 docs = query.stream()
 for doc in docs:
 doc.reference.delete()
 # Delete the user for safety-check 
 user_ref = db.collection("user")
 user_ref.document(inviter['id']).delete()
 # Send request with no token
 payload = ''
 headers = {
 'Content-Type': 'application/json',
 }
 response = client.request("POST", 'friend_request', headers=headers, data=payload)
 assert response.status_code == 403
 # Send request with false token
 payload = ''
 headers = {
 'Content-Type': 'application/json',
 'Authorization': 'Bearer amksckmasckmafvqnwfniqoniofv' 
 }
 response = client.request("POST", 'friend_request', headers=headers, data=payload)
 assert response.status_code == 401
 # Send request with unknown user
 payload = ''
 headers = {
 'Content-Type': 'application/json',
 'Authorization': 'Bearer ' + inviter['token']
 }
 response = client.request("POST", 'friend_request', headers=headers, data=payload)
 assert response.status_code == 500
 # Create request and re-send 
 user_ref.document(inviter['id']).set({"deviceId": deviceId})
 payload = ''
 headers = {
 'Content-Type': 'application/json',
 'Authorization': 'Bearer ' + inviter['token']
 }
 response = client.request("POST", 'friend_request', headers=headers, data=payload)
 # Check response status code
 assert response.status_code == 200
 result = mmcv.imfrombytes(response.read())
 # Check returned QR image
 assert result.shape[2] == 3
 # Write image for later read
 mmcv.imwrite(result, "qrcode.jpg")
 # Now test for the invitee aka the one that scan QR code
 # Delete invitee user (if existed)
 user_ref.document(invitee['id']).delete()
 # Test when the invitee is unknow user (no user entity in database)
 request_id = read_qr_code("qrcode.jpg")
 payload = ''
 headers = {
 'Content-Type': 'application/json',
 'Authorization': 'Bearer ' + invitee['token']
 }
 response = client.request("PATCH", 'friend_request/' + request_id, headers=headers, data=payload)
 assert response.status_code == 500

 # Create invitee user
 user_ref.document(invitee['id']).set({"deviceId": deviceId})
 # Send request
 request_id = read_qr_code("qrcode.jpg")
 payload = ''
 headers = {
 'Content-Type': 'application/json',
 'Authorization': 'Bearer ' + invitee['token']
 }
 response = client.request("PATCH", 'friend_request/' + request_id, headers=headers, data=payload)
 assert response.status_code == 200
 # Delete entity for next time test
 user_ref.document(inviter['id']).delete()
 user_ref.document(invitee['id']).delete()