def test_default_version(self):
request = self.factory.get('/')
scheme = versioning.HostNameVersioning()
scheme.default_version = 'v2'
scheme_version = scheme.get_version(request)
self.assertEqual(scheme_version, scheme.default_version)
def test_get_version(self):
headers = {
'HTTP_HOST': '{}.example.com'.format(self.version),
}
request = self.factory.get('/', **headers)
scheme = versioning.HostNameVersioning()
scheme_version = scheme.get_version(request)
self.assertEqual(scheme_version, self.version)
def test_version_not_allowed(self):
headers = {
'HTTP_HOST': '{}.example.com'.format(self.version),
}
request = self.factory.get('/', **headers)
scheme = versioning.HostNameVersioning()
scheme.allowed_versions = ('v2',)
with self.assertRaises(GraphQLPersistError):
scheme.get_version(request)