# 需要导入模块: import settings [as 别名]

# 或者: from settings import VERSION [as 别名]

def test_version(self):

host_info = self.mock_servers['mynewhost']

with timed('csr', 10):

data = {

'fqdn': host_info['fqdn'],

'nodename': host_info['nodename'],

'version': '1.0',

'capabilities': ['manage_targets'],

'address': 'mynewhost',

'csr': generate_csr(host_info['fqdn']),

}

with patch(settings, VERSION='2.0'):

# Try with a mis-matched version

token = RegistrationToken.objects.create(profile=ServerProfile.objects.get())

with timed('register fail', 10):

response = Client().post("/agent/register/%s/" % token.secret, data=json.dumps(data), content_type="application/json")

self.assertEqual(response.status_code, 400)

# Try with a matching version

token = RegistrationToken.objects.create(profile=ServerProfile.objects.get())

settings.VERSION = '1.1'

with timed('register pass', 10):

response = Client().post("/agent/register/%s/" % token.secret, data=json.dumps(data), content_type="application/json")

self.assertEqual(response.status_code, 201)

content = json.loads(response.content)

# reregistration should fail with unknown serial

data = {'address': 'mynewhost', 'fqdn': 'mynewhost.newcompany.com'}

headers = {'HTTP_X_SSL_CLIENT_NAME': host_info['fqdn'], 'HTTP_X_SSL_CLIENT_SERIAL': ''}

response = Client().post('/agent/reregister/', data=json.dumps(data), content_type='application/json', **headers)

self.assertEqual(response.status_code, 403)

# reregistration should update host's domain name

headers['HTTP_X_SSL_CLIENT_SERIAL'] = Crypto().get_serial(content['certificate'])

response = Client().post('/agent/reregister/', data=json.dumps(data), content_type='application/json', **headers)

self.assertEqual(response.status_code, 200)

host = ManagedHost.objects.get(id=content['host_id'])

self.assertEqual(host.fqdn, data['fqdn'])

# TOOD: reinstate selinux check, probably within the agent itself (it should fail

# its own registration step without even talking to the manager)

# def test_selinux_detection(self):

# """Test that a host with SELinux enabled fails setup."""

# MockAgentRpc.selinux_enabled = True

# try:

# import time

# host = self._create_host('myaddress')

# self.assertTrue(Command.objects.all().order_by("-id")[0].errored)

# self.assertState(host, 'unconfigured')

# finally:

# MockAgentRpc.selinux_enabled = False