import unittest

from fastapi.testclient import TestClient

from server.app import app
import server.app as server_app


class ServerApiTests(unittest.TestCase):
    def setUp(self):
        server_app.runtime_session = None
        self.client = TestClient(app)

    def test_health(self):
        response = self.client.get("/health")
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.json().get("status"), "ok")

    def test_reset_state_step_flow(self):
        reset_response = self.client.post("/reset", json={})
        self.assertEqual(reset_response.status_code, 200)
        reset_payload = reset_response.json()
        self.assertIn("observation", reset_payload)
        self.assertIn("step_count", reset_payload)
        self.assertEqual(reset_payload["step_count"], 0)

        state_response = self.client.get("/state")
        self.assertEqual(state_response.status_code, 200)
        state_payload = state_response.json()
        self.assertTrue(state_payload.get("initialized"))

        step_response = self.client.post(
            "/step",
            json={"action": "edit_config: replace npm tset with npm test"},
        )
        self.assertEqual(step_response.status_code, 200)
        step_payload = step_response.json()
        self.assertIn("reward", step_payload)
        self.assertIn("done", step_payload)

    def test_step_requires_reset(self):
        server_app.runtime_session = None
        client = TestClient(app)
        response = client.post("/step", json={"action": "read_logs: inspect logs"})
        self.assertEqual(response.status_code, 400)


if __name__ == "__main__":
    unittest.main()