def setUp(self): class MockDrop(MockLabKey): api = 'deleteDomain.api' default_action = domain_controller default_success_body = {} self.service = MockDrop() payload = { 'schemaName': self.schema_name, 'queryName': self.query_name } self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': json.dumps(payload), 'headers': { 'Content-Type': 'application/json' }, 'timeout': 300 } self.args = [ mock_server_context(self.service), self.schema_name, self.query_name ]
def setUp(self): domain_definition = { 'kind': 'IntList', 'domainDesign': { 'name': 'TheTestList', 'fields': [{ 'name': 'theKey', 'rangeURI': 'int' }] }, 'options': { 'keyName': 'theKey' } } class MockCreate(MockLabKey): api = 'createDomain.api' default_action = domain_controller default_success_body = domain_definition self.service = MockCreate() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': json.dumps(domain_definition), 'headers': { 'Content-Type': 'application/json' }, 'timeout': 300 } self.args = [mock_server_context(self.service), domain_definition]
def setUp(self): data_rows = [] # Generate the Run object(s) run = Run() run.name = 'python upload' run.data_rows = data_rows run.properties['RunFieldName'] = 'Run Field Value' # Generate the Batch object(s) batch = Batch() batch.runs = [run] batch.properties['PropertyName'] = 'Property Value' self.service = MockSaveBatch() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': '{"assayId": 12345, "batches": [{"batchProtocolId": 0, "comment": null, "created": null, "createdBy": null, "modified": null, "modifiedBy": null, "name": null, "properties": {"PropertyName": "Property Value"}, "runs": [{"comment": null, "created": null, "createdBy": null, "dataInputs": [], "dataRows": [], "experiments": [], "filePathRoot": null, "materialInputs": [], "materialOutputs": [], "modified": null, "modifiedBy": null, "name": "python upload", "properties": {"RunFieldName": "Run Field Value"}}]}]}', 'headers': { 'Content-type': 'application/json', 'Accept': 'text/plain' }, 'timeout': 300 } self.args = [mock_server_context(self.service), assay_id, batch]
def setUp(self): self.service = self.MockGetRoles() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': None, 'headers': None, 'timeout': 300 } self.args = [mock_server_context(self.service)]
def setUp(self): self.service = self.MockDeleteUser() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': { 'userId': [123] }, 'headers': None, 'timeout': 300 } self.args = [mock_server_context(self.service), self.__user_id]
def setUp(self): self.service = self.MockListGroups() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': { 'includeSiteGroups': True }, 'headers': None, 'timeout': 300 } self.args = [mock_server_context(self.service), True]
def setUp(self): self.service = self.MockResetPassword() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': { 'email': TestResetPassword.__email }, 'headers': None, 'timeout': 300 } self.args = [mock_server_context(self.service), self.__email]
def setUp(self): self.service = MockLoadBatch() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': '{"assayId": 12345, "batchId": 54321}', 'headers': { 'Content-type': 'application/json', 'Accept': 'text/plain' }, 'timeout': 300 } self.args = [mock_server_context(self.service), assay_id, batch_id]
def setUp(self): self.service = MockSelectRows() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': { "schemaName": schema, "query.queryName": query }, 'headers': None, 'timeout': 300 } self.args = [mock_server_context(self.service), schema, query]
def setUp(self): self.service = self.MockCreateUser() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': { 'email': TestCreateUser.__email, 'sendEmail': False }, 'headers': None, 'timeout': 300 } self.args = [mock_server_context(self.service), self.__email]
def setUp(self): self.service = MockExecuteSQL() sql = 'select * from ' + schema + '.' + query self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': { 'sql': sql, "schemaName": schema }, 'headers': None, 'timeout': 300 } self.args = [mock_server_context(self.service), schema, sql]
def setUp(self): self.service = self.MockRemoveGroupMember() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': { 'groupId': 123, 'principalIds': [321] }, 'headers': None, 'timeout': 300 } self.args = [ mock_server_context(self.service), self.__user_id, self.__group_id ]
def setUp(self): self.service = self.MockAddRole() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': { 'roleClassName': 'TestRole', 'principalId': 321, 'email': '*****@*****.**' }, 'headers': None, 'timeout': 300 } self.args = [ mock_server_context(self.service), self.__role, self.__user_id, self.__email ]
def setUp(self): self.service = MockDeleteRows() rows = '{id:1234}' self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': '{"queryName": "' + query + '", "rows": "' + rows + '", "schemaName": "' + schema + '"}', 'headers': { 'Content-Type': 'application/json' }, 'timeout': 300 } self.args = [mock_server_context(self.service), schema, query, rows]
def setUp(self): wiki_name = 'WikiToUpdate' wiki_body = 'Updated wiki body' self.service = MockUpdateWiki() self.expected_kwargs = { 'expected_args': [ self.service.get_server_url(), ], 'headers': { 'Content-type': 'application/json' }, 'params': { 'name': wiki_name } } self.args = [mock_server_context(self.service), wiki_name, wiki_body]
def setUp(self): class MockGet(MockLabKey): api = 'getDomain.api' default_action = domain_controller default_success_body = {} self.service = MockGet() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'headers': None, 'params': { 'schemaName': self.schema_name, 'queryName': self.query_name }, 'timeout': 300 } self.args = [ mock_server_context(self.service), self.schema_name, self.query_name ]
def setUp(self): message_title = 'Test Insert Message' message_body = 'The body of the message' render_as = 'HTML' expected_content = { 'body': message_body, 'title': message_title, 'rendererType': render_as } self.service = MockPostMessage() self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': expected_content, 'headers': None, 'timeout': 300 } self.args = [ mock_server_context(self.service), message_title, message_body, render_as ]
def setUp(self): class MockInferFields(MockLabKey): api = 'inferDomain.api' default_action = domain_controller default_success_body = {} self.service = MockInferFields() self.fd, self.path = tempfile.mkstemp() with os.fdopen(self.fd, 'w') as tmp: tmp.write('Name\tAge\nNick\t32\nBrian\t27\n') self.file = tmp self.expected_kwargs = { 'expected_args': [self.service.get_server_url()], 'data': None, 'files': { 'inferfile': self.file }, 'headers': None, 'timeout': 300 } self.args = [mock_server_context(self.service), self.file]