added check for file key

Signed-off-by: Pratiksha Sankhe <[email protected]>

tests/test_integration/test_operations.py

@@ -38,14 +38,15 @@ def test_get_files():
     with mock.patch("requests.get") as mock_get:
         mock_response = mock.Mock()
         mock_response.status_code = HTTPStatus.OK
+        mock_response.json.return_value = {"files": ["file1.txt", "file2.txt"]}
         mock_get.return_value = mock_response
 
         response = requests.get(server_url)
         print(f"Response status code: {response.status_code}")
 
-        assert (
-            response.status_code == HTTPStatus.OK
-        ), f"Expected status code 200, got {response.status_code}"
+        assert response.status_code == HTTPStatus.OK
+        assert "files" in response.json()
+        assert isinstance(response.json()["files"], list)
 
         mock_get.assert_called_once_with(server_url)