client = TestClient(app)


def test_get():
    response = client.get("/unicorns/shinny")
    assert response.status_code == 200, response.text
    assert response.json() == {"unicorn_name": "shinny"}


def test_get_exception():
    response = client.get("/unicorns/yolo")
    assert response.status_code == 418, response.text
    assert response.json() == {
        "message": "Oops! yolo did something. There goes a rainbow..."
    }

        // Then
        verify(response).setExpiration(expiration);
    }

    @Test
    @DisplayName("Test Response interface methods - getException")
    void testGetException() {
        // Given
        Exception expectedException = new Exception("Test exception");
        when(response.getException()).thenReturn(expectedException);

        // When