Include test for --include-only

tests/test_cli.py

@@ -63,3 +63,19 @@ def test_errors_if_neither_demo_or_connection_string_included(self):
             "Exactly one of --demo and --connection-string should be used.",
             completed_process.stderr
         )
+
+    def test_can_pass_in_list_of_tables_to_include(self):
+        command = (
+            "fk-graph"
+            " --demo"
+            " --table=table_a"
+            " --primary-key=1"
+            " --include-only=\"['table_a']\""
+        )
+        # PIPE just stops stdout from printing to screen.
+        process = Popen(shlex_split(command), stdout=PIPE)
+        self.addCleanup(process.terminate)
+        response = self.session.get("http://localhost:8050")
+        # Because of the javascripty-nature of the app, we can
+        # only really inspect the status code.
+        self.assertEqual(response.status_code, 200)