[checkmk-commits] Add add_folder(), delete_folder(), folder_exists(), delete_folder() and host_exists() methods to web fixture

Module: check_mk Branch: master Commit: a2f9d6480745771f32976fa0bef2c05e6f30f161 URL: http://git.mathias-kettner.de/git/?p=check_mk.git;a=commit;h=a2f9d648074577… Author: Lars Michelsen <lm(a)mathias-kettner.de> Date: Tue Feb 20 11:27:24 2018 +0100 Add add_folder(), delete_folder(), folder_exists(), delete_folder() and host_exists() methods to web fixture Change-Id: I0ea581b7d061d938bb39df55f1d5df56100a7868 --- tests/testlib/__init__.py | 77 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 77 insertions(+) diff --git a/tests/testlib/__init__.py b/tests/testlib/__init__.py index a6c99ea..556749c 100644 --- a/tests/testlib/__init__.py +++ b/tests/testlib/__init__.py @@ -1148,6 +1148,83 @@ class CMKWebSession(WebSession): return result + def host_exists(self, hostname): + try: + result = self._api_request("webapi.py?action=get_host", { + "request": json.dumps({ + "hostname": hostname, + }), + }) + except AssertionError, e: + if "No such host" in "%s" % e: + return False + else: + raise + + assert type(result) == dict + return "hostname" in result + + + def add_folder(self, folder_path, attributes=None, create_folders=True, expect_error=False): + if attributes == None: + attributes = {} + + result = self._api_request("webapi.py?action=add_folder", { + "request": json.dumps({ + "folder" : folder_path, + "attributes" : attributes or {}, + "create_parent_folders" : create_folders, + }), + }, expect_error=expect_error) + + assert result == None + + folder = self.get_folder(folder_path) + assert folder["attributes"] == attributes + + + def get_folder(self, folder_path, effective_attributes=False): + result = self._api_request("webapi.py?action=get_folder", { + "request": json.dumps({ + "folder" : folder_path, + "effective_attributes" : effective_attributes, + }), + }) + + assert type(result) == dict + assert "attributes" in result + + return result + + + def folder_exists(self, folder_path): + try: + result = self._api_request("webapi.py?action=get_folder", { + "request": json.dumps({ + "folder": folder_path, + }), + }) + except AssertionError, e: + if "does not exist" in "%s" % e: + return False + else: + raise + + assert type(result) == dict + return "folder" in result + + + def delete_folder(self, folder_path): + result = self._api_request("webapi.py?action=delete_folder", { + "request": json.dumps({ + "folder": folder_path, + }), + }) + + assert result == None + assert not self.folder_exists(folder_path) + + def get_ruleset(self, ruleset_name): result = self._api_request("webapi.py?action=get_ruleset&output_format=python", { "request": json.dumps({