import json
from typing import Dict, Optional, Tuple, List, Any
from unittest import TestCase

import optuna
from .wsgi_utils import create_wsgi_env
from optuna_dashboard.app import create_app


def create_json_api_wsgi_env(
    path: str,
    method: str,
    json_body: Optional[Dict[str, Any]] = None,
    content_type: str = "application/json",
    headers: Optional[Dict[str, str]] = None,
):
    body = json.dumps(json_body) if json_body else ""
    return create_wsgi_env(
        path, method, body, content_type=content_type, headers=headers
    )


class APITestCase(TestCase):
    def setUp(self):
        self.status: Optional[str] = None
        self.headers: Optional[List[Tuple[str, str]]] = None

    def _start_response(self, status, headers):
        self.status = status
        self.headers = headers

    def test_create_study(self):
        storage = optuna.storages.InMemoryStorage()
        self.assertEqual(len(storage.get_all_study_summaries()), 0)

        app = create_app(storage)
        env = create_json_api_wsgi_env(
            "/api/studies",
            "POST",
            json_body={
                "study_name": "foo",
                "direction": "minimize",
            },
        )
        _ = app(env, self._start_response)
        self.assertEqual(self.status, "201 Created")
        self.assertEqual(len(storage.get_all_study_summaries()), 1)

    def test_create_study_duplicated(self):
        storage = optuna.storages.InMemoryStorage()
        storage.create_new_study("foo")
        self.assertEqual(len(storage.get_all_study_summaries()), 1)

        app = create_app(storage)
        env = create_json_api_wsgi_env(
            "/api/studies",
            "POST",
            json_body={
                "study_name": "foo",
                "direction": "minimize",
            },
        )
        _ = app(env, self._start_response)
        self.assertEqual(self.status, "400 Bad Request")
        self.assertEqual(len(storage.get_all_study_summaries()), 1)

    def test_delete_study(self):
        storage = optuna.storages.InMemoryStorage()
        storage.create_new_study("foo1")
        storage.create_new_study("foo2")
        self.assertEqual(len(storage.get_all_study_summaries()), 2)

        app = create_app(storage)
        env = create_json_api_wsgi_env(
            "/api/studies/1",
            "DELETE",
        )
        _ = app(env, self._start_response)
        self.assertEqual(self.status, "204 No Content")
        self.assertEqual(len(storage.get_all_study_summaries()), 1)

    def test_delete_study_not_found(self):
        storage = optuna.storages.InMemoryStorage()
        app = create_app(storage)
        env = create_json_api_wsgi_env(
            "/api/studies/1",
            "DELETE",
        )
        _ = app(env, self._start_response)
        self.assertEqual(self.status, "404 Not Found")