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]