def test_create_action_body_by_host(requests_mock): client = Client(BASE_URL, 'username', 'password', 'domain') requests_mock.get(BASE_URL + 'session/login', json={'data': { 'session': 'session-id' }}) requests_mock.get(BASE_URL + 'packages/20', json={'data': { 'id': 12345, 'expire_seconds': 360 }}) requests_mock.post(BASE_URL + 'parse_question', json=parse_question_res) body = client.build_create_action_body(True, 'action-name', '', package_id=20, action_group_id=1, hostname='host') body = json.dumps(body) res = json.dumps(CREATE_ACTION_BY_HOST_RES) assert res == body
def test_create_action_body_with_parameters(requests_mock): client = Client(BASE_URL, 'username', 'password', 'domain') requests_mock.get(BASE_URL + 'session/login', json={'data': { 'session': 'session-id' }}) requests_mock.get(BASE_URL + 'packages/by-name/package-name', json={'data': { 'id': 12345, 'expire_seconds': 360 }}) body = client.build_create_action_body( False, 'action-name', '$1=true;$2=value;$3=otherValue', package_name='package-name', action_group_id=1, target_group_name='target-group-name') body = json.dumps(body) res = json.dumps(CREATE_ACTION_WITH_PARAMETERS_RES) assert res == body
def test_create_action_body_by_target_group_name(requests_mock): client = Client(BASE_URL, 'username', 'password', 'domain') requests_mock.get(BASE_URL + 'session/login', json={'data': { 'session': 'SESSION-ID' }}) requests_mock.get(BASE_URL + 'packages/by-name/package-name', json={'data': { 'id': 12345, 'expire_seconds': 360 }}) body = client.build_create_action_body( False, 'action-name', '', package_name='package-name', action_group_id=1, target_group_name='target-group-name') body = json.dumps(body) res = json.dumps(CREATE_ACTION_BY_TARGET_GROUP_RES) assert res == body