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)