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