from fastapi.testclient import TestClient
import mmcv
from app.main import app
import pytest
import json
import os
import site
import shutil
from fastapi.routing import APIRoute
from app import firebase_app
import requests
def get_site_packages():
    # Get the list of directories 
    site_packages_dirs = site.getsitepackages()

    # Find the "site-packages" directory in the list
    for dir in site_packages_dirs:
        if dir.endswith("site-packages"):
            target_dir = dir
            break
        else:
            target_dir=None
    return target_dir

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 token():
    url = "https://identitytoolkit.googleapis.com/v1/accounts:signInWithPassword?key=AIzaSyArSoK9Wx9Hpe1R9ZywuLEIMVjCtHjO8Os"

    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)
    token = response.json()["idToken"]
    yield token
class TestFireBaseAPI():
    def test_get_me(self, client, token):
        if "/me" not in endpoints():
            pytest.skip("This route isn't defined")
        else:
            if token != "":
                payload = ""
                headers = {
                    'accept': 'application/json',
                    "Authorization":"Bearer " + token
                }
                response = client.request("get", 'me', headers=headers, data=payload)
                assert response.status_code == 200
            payload = ""
            headers = {
                'accept': 'application/json',
                'Content-Type': 'application/json',
            }
            response = client.request("get", 'me', headers=headers, data=payload)
            assert response.status_code == 403
    def test_invitation(self,client, token):
        if"/invitation" not in endpoints():
            pytest.skip("This route isn't defined")
        else:
            payload = ''
            headers = {
                'accept': 'application/json',
                'Content-Type': 'application/json',
                "Authorization": "Bearer " + token
            }
            response = client.request("post", 'invitation', headers=headers, data=payload)
            assert response.status_code == 200
            result = mmcv.imfrombytes(response.read())
            assert result.shape[0] == 3