def __init__(self, master=None, controller_mailbox=None): self.master = master if master else Master.get_master(mailbox=controller_mailbox) self.controller_mailbox = controller_mailbox self.config_handler = ConfigHandler() initialize_storage_db() check_tool_availability() self.access_handler = AccessHandler() acl_parser = ACLConfigHandler() #database access authorization self.table_to_class = dict(workspaces=Workspaces, storages=Storage) self.cls_to_identifier_method = { Workspaces: Workspaces.get_by_full_name, Storage: Storage.get_storage_by_name} self.cls_to_acl = {Workspaces: acl_parser.get_workspaces_setter_acl, Storage: acl_parser.get_storages_setter_acl}
def test_cancel_job(): try: johann = User.get_user_by_username('johann') except UserNotFoundException: johann = User('johann', 'jtoken') job = Job('dmd_home', 'dmd1', 'home/centos/testdata/johann-directory-1', [False, False, False], johann) job_id = job.id master = Master.get_master() master.add_to_queue(job) master.cancel_job(job) worker = master.get_workers()[0] assert worker.name == 'worker1' assert worker.status.name == 'WAITING' assert Job.get_job_by_id(job_id).status.name == 'CANCELED'
def test_success_job(env): try: johann = User.get_user_by_username('johann') except UserNotFoundException: johann = User('johann', 'jtoken') job = Job(source_alias = 'dmd_home', target_alias = 'archive1', source_relative_path = 'export/master_data/tests/johann-directory-1', notification = [False, False, False], user = johann) job_id = job.id master = Master.get_master() worker = master.get_workers()[0] assert worker.name == 'worker1' assert worker.status.name == 'WAITING' master.add_to_queue(job) time.sleep(5) while (worker.status.name == 'ACTIVE'): #wait for worker to finish time.sleep(5) assert worker.status.name == 'WAITING' assert Job.get_job_by_id(job_id).status.name == 'DONE' #make sure the file was actually copied assert os.path.exists("/export/archive1/1-johann-directory.tar")
def test_constructor(): master = Master.get_master() assert (type(master) is Master) master2 = Master.get_master() assert (master == master2)