Move test into model

src/handlers.rs

@@ -19,40 +19,3 @@ pub async fn graphql_handler<T: Client + Send + Sync + 'static>(
 pub async fn graphiql_handler() -> impl IntoResponse {
     Html(GraphiQLSource::build().endpoint("/graphql").finish())
 }
-
-#[cfg(test)]
-mod tests {
-    use async_graphql::{EmptyMutation, EmptySubscription, Schema};
-    use httpmock::MockServer;
-    use url::Url;
-
-    use crate::TiledQuery;
-    use crate::clients::TiledClient;
-
-    #[tokio::test]
-    async fn test_api_version_query() {
-        let mock_server = MockServer::start();
-
-        let mock = mock_server
-            .mock_async(|when, then| {
-                when.method("GET").path("/api/v1/");
-                then.status(200)
-                    .body_from_file("resources/tiled_metadata.json");
-            })
-            .await;
-
-        let schema = Schema::build(
-            TiledQuery(TiledClient {
-                address: Url::parse(&mock_server.base_url()).unwrap(),
-            }),
-            EmptyMutation,
-            EmptySubscription,
-        )
-        .finish();
-
-        let response = schema.execute("{metadata { apiVersion } }").await;
-
-        assert_eq!(response.data.to_string(), "{metadata: {apiVersion: 0}}");
-        mock.assert();
-    }
-}

src/model.rs

@@ -12,3 +12,121 @@ impl TiledQuery {
         self.0.metadata().await
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use async_graphql::{EmptyMutation, EmptySubscription, Schema, Value};
+    use httpmock::MockServer;
+    use url::Url;
+
+    use crate::TiledQuery;
+    use crate::clients::TiledClient;
+
+    #[tokio::test]
+    async fn test_api_version_query() {
+        let mock_server = MockServer::start();
+        let mock = mock_server
+            .mock_async(|when, then| {
+                when.method("GET").path("/api/v1/");
+                then.status(200)
+                    .body_from_file("resources/tiled_metadata.json");
+            })
+            .await;
+
+        let schema = Schema::build(
+            TiledQuery(TiledClient {
+                address: Url::parse(&mock_server.base_url()).unwrap(),
+            }),
+            EmptyMutation,
+            EmptySubscription,
+        )
+        .finish();
+
+        let response = schema.execute("{metadata { apiVersion } }").await;
+
+        assert_eq!(response.data.to_string(), "{metadata: {apiVersion: 0}}");
+        assert_eq!(response.errors, &[]);
+        mock.assert();
+    }
+
+    #[tokio::test]
+    async fn test_server_unavailable() {
+        let schema = Schema::build(
+            TiledQuery(TiledClient {
+                address: Url::parse("http://tiled.example.com").unwrap(),
+            }),
+            EmptyMutation,
+            EmptySubscription,
+        )
+        .finish();
+
+        let response = schema.execute("{metadata { apiVersion } }").await;
+        assert_eq!(response.data, Value::Null);
+        assert_eq!(
+            response.errors[0].message,
+            "Tiled server error: error sending request for url (http://tiled.example.com/api/v1/)"
+        );
+        assert_eq!(response.errors.len(), 1);
+    }
+
+    #[tokio::test]
+    async fn test_internal_tiled_error() {
+        let mock_server = MockServer::start();
+        let mock = mock_server
+            .mock_async(|when, then| {
+                when.method("GET").path("/api/v1/");
+                then.status(503);
+            })
+            .await;
+
+        let schema = Schema::build(
+            TiledQuery(TiledClient {
+                address: Url::parse(&mock_server.base_url()).unwrap(),
+            }),
+            EmptyMutation,
+            EmptySubscription,
+        )
+        .finish();
+
+        let response = schema.execute("{metadata { apiVersion } }").await;
+        let actual = &response.errors[0].message;
+        let expected =
+            "Tiled server error: HTTP status server error (503 Service Unavailable) for url";
+        assert_eq!(response.data, Value::Null);
+        assert!(
+            actual.starts_with(expected),
+            "Unexpected error: {actual} \nExpected: {expected} [...]"
+        );
+        assert_eq!(response.errors.len(), 1);
+        mock.assert();
+    }
+
+    #[tokio::test]
+    async fn test_invalid_server_response() {
+        let mock_server = MockServer::start();
+        let mock = mock_server
+            .mock_async(|when, then| {
+                when.method("GET").path("/api/v1/");
+                then.status(200).body("{}");
+            })
+            .await;
+
+        let schema = Schema::build(
+            TiledQuery(TiledClient {
+                address: Url::parse(&mock_server.base_url()).unwrap(),
+            }),
+            EmptyMutation,
+            EmptySubscription,
+        )
+        .finish();
+
+        let response = schema.execute("{metadata { apiVersion } }").await;
+        assert_eq!(response.data, Value::Null);
+        assert_eq!(response.errors.len(), 1);
+        assert_eq!(
+            response.errors[0].message,
+            "Invalid response: missing field `api_version` at line 1 column 2, response: {}"
+        );
+        mock.assert();
+    }
+}