Add a test to force us to update default schema!

plotly · theengineear · Sep 29, 2015 · Sep 29, 2015 · Sep 29, 2015 · Sep 29, 2015
commit 174647780a4ace954bb1e425777e66a3c5bbdfab
diff --git a/plotly/tests/test_core/test_graph_reference/test_graph_reference.py b/plotly/tests/test_core/test_graph_reference/test_graph_reference.py
@@ -9,6 +9,7 @@
 from pkg_resources import resource_string
 from unittest import TestCase

+import requests
 from nose.plugins.attrib import attr

 from plotly import files, graph_reference as gr, tools, utils
@@ -57,6 +58,25 @@ def test_get_graph_reference_bad_request_no_copy(self):
        graph_reference = gr.get_graph_reference()
        self.assertEqual(graph_reference, default_graph_reference)

+    @attr('slow')
+    def test_default_schema_is_up_to_date(self):
+        api_domain = files.FILE_CONTENT[files.CONFIG_FILE]['plotly_api_domain']
+        graph_reference_url = '{}{}?sha1'.format(api_domain,
+                                                 gr.GRAPH_REFERENCE_PATH)
+        response = requests.get(graph_reference_url)
+        schema = json.loads(response.content)['schema']
+
+        path = os.path.join('graph_reference', 'default-schema.json')
+        s = resource_string('plotly', path).decode('utf-8')
+        default_schema = json.loads(s)
+
+        msg = (
+            'The default, hard-coded plot schema we ship with pip is out of '
+            'sync with the prod plot schema!\n'
+            'Run `make update_default_schema` to fix it!'
+        )
+        self.assertEqual(schema, default_schema, msg=msg)
+

 class TestStringToClass(PlotlyTestCase):