async def test_pattern_with_non_existing_folder(self):
corpus = TestHelper.get_local_corpus(
self.test_subpath, "test_pattern_with_non_existing_folder")
content = TestHelper.get_input_file_content(
self.test_subpath, "test_pattern_with_non_existing_folder",
"entities.manifest.cdm.json")
manifest_content = ManifestContent()
manifest_content.decode(content)
cdmManifest = ManifestPersistence.from_object(
CdmCorpusContext(corpus, None), "entities", "local", "/",
manifest_content)
error_logged = 0
def callback(level: CdmStatusLevel, message: str):
if 'Failed to fetch all files in the folder location \'local:/testLocation\' described by a partition pattern. Exception:' in message:
nonlocal error_logged
error_logged += 1
corpus.set_event_callback(callback, CdmStatusLevel.WARNING)
await cdmManifest.file_status_check_async()
self.assertEqual(1, error_logged)
self.assertEqual(len(cdmManifest.entities[0].data_partitions), 0)
# make sure the last check time is still being set
self.assertIsNotNone(
cdmManifest.entities[0].data_partition_patterns[0].
last_file_status_check_time)
def test_load_local_entity_with_data_partition_pattern(self):
content = TestHelper.get_input_file_content(
self.test_subpath,
"test_load_local_entity_with_data_partition_pattern",
"entities.manifest.cdm.json")
manifest_content = ManifestContent()
manifest_content.decode(content)
cdmManifest = ManifestPersistence.from_object(
CdmCorpusContext(CdmCorpusDefinition(), None), "entities",
"testNamespace", "/", manifest_content)
self.assertEqual(len(cdmManifest.entities), 1)
self.assertEqual(cdmManifest.entities[0].object_type,
CdmObjectType.LOCAL_ENTITY_DECLARATION_DEF)
entity = cdmManifest.entities[0]
self.assertEqual(len(entity.data_partition_patterns), 1)
pattern = entity.data_partition_patterns[0]
self.assertEqual(pattern.name, "testPattern")
self.assertEqual(pattern.explanation, "test explanation")
self.assertEqual(pattern.root_location, "test location")
self.assertEqual(pattern.regular_expression, "\\s*")
self.assertEqual(len(pattern.parameters), 2)
self.assertEqual(pattern.parameters[0], "testParam1")
self.assertEqual(pattern.parameters[1], "testParam2")
self.assertEqual(pattern.specialized_schema, "test special schema")
self.assertEqual(len(pattern.exhibits_traits), 1)
async def test_from_and_to_data(self):
with open(os.path.join(ROOT_PATH,
'default.manifest.cdm.json')) as manifestFile:
manifestContent = ManifestContent()
manifestContent.decode(manifestFile.read())
corpus = CdmCorpusDefinition()
corpus.storage.default_namespace = 'local'
corpus.ctx.update_logging_options(level='WARNING')
corpus.storage.mount('cdm',
LocalAdapter(root="../CDM.SchemaDocuments"))
corpus.storage.mount('local', LocalAdapter(root=ROOT_PATH))
corpus.storage.mount(
'remote', RemoteAdapter(hosts={"contoso": "http://contoso.com"}))
folder = corpus.storage.fetch_root_folder('local')
manifest = await corpus.fetch_object_async('default.manifest.cdm.json',
folder)
data = await ManifestPersistence.to_data(manifest, None, None)
for entity in manifest.entities:
entity_def = await corpus.fetch_object_async(
entity.entity_path, manifest)
await manifest.save_as_async('test_output/new.manifest.cdm.json',
save_referenced=True)
self.assertDictEqual(json.loads(manifestContent.encode()),
json.loads(data.encode()))
def test_load_local_entity_with_data_partition_pattern(self):
content = TestHelper.get_input_file_content(
self.test_subpath,
'test_load_local_entity_with_data_partition_pattern',
'entities.manifest.cdm.json')
manifest_content = ManifestContent()
manifest_content.decode(content)
cdm_manifest = ManifestPersistence.from_object(
CdmCorpusContext(CdmCorpusDefinition(), None), 'entities',
'testNamespace', '/', manifest_content)
self.assertEqual(len(cdm_manifest.entities), 2)
entity1 = cdm_manifest.entities[0]
self.assertEqual(entity1.object_type,
CdmObjectType.LOCAL_ENTITY_DECLARATION_DEF)
self.assertEqual(len(entity1.data_partition_patterns), 1)
pattern1 = entity1.data_partition_patterns[0]
self.assertEqual(pattern1.name, 'testPattern')
self.assertEqual(pattern1.explanation, 'test explanation')
self.assertEqual(pattern1.root_location, 'test location')
self.assertEqual(pattern1.regular_expression, '\\s*')
self.assertEqual(len(pattern1.parameters), 2)
self.assertEqual(pattern1.parameters[0], 'testParam1')
self.assertEqual(pattern1.parameters[1], 'testParam2')
self.assertEqual(pattern1.specialized_schema, 'test special schema')
self.assertEqual(len(pattern1.exhibits_traits), 1)
entity2 = cdm_manifest.entities[1]
self.assertEqual(entity2.object_type,
CdmObjectType.LOCAL_ENTITY_DECLARATION_DEF)
self.assertEqual(len(entity2.data_partition_patterns), 1)
pattern2 = entity2.data_partition_patterns[0]
self.assertEqual(pattern2.name, 'testPattern2')
self.assertEqual(pattern2.root_location, 'test location2')
self.assertEqual(pattern2.glob_pattern, '/*.csv')
manifest_data = ManifestPersistence.to_data(cdm_manifest, None, None)
self.assertEqual(len(manifest_data.entities), 2)
entity_data1 = manifest_data.entities[0]
self.assertEqual(len(entity_data1.dataPartitionPatterns), 1)
pattern_data1 = entity_data1.dataPartitionPatterns[0]
self.assertEqual(pattern_data1.name, 'testPattern')
self.assertEqual(pattern_data1.explanation, 'test explanation')
self.assertEqual(pattern_data1.rootLocation, 'test location')
self.assertEqual(pattern_data1.regularExpression, '\\s*')
self.assertEqual(len(pattern_data1.parameters), 2)
self.assertEqual(pattern_data1.parameters[0], 'testParam1')
self.assertEqual(pattern_data1.parameters[1], 'testParam2')
self.assertEqual(pattern_data1.specializedSchema,
'test special schema')
self.assertEqual(len(pattern_data1.exhibitsTraits), 1)
pattern_data2 = manifest_data.entities[1].dataPartitionPatterns[0]
self.assertEqual(pattern_data2.name, 'testPattern2')
self.assertEqual(pattern_data2.rootLocation, 'test location2')
self.assertEqual(pattern_data2.globPattern, '/*.csv')
async def test_pattern_with_non_existing_folder(self):
corpus = TestHelper.get_local_corpus(self.test_subpath, "test_pattern_with_non_existing_folder")
content = TestHelper.get_input_file_content(self.test_subpath, "test_pattern_with_non_existing_folder", "entities.manifest.cdm.json")
manifest_content = ManifestContent()
manifest_content.decode(content)
cdmManifest = ManifestPersistence.from_object(CdmCorpusContext(corpus, None), "entities", "local", "/", manifest_content)
await cdmManifest.file_status_check_async()
self.assertEqual(len(cdmManifest.entities[0].data_partitions), 0)
# make sure the last check time is still being set
self.assertIsNotNone(cdmManifest.entities[0].data_partition_patterns[0].last_file_status_check_time)
async def test_load_local_entitiy_with_data_partition(self):
content = TestHelper.get_input_file_content(
self.test_subpath, 'test_load_local_entity_with_data_partition',
'entities.manifest.cdm.json')
manifest_content = ManifestContent()
manifest_content.decode(content)
cdm_manifest = await ManifestPersistence.from_data(
CdmCorpusContext(CdmCorpusDefinition()), 'entities',
'testNamespace', '/', manifest_content)
self.assertEqual(len(cdm_manifest.entities), 1)
self.assertEqual(cdm_manifest.entities[0].object_type,
CdmObjectType.LOCAL_ENTITY_DECLARATION_DEF)
entity = cdm_manifest.entities[0]
self.assertEqual(len(entity.data_partitions), 2)
relative_partition = entity.data_partitions[0]
self.assertEqual(relative_partition.name, 'Sample data partition')
self.assertEqual(relative_partition.location, 'test/location')
# self.assertEqual(TimeUtils.GetFormattedDateString(relative_partition.LastFileModifiedTime), '2008-09-15T23:53:23.000Z')
self.assertEqual(len(relative_partition.exhibits_traits), 1)
self.assertEqual(relative_partition.specialized_schema, 'teststring')
test_list = relative_partition.arguments['test']
self.assertEqual(len(test_list), 3)
self.assertEqual(test_list[0], 'something')
self.assertEqual(test_list[1], 'somethingelse')
self.assertEqual(test_list[2], 'anotherthing')
key_list = relative_partition.arguments['KEY']
self.assertEqual(len(key_list), 1)
self.assertEqual(key_list[0], 'VALUE')
self.assertFalse('wrong' in relative_partition.arguments)
absolute_partition = entity.data_partitions[1]
self.assertEqual(absolute_partition.location,
"local:/some/test/location")
async def _load_document_from_path_async(self, folder: 'CdmFolderDefinition', doc_name: str,
doc_container: 'CdmDocumentDefinition',
res_opt: Optional[ResolveOptions] = None) \
-> 'CdmDocumentDefinition':
# go get the doc
doc_content = None # type: Optional[CdmDocumentDefinition]
json_data = None
fs_modified_time = None
doc_path = folder.folder_path + doc_name
adapter = self._ctx.corpus.storage.fetch_adapter(
folder.namespace) # type: StorageAdapter
try:
if adapter.can_read():
# log message used by navigator, do not change or remove
logger.debug(self._ctx, self._TAG,
self._load_document_from_path_async.__name__,
doc_path, 'request file: {}'.format(doc_path))
json_data = await adapter.read_async(doc_path)
# log message used by navigator, do not change or remove
logger.debug(self._ctx, self._TAG,
self._load_document_from_path_async.__name__,
doc_path, 'received file: {}'.format(doc_path))
else:
raise Exception('Storage Adapter is not enabled to read.')
except Exception as e:
# log message used by navigator, do not change or remove
logger.debug(self._ctx, self._TAG,
self._load_document_from_path_async.__name__,
doc_path, 'fail file: {}'.format(doc_path))
# when shallow validation is enabled, log messages about being unable to find referenced documents as warnings instead of errors.
if res_opt and res_opt.shallow_validation:
logger.warning(
self._ctx, self._TAG,
PersistenceLayer._load_document_from_path_async.__name__,
doc_path, CdmLogCode.WARN_PERSIST_FILE_READ_FAILURE,
doc_path, folder.Namespace, e)
else:
logger.error(self._ctx, self._TAG,
self._load_document_from_path_async.__name__,
doc_path,
CdmLogCode.ERR_PERSIST_FILE_READ_FAILURE,
doc_path, str(folder.namespace), e)
return None
try:
fs_modified_time = await adapter.compute_last_modified_time_async(
doc_path)
except Exception as e:
logger.warning(
self._ctx, self._TAG,
PersistenceLayer._load_document_from_path_async.__name__,
doc_path, CdmLogCode.WARN_PERSIST_FILE_MOD_COMPUTE_FAILED,
e.Message)
if not doc_name:
logger.error(self._ctx, self._TAG,
self._load_document_from_path_async.__name__,
doc_path, CdmLogCode.ERR_PERSIST_NULL_DOC_NAME)
return None
doc_name_lower = doc_name.lower()
# If loading an model.json file, check that it is named correctly.
if doc_name_lower.endswith(
self.MODEL_JSON_EXTENSION
) and not doc_name.lower() == self.MODEL_JSON_EXTENSION:
logger.error(self._ctx, self._TAG,
self._load_document_from_path_async.__name__,
doc_path,
CdmLogCode.ERR_PERSIST_DOC_NAME_LOAD_FAILURE,
doc_name, self.MODEL_JSON_EXTENSION)
return None
try:
if doc_name_lower.endswith(PersistenceLayer.MANIFEST_EXTENSION
) or doc_name_lower.endswith(
PersistenceLayer.FOLIO_EXTENSION):
from cdm.persistence.cdmfolder import ManifestPersistence
from cdm.persistence.cdmfolder.types import ManifestContent
manifest = ManifestContent()
manifest.decode(json_data)
doc_content = ManifestPersistence.from_object(
self._ctx, doc_name, folder.namespace, folder.folder_path,
manifest)
elif doc_name_lower.endswith(
PersistenceLayer.MODEL_JSON_EXTENSION):
if doc_name_lower != PersistenceLayer.MODEL_JSON_EXTENSION:
logger.error(self._ctx, self._TAG,
self._load_document_from_path_async.__name__,
doc_path,
CdmLogCode.ERR_PERSIST_DOC_NAME_LOAD_FAILURE,
doc_name, self.MODEL_JSON_EXTENSION)
return None
from cdm.persistence.modeljson import ManifestPersistence
from cdm.persistence.modeljson.types import Model
model = Model()
model.decode(json_data)
doc_content = await ManifestPersistence.from_object(
self._ctx, model, folder)
elif doc_name_lower.endswith(PersistenceLayer.CDM_EXTENSION):
from cdm.persistence.cdmfolder import DocumentPersistence
from cdm.persistence.cdmfolder.types import DocumentContent
document = DocumentContent()
document.decode(json_data)
doc_content = DocumentPersistence.from_object(
self._ctx, doc_name, folder.namespace, folder.folder_path,
document)
else:
# Could not find a registered persistence class to handle this document type.
logger.error(self._ctx, self._TAG,
self._load_document_from_path_async.__name__,
doc_path, CdmLogCode.ERR_PERSIST_CLASS_MISSING,
doc_name)
return None
except Exception as e:
logger.error(self._ctx, self._TAG,
self._load_document_from_path_async.__name__,
doc_path,
CdmLogCode.ERR_PERSIST_DOC_CONVERSION_FAILURE,
doc_path, e)
return None
# add document to the folder, this sets all the folder/path things, caches name to content association and may trigger indexing on content
if doc_content is not None:
if doc_container:
# there are situations where a previously loaded document must be re-loaded.
# the end of that chain of work is here where the old version of the document has been removed from
# the corpus and we have created a new document and loaded it from storage and after this call we will probably
# add it to the corpus and index it, etc.
# it would be really rude to just kill that old object and replace it with this replicant, especially because
# the caller has no idea this happened. so... sigh ... instead of returning the new object return the one that
# was just killed off but make it contain everything the new document loaded.
doc_content = doc_content.copy(
ResolveOptions(wrt_doc=doc_container,
directives=self._ctx.corpus.
default_resolution_directives),
doc_container)
folder.documents.append(doc_content, doc_name)
doc_content._file_system_modified_time = fs_modified_time
doc_content._is_dirty = False
return doc_content
async def _fetch_document_from_folder_path_async(
self, document_path: str, adapter: 'StorageAdapterBase',
force_reload: bool) -> 'CdmDocumentDefinition':
"""Gets the document from folder path.
arguments:
path: The path.
adapter: The storage adapter where the document can be found."""
from .cdm_corpus_def import FOLIO_EXTENSION, MANIFEST_EXTENSION, MODEL_JSON_EXTENSION
doc_name = None
first = document_path.find('/')
if first < 0:
doc_name = document_path
else:
doc_name = document_path[0:first]
if doc_name in self._document_lookup and not force_reload:
return self._document_lookup[doc_name]
is_cdm_folder = doc_name.endswith(
FOLIO_EXTENSION) or doc_name.endswith(MANIFEST_EXTENSION)
is_model_json = doc_name.endswith(MODEL_JSON_EXTENSION)
# got that doc?
doc = None
# go get the doc
doc_path = '{}{}'.format(self.folder_path, doc_name)
json_data = None
fs_modified_time = None
try:
if adapter.can_read():
json_data = await adapter.read_async(doc_path)
fs_modified_time = await adapter.compute_last_modified_time_async(
adapter.create_adapter_path(doc_path))
except Exception as e:
self.corpus.ctx.logger.exception(
'Could not read %s from the \'%s\' namespace.', doc_path,
self.namespace)
return None
if is_cdm_folder:
from cdm.persistence.cdmfolder import ManifestPersistence
from cdm.persistence.cdmfolder.types import ManifestContent
manifest = ManifestContent()
manifest.decode(json_data)
doc = await ManifestPersistence.from_data(self.ctx, doc_name,
self.namespace,
self.folder_path,
manifest)
doc.folder = self
self.documents.append(doc)
self.corpus._add_document_objects(self, doc)
self._document_lookup[doc_name] = doc
elif is_model_json:
from cdm.persistence.modeljson import ManifestPersistence
from cdm.persistence.modeljson.types import Model
model = Model()
model.decode(json_data)
doc = await ManifestPersistence.from_data(self.ctx, model, self)
doc.folder = self
doc.folder_path = self.folder_path
else:
from cdm.persistence.cdmfolder import DocumentPersistence
from cdm.persistence.cdmfolder.types import DocumentContent
document = DocumentContent()
document.decode(json_data)
doc = self.documents.append(await DocumentPersistence.from_data(
self.ctx, doc_name, self.namespace, self.folder_path,
document))
doc._file_system_modified_time = fs_modified_time
doc._is_dirty = False
return doc
async def load_document_from_path_async(
folder: 'CdmFolderDefinition', doc_name: str,
doc_container: 'CdmDocumentDefinition') -> 'CdmDocumentDefinition':
is_cdm_folder = doc_name.lower().endswith(
CdmCorpusDefinition._FOLIO_EXTENSION) or doc_name.lower().endswith(
CdmCorpusDefinition._MANIFEST_EXTENSION)
is_model_json = doc_name.lower().endswith(
CdmCorpusDefinition._MODEL_JSON_EXTENSION)
# go get the doc
doc_content = None # type: Optional[CdmDocumentDefinition]
json_data = None
fs_modified_time = None
ctx = folder.ctx
doc_path = folder.folder_path + doc_name
adapter = ctx.corpus.storage.fetch_adapter(
folder.namespace) # type: StorageAdapter
try:
if adapter.can_read():
json_data = await adapter.read_async(doc_path)
fs_modified_time = await adapter.compute_last_modified_time_async(
adapter.create_adapter_path(doc_path))
ctx.logger.info('read file: {}'.format(doc_path))
except Exception as e:
ctx.logger.error(
'could not read %s from the \'%s\' namespace.\n Reason: \n%s',
doc_path, folder.namespace, e)
return None
try:
if is_cdm_folder:
from cdm.persistence.cdmfolder import ManifestPersistence
from cdm.persistence.cdmfolder.types import ManifestContent
manifest = ManifestContent()
manifest.decode(json_data)
doc_content = await ManifestPersistence.from_data(
ctx, doc_name, folder.namespace, folder.folder_path, manifest)
elif is_model_json:
if doc_name.lower() != CdmCorpusDefinition._MODEL_JSON_EXTENSION:
ctx.logger.error(
'Failed to load \'{}\', as it\'s not an acceptable filename. It must be model.json'
.format(doc_name))
return None
from cdm.persistence.modeljson import ManifestPersistence
from cdm.persistence.modeljson.types import Model
model = Model()
model.decode(json_data)
doc_content = await ManifestPersistence.from_data(
ctx, model, folder)
else:
from cdm.persistence.cdmfolder import DocumentPersistence
from cdm.persistence.cdmfolder.types import DocumentContent
document = DocumentContent()
document.decode(json_data)
doc_content = await DocumentPersistence.from_data(
ctx, doc_name, folder.namespace, folder.folder_path, document)
except Exception as e:
ctx.logger.error('Could not convert \'{}\'. Reason \'{}\''.format(
doc_path, e))
return None
# add document to the folder, this sets all the folder/path things, caches name to content association and may trigger indexing on content
if doc_content is not None:
if doc_container:
# there are situations where a previously loaded document must be re-loaded.
# the end of that chain of work is here where the old version of the document has been removed from
# the corpus and we have created a new document and loaded it from storage and after this call we will probably
# add it to the corpus and index it, etc.
# it would be really rude to just kill that old object and replace it with this replicant, especially because
# the caller has no idea this happened. so... sigh ... instead of returning the new object return the one that
# was just killed off but make it contain everything the new document loaded.
doc_content = doc_content.copy(
ResolveOptions(wrt_doc=doc_container), doc_container)
folder.documents.append(doc_content, doc_name)
doc_content._file_system_modified_time = fs_modified_time
doc_content._is_dirty = False
return doc_content
entityJson = entity.encode()
entityCheck = JObject(entityJson)
entityCheckJson = entityCheck.encode()
assert entityJson == entityCheckJson
# Test loading a manifest from a file, and printing it out
manifestJson = ''
with open('tests/cdmfolder/default.manifest.cdm.json') as manifestJsonFile:
manifestJson = manifestJsonFile.read()
manifestContent = ManifestContent()
manifestContent.decode(manifestJson)
kvprint('name', manifestContent.manifestName, 3, 0)
kvprint('explanation', str(manifestContent.get('explanation')), 3, 0)
kvprint('jsonSchemaSemanticVersion', manifestContent.jsonSchemaSemanticVersion,
3, 0)
if manifestContent.get('imports'):
kvprint('imports', '', 6, 0)
for the_import in manifestContent.get('imports'):
kvprint('corpusPath', the_import.corpusPath, 3, 4)
kvprint('moniker', the_import.moniker, 3, 4)
if manifestContent.get('entities'):
kvprint('entities', '', 6, 0)
for entity in manifestContent.get('entities'):