189 lines
8.1 KiB
Python
189 lines
8.1 KiB
Python
from tests.base_test import BaseTest
|
|
|
|
from crc.models.data_store import DataStoreModel, DataStoreSchema
|
|
from crc.services.user_file_service import UserFileService
|
|
from crc.models.file import FileModel
|
|
from crc import session
|
|
|
|
import json
|
|
|
|
|
|
class DataStoreTest(BaseTest):
|
|
TEST_STUDY_ITEM = {
|
|
"key": "MyKey",
|
|
"workflow_id": 12,
|
|
"study_id": 42,
|
|
"task_id": "MyTask",
|
|
"spec_id": "My Spec Name",
|
|
"value": "Some Value"
|
|
}
|
|
TEST_FILE_ITEM = {
|
|
"key": "MyKey",
|
|
"workflow_id": 12,
|
|
"task_id": "MyTask",
|
|
"spec_id": "My Spec Name",
|
|
"value": "Some Value"
|
|
}
|
|
|
|
def add_test_study_data(self):
|
|
study_data = DataStoreSchema().dump(self.TEST_STUDY_ITEM)
|
|
rv = self.app.post('/v1.0/datastore',
|
|
content_type="application/json",
|
|
headers=self.logged_in_headers(),
|
|
data=json.dumps(study_data))
|
|
self.assert_success(rv)
|
|
return json.loads(rv.get_data(as_text=True))
|
|
|
|
def add_test_user_data(self):
|
|
study_data = DataStoreSchema().dump(self.TEST_STUDY_ITEM)
|
|
study_data['user_id'] = 'dhf8r'
|
|
del(study_data['study_id'])
|
|
study_data['value'] = 'User Value'
|
|
rv = self.app.post('/v1.0/datastore',
|
|
content_type="application/json",
|
|
headers=self.logged_in_headers(),
|
|
data=json.dumps(study_data))
|
|
self.assert_success(rv)
|
|
return json.loads(rv.get_data(as_text=True))
|
|
|
|
def add_test_file_data(self, file_id, value):
|
|
file_data = DataStoreSchema().dump(self.TEST_FILE_ITEM)
|
|
file_data['file_id'] = file_id
|
|
file_data['value'] = value
|
|
rv = self.app.post('/v1.0/datastore',
|
|
content_type="application/json",
|
|
headers=self.logged_in_headers(),
|
|
data=json.dumps(file_data))
|
|
self.assert_success(rv)
|
|
return json.loads(rv.get_data(as_text=True))
|
|
|
|
def test_get_study_data(self):
|
|
"""Generic test, but pretty detailed, in that the study should return a categorized list of workflows
|
|
This starts with out loading the example data, to show that all the bases are covered from ground 0."""
|
|
|
|
"""NOTE: The protocol builder is not enabled or mocked out. As the master workflow (which is empty),
|
|
and the test workflow do not need it, and it is disabled in the configuration."""
|
|
|
|
new_study = self.add_test_study_data()
|
|
new_study = session.query(DataStoreModel).filter_by(id=new_study["id"]).first()
|
|
|
|
api_response = self.app.get('/v1.0/datastore/%i' % new_study.id,
|
|
headers=self.logged_in_headers(), content_type="application/json")
|
|
self.assert_success(api_response)
|
|
d = api_response.get_data(as_text=True)
|
|
study_data = DataStoreSchema().loads(d)
|
|
|
|
self.assertEqual(study_data.key, self.TEST_STUDY_ITEM['key'])
|
|
self.assertEqual(study_data.value, self.TEST_STUDY_ITEM['value'])
|
|
self.assertEqual(study_data.user_id, None)
|
|
|
|
def test_update_datastore(self):
|
|
|
|
new_study = self.add_test_study_data()
|
|
new_study = session.query(DataStoreModel).filter_by(id=new_study["id"]).first()
|
|
new_study.value = 'MyNewValue'
|
|
api_response = self.app.put('/v1.0/datastore/%i' % new_study.id,
|
|
data=DataStoreSchema().dumps(new_study),
|
|
headers=self.logged_in_headers(), content_type="application/json")
|
|
self.assert_success(api_response)
|
|
|
|
api_response = self.app.get('/v1.0/datastore/%i' % new_study.id,
|
|
headers=self.logged_in_headers(), content_type="application/json")
|
|
self.assert_success(api_response)
|
|
study_data = DataStoreSchema().loads(api_response.get_data(as_text=True))
|
|
|
|
self.assertEqual(study_data.key, self.TEST_STUDY_ITEM['key'])
|
|
self.assertEqual(study_data.value, 'MyNewValue')
|
|
self.assertEqual(study_data.user_id, None)
|
|
|
|
def test_delete_datastore(self):
|
|
|
|
new_study = self.add_test_study_data()
|
|
oldid = new_study['id']
|
|
new_study = session.query(DataStoreModel).filter_by(id=new_study["id"]).first()
|
|
rv = self.app.delete('/v1.0/datastore/%i' % new_study.id, headers=self.logged_in_headers())
|
|
self.assert_success(rv)
|
|
studyreponse = session.query(DataStoreModel).filter_by(id=oldid).first()
|
|
self.assertEqual(studyreponse,None)
|
|
|
|
def test_data_crosstalk(self):
|
|
"""Test to make sure that data saved for user or study is not accessible from the other method"""
|
|
|
|
|
|
new_study = self.add_test_study_data()
|
|
new_user = self.add_test_user_data()
|
|
|
|
api_response = self.app.get(f'/v1.0/datastore/user/{new_user["user_id"]}',
|
|
headers=self.logged_in_headers(), content_type="application/json")
|
|
self.assert_success(api_response)
|
|
d = json.loads(api_response.get_data(as_text=True))
|
|
self.assertEqual(d[0]['value'],'User Value')
|
|
|
|
api_response = self.app.get(f'/v1.0/datastore/study/{new_study["study_id"]}',
|
|
headers=self.logged_in_headers(), content_type="application/json")
|
|
|
|
self.assert_success(api_response)
|
|
d = json.loads(api_response.get_data(as_text=True))
|
|
self.assertEqual(d[0]['value'],'Some Value')
|
|
|
|
def test_datastore_file(self):
|
|
|
|
workflow = self.create_workflow('random_fact')
|
|
self.add_test_user_data()
|
|
test_file = UserFileService.add_workflow_file(workflow.id, 'xxx', 'xxx', 'my_file.docx', 'docx', b'this is it.')
|
|
|
|
# make sure we don't already have a datastore
|
|
api_response = self.app.get(f'/v1.0/datastore/file/{test_file.id}',
|
|
headers=self.logged_in_headers(),
|
|
content_type="application/json")
|
|
self.assert_success(api_response)
|
|
data = json.loads(api_response.get_data(as_text=True))
|
|
self.assertEqual(0, len(data))
|
|
|
|
# add datastore
|
|
self.add_test_file_data(test_file.id, 'Some File Data Value')
|
|
|
|
# make sure we can get the datastore
|
|
api_response = self.app.get(f'/v1.0/datastore/file/{test_file.id}',
|
|
headers=self.logged_in_headers(),
|
|
content_type="application/json")
|
|
self.assert_success(api_response)
|
|
data = json.loads(api_response.get_data(as_text=True))
|
|
|
|
self.assertEqual(1, len(data))
|
|
self.assertEqual('MyKey', data[0]['key'])
|
|
self.assertEqual('Some File Data Value', data[0]['value'])
|
|
|
|
def test_datastore_files(self):
|
|
|
|
workflow = self.create_workflow('random_fact')
|
|
self.add_test_user_data()
|
|
test_file = UserFileService.add_workflow_file(workflow.id, 'xxx', 'xxx', 'my_file.docx', 'docx', b'this is it.')
|
|
|
|
# add datastore
|
|
value_1 = 'Some File Data Value 1'
|
|
self.add_test_file_data(test_file.id, value_1)
|
|
|
|
# make sure we have 1 item in the datastore
|
|
api_response_1 = self.app.get(f'/v1.0/datastore/file/{test_file.id}',
|
|
headers=self.logged_in_headers(), content_type="application/json")
|
|
self.assert_success(api_response_1)
|
|
data_1 = json.loads(api_response_1.get_data(as_text=True))
|
|
|
|
self.assertEqual(1, len(data_1))
|
|
self.assertEqual('MyKey', data_1[0]['key'])
|
|
self.assertEqual(value_1, data_1[0]['value'])
|
|
|
|
# add second datastore
|
|
value_2 = 'Some File Data Value 2'
|
|
self.add_test_file_data(test_file.id, value_2)
|
|
|
|
# make sure we have 2 items in the datastore
|
|
api_response_2 = self.app.get(f'/v1.0/datastore/file/{test_file.id}',
|
|
headers=self.logged_in_headers(), content_type="application/json")
|
|
self.assert_success(api_response_2)
|
|
data_2 = json.loads(api_response_2.get_data(as_text=True))
|
|
self.assertEqual(2, len(data_2))
|
|
self.assertEqual(value_1, data_2[0]['value'])
|
|
self.assertEqual(value_2, data_2[1]['value'])
|