class MessagesModule(TaskSet): def __init__(self, parent): super().__init__(parent) self.requests_helper = RequestsHelper(self.client) @task(1) def navigate_to_messages(self): url = target["admin"]["messages"]["endpoint"] self.requests_helper.send_get_request(url, "/MESSAGES") @task(1) def read_message(self): url = target["admin"]["read_message"]["endpoint"] + "/" + target["admin"]["read_message"]["id"] self.requests_helper.send_get_request(url, "/MESSAGE READ") @task(1) def send_message(self): url = target["admin"]["send_message"]["endpoint"] filename = target["admin"]["send_message"]["filename"] self.requests_helper.send_post_request(url, "NEW MESSAGE", filename) @task(1) def reply_to_message(self): url = target["admin"]["reply_to_message"]["endpoint"] + "/" + target["admin"]["reply_to_message"]["id"] filename = target["admin"]["reply_to_message"]["filename"] self.requests_helper.send_post_request(url, "/REPLY TO MESSAGE", filename) @task(2) def stop(self): self.interrupt()
class UserProfileModule(TaskSet): def __init__(self, parent): super().__init__(parent) self.requests_helper = RequestsHelper(self.client) @task(1) def approve_registration_request(self): url = target["admin"]["approve_reg_request"]["endpoint"] filename = target["admin"]["approve_reg_request"]["filename"] self.requests_helper.send_post_request(url, "/APPROVE REGISTRATION REQUEST", filename) @task(1) def create_user_profile(self): url = target["admin"]["create_user_profile"]["endpoint"] filename = target["admin"]["create_user_profile"]["filename"] self.requests_helper.send_post_request(url, "/CREATE USER PROFILE", filename) @task(1) def view_account_types(self): url = target["admin"]["account_types"]["endpoint"] self.requests_helper.send_get_request(url, "/ACCOUNT TYPES") @task(1) def create_new_account(self): url = target["admin"]["new_account"]["endpoint"] filename = target["admin"]["create_user_profile"]["filename"] self.requests_helper.send_post_request(url, "/NEW ACCOUNT", filename) @task(1) def unblock_user_profile(self): url = target["admin"]["unblock_user_profile"]["endpoint"] filename = target["admin"]["reply_to_message"]["filename"] self.requests_helper.send_post_request(url, "/USER UNBLOCK", filename) @task(2) def stop(self): self.interrupt()
class RequestsModule(TaskSet): def __init__(self, parent): super().__init__(parent) self.requests_helper = RequestsHelper(self.client) @task(1) def create_tba_request(self): url = target["user"]["tba_request"]["endpoint"] filename_1 = target["user"]["tba_request"]["fill_in_filename"] filename_2 = target["user"]["tba_request"]["finish_filename"] self.requests_helper.send_post_request(url, "/TBA FILL IN DATA", filename_1) self.requests_helper.send_post_request(url, "/TBA ACCEPT DATA", filename_2) @task(1) def create_tbu_request(self): url = target["user"]["tbu_request"]["endpoint"] filename_1 = target["user"]["tbu_request"]["fill_in_filename"] filename_2 = target["user"]["tbu_request"]["finish_filename"] self.requests_helper.send_post_request(url, "/TBU FILL IN DATA", filename_1) self.requests_helper.send_post_request(url, "/TBU ACCEPT DATA", filename_2) @task(1) def create_owt_request(self): url = target["user"]["owt_request"]["endpoint"] filename_1 = target["user"]["owt_request"]["fill_in_filename"] filename_2 = target["user"]["owt_request"]["finish_filename"] self.requests_helper.send_post_request(url, "/OWT FILL IN DATA", filename_1) self.requests_helper.send_post_request(url, "/OWT ACCEPT DATA", filename_2) @task(2) def stop(self): self.interrupt()
class TransferRequestsModule(TaskSet): def __init__(self, parent): super().__init__(parent) self.requests_helper = RequestsHelper(self.client) @task(1) def create_tba_request(self): url = target["admin"]["tba_request"]["endpoint"] filename_1 = target["admin"]["tba_request"]["fill_data_filename"] filename_2 = target["admin"]["tba_request"]["finish_filename"] self.requests_helper.send_post_request(url, "/TBA FILL IN DATA", filename_1) self.requests_helper.send_post_request(url, "/TBA ACCEPT DATA", filename_2) @task(1) def create_tbu_request(self): url = target["admin"]["tbu_request"]["endpoint"] filename_1 = target["admin"]["tbu_request"]["fill_data_filename"] filename_2 = target["admin"]["tbu_request"]["finish_filename"] self.requests_helper.send_post_request(url, "/TBU FILL IN DATA", filename_1) self.requests_helper.send_post_request(url, "/TBU ACCEPT DATA", filename_2) @task(1) def create_owt_request(self): url = target["admin"]["owt_request"]["endpoint"] filename_1 = target["admin"]["owt_request"]["fill_data_filename"] filename_2 = target["admin"]["owt_request"]["finish_filename"] self.requests_helper.send_post_request(url, "/OWT FILL IN DATA", filename_1) self.requests_helper.send_post_request(url, "/OWT ACCEPT DATA", filename_2) @task(1) def perform_debit(self): url = target["admin"]["debit"]["endpoint"] filename = target["admin"]["debit"]["filename"] self.requests_helper.send_post_request(url, "/DEBIT", filename) @task(1) def perform_credit(self): url = target["admin"]["credit"]["endpoint"] filename = target["admin"]["credit"]["filename"] self.requests_helper.send_post_request(url, "/CREDIT", filename) @task(1) def approve_transfer_request(self): url = target["admin"]["approve_request"]["endpoint"] filename = target["admin"]["approve_request"]["filename"] self.requests_helper.send_post_request(url, "/EXECUTE TRANSFER REQUEST", filename) @task(1) def cancel_transfer_request(self): url = target["admin"]["cancel_request"]["endpoint"] filename = target["admin"]["cancel_request"]["filename"] self.requests_helper.send_post_request(url, "/CANCEL TRANSFER REQUEST", filename) # @task(2) def stop(self): self.interrupt()