def test_get_article_file(setup, change_service, inmemory_receive_package,
test_package_A):
article_services = ArticleServices(change_service[0], change_service[1])
article_check = article_services.get_article_file('ID')
assert article_check is not None
xml_tree = XMLTree()
xml_tree.content = test_package_A[0].content
assert xml_tree.compare(article_check)
def inmemory_receive_package(change_service, test_package_A,
test_packA_assets_files):
article_services = ArticleServices(change_service[0], change_service[1])
xml_file = test_package_A[0]
return article_services.receive_package(id='ID',
files=test_packA_assets_files,
filename=xml_file.name,
content=xml_file.content,
content_size=xml_file.size)
def test_get_article_file_in_database(mocked_get_attachment, setup,
change_service, inmemory_receive_package,
xml_test, test_packA_filenames):
mocked_get_attachment.return_value = xml_test.encode('utf-8')
article_id = 'ID'
article_services = ArticleServices(change_service[0], change_service[1])
article_services.get_article_file(article_id)
mocked_get_attachment.assert_called_with(document_id=article_id,
file_id=test_packA_filenames[0])
def test_get_article_in_database(mocked_dataservices_read, setup,
change_service, inmemory_receive_package):
article_id = 'ID'
mocked_dataservices_read.return_value = {'document_id': article_id}
article_services = ArticleServices(change_service[0], change_service[1])
article_check = article_services.get_article_data(article_id)
assert article_check is not None
assert isinstance(article_check, dict)
mocked_dataservices_read.assert_called_with(article_id)
def couchdb_receive_package(dbserver_service, test_package_A,
test_packA_assets_files):
article_services = ArticleServices(dbserver_service[0],
dbserver_service[1])
xml_file = test_package_A[0]
return article_services.receive_package(id='ID',
files=test_packA_assets_files,
filename=xml_file.name,
content=xml_file.content,
content_size=xml_file.size)
def test_get_asset_file(change_service, test_package_A, test_packA_filenames,
test_packA_assets_files):
xml_file = test_package_A[0]
article_services = ArticleServices(change_service[0], change_service[1])
article_services.receive_package(id='ID',
files=test_packA_assets_files,
filename=xml_file.name,
content=xml_file.content,
content_size=xml_file.size)
for file in test_package_A[1:]:
content_type, content = article_services.get_asset_file(
'ID', file.name)
assert file.content == content
def test_get_article_record(setup, change_service, inmemory_receive_package):
article_services = ArticleServices(change_service[0], change_service[1])
article_id = 'ID'
article_check = article_services.get_article_data(article_id)
assert article_check is not None
assert isinstance(article_check, dict)
assert article_check.get('document_id') == article_id
assert article_check.get('document_type') == RecordType.ARTICLE.value
assert article_check.get('content') is not None
assert isinstance(article_check['content'], dict)
assert article_check['content'].get('xml') is not None
assert article_check.get('created_date') is not None
assert article_check.get('attachments') is not None
assert isinstance(article_check['attachments'], list)
def test_receive_package(change_service, test_package_A,
test_packA_assets_files):
article = Article('ID')
xml_file = test_package_A[0]
article.xml_file = xml_file
article.update_asset_files(test_packA_assets_files)
article_services = ArticleServices(change_service[0], change_service[1])
unexpected, missing = article_services.receive_package(
id='ID',
files=test_packA_assets_files,
filename=xml_file.name,
content=xml_file.content,
content_size=xml_file.size)
assert unexpected == []
assert missing == []
def test_get_article_in_database_not_found(mocked_dataservices_read, setup,
change_service,
inmemory_receive_package):
article_id = 'ID'
mocked_dataservices_read.return_value = {'document_id': article_id}
article_services = ArticleServices(change_service[0], change_service[1])
pytest.raises(ArticleServicesException, article_services.get_article_data,
article_id)
def test_get_asset_files(change_service, test_package_A,
test_packA_assets_files):
xml_file, files = test_package_A[0], test_package_A[1:]
article_services = ArticleServices(change_service[0], change_service[1])
article_services.receive_package(id='ID',
files=test_packA_assets_files,
filename=xml_file.name,
content=xml_file.content,
content_size=xml_file.size)
items, msg = article_services.get_asset_files('ID')
asset_contents = [
asset_data[1] for name, asset_data in items.items()
if len(asset_data) == 2
]
assert len(items) == len(files)
assert len(msg) == 0
for asset in files:
assert asset.content in asset_contents
def test_receive_xml_file(change_service, test_package_A,
test_packA_filenames):
file = test_package_A[0]
article_services = ArticleServices(change_service[0], change_service[1])
expected = {
'attachments': [test_packA_filenames[0]],
'content': {
'xml': test_packA_filenames[0],
},
'document_type': 'ART',
'document_id': 'ID',
}
expected_assets = test_packA_filenames[1:]
article_services.receive_xml_file(id='ID',
filename=file.name,
content=file.content,
content_size=file.size)
got = article_services.article_db_service.read('ID')
assert got['content']['xml'] == expected['content']['xml']
assert sorted(got['content'].get('assets')) == sorted(expected_assets)
assert sorted(got['attachments']) == sorted(expected['attachments'])
def _get_article_service(db_host, db_port, username, password):
database_config = {
'database_uri': '{}:{}'.format(db_host, db_port),
'database_username': username,
'database_password': password,
}
articles_database_config = database_config.copy()
articles_database_config['database_name'] = "articles"
changes_database_config = database_config.copy()
changes_database_config['database_name'] = "changes"
return ArticleServices(
CouchDBManager(**articles_database_config),
CouchDBManager(**changes_database_config)
)
def test_get_asset_file_not_found(mocked_get_attachment, setup, change_service,
inmemory_receive_package):
article_services = ArticleServices(change_service[0], change_service[1])
pytest.raises(ArticleServicesException, article_services.get_asset_file,
'ID', 'file_id')