From ce9d58ad5c60b4747ac319cf2bb988d475d36a30 Mon Sep 17 00:00:00 2001
From: Ramon Wenger <ramon.wenger@iterativ.ch>
Date: Thu, 19 May 2022 18:33:45 +0200
Subject: [PATCH] Add unit test

---
 server/books/tests/test_404.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 server/books/tests/test_404.py

diff --git a/server/books/tests/test_404.py b/server/books/tests/test_404.py
new file mode 100644
index 00000000..116bd2f8
--- /dev/null
+++ b/server/books/tests/test_404.py
@@ -0,0 +1,33 @@
+from graphql.error import GraphQLLocatedError
+
+from core.tests.base_test import SkillboxTestCase
+
+TOPIC_QUERY = """
+query TopicQuery($slug: String!) {
+  topic(slug: $slug) {
+    __typename
+    ...on TopicNode {
+      title
+    }
+    ...on NotFound {
+      reason
+    }
+  }
+}
+"""
+
+
+class ContentBlockTestCase(SkillboxTestCase):
+    def setUp(self) -> None:
+        self.createDefault()
+        self.client = self.get_client()
+
+    def test_topic(self):
+        slug = "non-existing"
+        result = self.client.execute(TOPIC_QUERY, variables={
+            "slug": slug
+        })
+        self.assertIsNone(result.get('errors'))
+        topic = result.get('data').get('topic')
+        self.assertEqual(topic.get('__typename'), 'NotFound')
+        self.assertEqual(topic.get('reason'), 'Not Found')