def test_get_policies_error(config):
compliance_client = ComplianceClient(config)
compliance_client.hostname = 'foo'
compliance_client.conn.session.get = Mock(return_value=Mock(
status_code=500))
assert compliance_client.get_policies() == []
compliance_client.conn.session.get.assert_called_with(
'https://localhost/app/compliance/systems',
params={'search': 'name=foo'})
def test_get_policies_no_policies(config):
compliance_client = ComplianceClient(config)
compliance_client.hostname = 'foo'
compliance_client.conn.session.get = Mock(return_value=Mock(
status_code=200, json=Mock(return_value={'data': []})))
assert compliance_client.get_policies() == []
compliance_client.conn.session.get.assert_called_with(
'https://localhost/app/compliance/profiles',
params={'search': 'system_names=foo'})
def test_get_initial_profiles(config):
compliance_client = ComplianceClient(config)
compliance_client.hostname = 'foo'
compliance_client.conn.session.get = Mock(return_value=Mock(status_code=200, json=Mock(return_value={'data': [{'attributes': 'data'}]})))
assert compliance_client.get_initial_profiles() == [{'attributes': 'data'}]
compliance_client.conn.session.get.assert_called_with('https://localhost/app/compliance/profiles', params={'search': 'system_names=foo canonical=false external=false'})
def test_get_profiles_matching_os(config, linux_distro_mock):
compliance_client = ComplianceClient(config)
compliance_client.hostname = 'foo'
compliance_client.conn.session.get = Mock(return_value=Mock(status_code=200, json=Mock(return_value={'data': [{'attributes': 'data'}]})))
assert compliance_client.get_profiles_matching_os() == [{'attributes': 'data'}]
compliance_client.conn.session.get.assert_called_with('https://localhost/app/compliance/profiles', params={'search': 'system_names=foo canonical=false os_minor_version=5'})
def test_get_profiles(config):
compliance_client = ComplianceClient(config)
compliance_client.hostname = 'foo'
compliance_client.conn.session.get = Mock(return_value=Mock(status_code=200, json=Mock(return_value={'data': [{'attributes': 'data'}]})))
assert compliance_client.get_profiles('search string') == [{'attributes': 'data'}]
compliance_client.conn.session.get.assert_called_with('https://localhost/app/compliance/profiles', params={'search': 'search string'})
