def test_ignores_documents_with_different_titles(self, dc_mock, conn, fda_file):
file_attrs = {
'id': fda_file,
'documentcloud_id': '100-bar',
}
conn['database']['files'].update(file_attrs, ['id'])
remote_document_ids = ['100-foo']
conf = {
'DOCUMENTCLOUD_USERNAME': '******',
'DOCUMENTCLOUD_PASSWORD': '******',
'DOCUMENTCLOUD_PROJECT': 'project name',
}
_enable_documentcloud_mock(dc_mock, remote_document_ids)
processor.process(conf, conn)
assert not dc_mock().documents.get().delete.called
def test_ignores_documents_with_different_titles(self, dc_mock):
conf = {
'DOCUMENTCLOUD_USERNAME': '******',
'DOCUMENTCLOUD_PASSWORD': '******',
'DOCUMENTCLOUD_PROJECT': 'project name',
}
conn = {
'database': mock.Mock()
}
document_ids = ['100-foo']
conn['database'].query.return_value = [
{'documentcloud_id': '100-bar'},
]
_enable_documentcloud_mock(dc_mock, document_ids)
processor.process(conf, conn)
assert not dc_mock().documents.get().delete.called
def test_deletes_all_documents_not_in_the_database(self, dc_mock, conn, fda_file):
file_attrs = {
'id': fda_file,
'documentcloud_id': '200-bar',
}
conn['database']['files'].update(file_attrs, ['id'])
remote_document_ids = ['100-foo', '200-bar', '300-baz']
conf = {
'DOCUMENTCLOUD_USERNAME': '******',
'DOCUMENTCLOUD_PASSWORD': '******',
'DOCUMENTCLOUD_PROJECT': 'project name',
}
_enable_documentcloud_mock(dc_mock, remote_document_ids)
processor.process(conf, conn)
dc_mock().documents.delete.assert_has_calls([
mock.call('300'),
mock.call('100'),
], any_order=True)
def test_deletes_all_documents_not_in_the_database(self, dc_mock):
conf = {
'DOCUMENTCLOUD_USERNAME': '******',
'DOCUMENTCLOUD_PASSWORD': '******',
'DOCUMENTCLOUD_PROJECT': 'project name',
}
conn = {
'database': mock.Mock()
}
document_ids = ['100-foo', '200-bar', '300-baz']
conn['database'].query.return_value = [
{'documentcloud_id': '200-bar'},
]
_enable_documentcloud_mock(dc_mock, document_ids)
processor.process(conf, conn)
dc_mock().documents.delete.assert_has_calls([
mock.call('300'),
mock.call('100'),
])
