Update tests

tomchristie · tomchristie · commit d6389c7c3ac9 · 2016-09-28T10:46:28.000+01:00
diff --git a/tests/test_encode.py b/tests/test_encode.py
@@ -5,7 +5,7 @@
 
 class TestBasicInfo(TestCase):
     def setUp(self):
-        self.document = coreapi.Document(title='Example API')
+        self.document = coreapi.Document(title='Example API', url='https://www.example.com/')
         self.swagger = generate_swagger_object(self.document)
 
     def test_info(self):
@@ -23,9 +23,14 @@ def test_swagger_version(self):
 
     def test_host(self):
         self.assertIn('host', self.swagger)
-        expected = ''
+        expected = 'www.example.com'
         self.assertEquals(self.swagger['host'], expected)
 
+    def test_schemes(self):
+        self.assertIn('schemes', self.swagger)
+        expected = ['https']
+        self.assertEquals(self.swagger['schemes'], expected)
+
 
 class TestPaths(TestCase):
     def setUp(self):
