def setUp(self):
unittest.TestCase.setUp(self)
self.options = {}
self.global_data = {}
self.msg_data = {}
patch("pad.plugins.pdf_info.PDFInfoPlugin.options",
self.options).start()
self.mock_ctxt = MagicMock(
**{
"get_plugin_data.side_effect":
lambda p, k: self.msg_data[k],
"set_plugin_data."
"side_effect":
lambda p, k, v: self.msg_data.setdefault(k, v),
})
self.mock_msg = MagicMock(
**{
"get_plugin_data.side_effect":
lambda p, k: self.msg_data[k],
"set_plugin_data."
"side_effect":
lambda p, k, v: self.msg_data.update({k: v}),
})
self.mock_msg.msg = None
self.plugin = pad.plugins.pdf_info.PDFInfoPlugin(self.mock_ctxt)
def setUp(self):
unittest.TestCase.setUp(self)
self.options = {}
self.global_data = {}
self.msg_data = {}
patch("pad.plugins.pdf_info.PDFInfoPlugin.options",
self.options).start()
self.mock_ctxt = MagicMock(**{
"get_plugin_data.side_effect": lambda p, k: self.msg_data[k],
"set_plugin_data."
"side_effect": lambda p, k, v: self.msg_data.setdefault(k, v),
})
self.mock_msg = MagicMock(**{
"get_plugin_data.side_effect": lambda p, k: self.msg_data[k],
"set_plugin_data."
"side_effect": lambda p, k, v: self.msg_data.update({k: v}),
})
self.mock_msg.msg = None
self.plugin = pad.plugins.pdf_info.PDFInfoPlugin(self.mock_ctxt)
def test_save_stats_encrypted(self):
"""Test the _save_stats method"""
patch("pad.plugins.pdf_info.PDFInfoPlugin._update_details").start()
patch(
"pad.plugins.pdf_info.PDFInfoPlugin._update_image_counts").start()
patch("pad.plugins.pdf_info.PDFInfoPlugin._update_pixel_coverage"
).start()
datastore = BytesIO()
with open("tests/data/pdftest_encrypted.pdf", "rb") as pdf_file:
datastore.write(pdf_file.read())
if not datastore.getvalue():
return
pdfs = {1: {"data": datastore, "name": "pdftest.pdf"}}
self.mock_msg.msg = new_email(pdfs)
results = []
for part in self.mock_msg.msg.walk():
payload = part.get_payload(decode=True)
if payload is None:
continue
results.append(self.plugin._save_stats(self.mock_msg, payload))
self.plugin._update_details.assert_not_called()
self.plugin._update_image_counts.assert_not_called()
self.plugin._update_pixel_coverage.assert_not_called()
self.assertEqual(results, [None])
def test_save_stats_text(self):
"""Test the _save_stats method"""
patch("pad.plugins.pdf_info.PDFInfoPlugin._update_details").start()
patch(
"pad.plugins.pdf_info.PDFInfoPlugin._update_image_counts").start()
patch("pad.plugins.pdf_info.PDFInfoPlugin._update_pixel_coverage"
).start()
datastore = BytesIO()
with open("tests/data/pdftest_text.pdf", "rb") as pdf_file:
datastore.write(pdf_file.read())
if not datastore.getvalue():
return
pdf_id = md5(datastore.getvalue()).hexdigest()
pdf_object = PyPDF2.PdfFileReader(datastore)
info = pdf_object.getDocumentInfo()
update_details_calls = [
call(self.mock_msg, pdf_id, "author", info.author),
call(self.mock_msg, pdf_id, "creator", info.creator),
call(self.mock_msg, pdf_id, "producer", info.producer),
call(self.mock_msg, pdf_id, "title", info.title)
]
pdfs = {1: {"data": datastore, "name": "pdftest.pdf"}}
self.mock_msg.msg = new_email(pdfs)
for part in self.mock_msg.msg.walk():
payload = part.get_payload(decode=True)
if payload is None:
continue
self.plugin._save_stats(self.mock_msg, payload)
self.plugin._update_details.assert_has_calls(update_details_calls)
def test_save_stats_encrypted(self):
"""Test the _save_stats method"""
patch("pad.plugins.pdf_info.PDFInfoPlugin._update_details").start()
patch(
"pad.plugins.pdf_info.PDFInfoPlugin._update_image_counts").start()
patch(
"pad.plugins.pdf_info.PDFInfoPlugin._update_pixel_coverage").start()
datastore = BytesIO()
with open("tests/data/pdftest_encrypted.pdf", "rb") as pdf_file:
datastore.write(pdf_file.read())
if not datastore.getvalue():
return
pdfs = {1: {"data": datastore, "name": "pdftest.pdf"}}
self.mock_msg.msg = new_email(pdfs)
results = []
for part in self.mock_msg.msg.walk():
payload = part.get_payload(decode=True)
if payload is None:
continue
results.append(self.plugin._save_stats(self.mock_msg, payload))
self.plugin._update_details.assert_not_called()
self.plugin._update_image_counts.assert_not_called()
self.plugin._update_pixel_coverage.assert_not_called()
self.assertEqual(results, [None])
def test_save_stats_text(self):
"""Test the _save_stats method"""
patch("pad.plugins.pdf_info.PDFInfoPlugin._update_details").start()
patch(
"pad.plugins.pdf_info.PDFInfoPlugin._update_image_counts").start()
patch(
"pad.plugins.pdf_info.PDFInfoPlugin._update_pixel_coverage").start()
datastore = BytesIO()
with open("tests/data/pdftest_text.pdf", "rb") as pdf_file:
datastore.write(pdf_file.read())
if not datastore.getvalue():
return
pdf_id = md5(datastore.getvalue()).hexdigest()
pdf_object = PyPDF2.PdfFileReader(datastore)
info = pdf_object.getDocumentInfo()
update_details_calls = [
call(self.mock_msg, pdf_id, "author", info.author),
call(self.mock_msg, pdf_id, "creator", info.creator),
call(self.mock_msg, pdf_id, "producer", info.producer),
call(self.mock_msg, pdf_id, "title", info.title)
]
pdfs = {1: {"data": datastore, "name": "pdftest.pdf"}}
self.mock_msg.msg = new_email(pdfs)
for part in self.mock_msg.msg.walk():
payload = part.get_payload(decode=True)
if payload is None:
continue
self.plugin._save_stats(self.mock_msg, payload)
self.plugin._update_details.assert_has_calls(update_details_calls)
def test_extract_metadata(self):
patch("oa.plugins.pdf_info.PDFInfoPlugin._add_name").start()
patch("oa.plugins.pdf_info.PDFInfoPlugin._update_counts").start()
patch("oa.plugins.pdf_info.PDFInfoPlugin._save_stats").start()
add_name_calls = []
update_counts_calls = []
save_stats_calls = []
pdfs = {}
allpdfinfo = (
{"details": {"/Author": "Author1", "/Creator": "unittest",
"/Created": "2016-02-11", "/Title": "pdftest"}},
{"details": {"/Author": "Author2", "/Creator": "unittest",
"/Created": "2016-02-11", "/Title": "pdftest2"}},
{"details": {"/Author": "Author3", "/Creator": "unittest",
"/Created": "2016-02-11", "/Title": "pdftest3"}},
{"details": {"/Author": "Author4", "/Creator": "unittest",
"/Created": "2016-02-11", "/Title": "pdftest4"},
"images": ("image1", (100, 100))},
)
for i, pdf_info in enumerate(allpdfinfo):
name = "%d.pdf" % i
if "images" not in pdf_info:
pdf_object = new_pdf(details=pdf_info["details"], name=name)
else:
pdfc = PDFWithAttachments(
details=pdf_info["details"], name=name)
image = new_image_string(pdf_info["images"][1])
pdfc.addAttachment(pdf_info["images"][0], image)
pdf_object = {"data": pdfc.as_file(), "name": name}
pdfs.update({i: pdf_object})
add_name_calls.append(call(self.mock_msg, name))
update_counts_calls.append(call(self.mock_msg, incr=1))
save_stats_calls.append(call(self.mock_msg,
pdf_object["data"].read()))
self.mock_msg.msg = new_email(pdfs)
for part in self.mock_msg.msg.walk():
payload = part.get_payload(decode=True)
self.plugin.extract_metadata(self.mock_msg, payload, None, part)
self.plugin._add_name.assert_has_calls(add_name_calls)
self.plugin._update_counts.assert_has_calls(update_counts_calls)
self.plugin._save_stats.assert_has_calls(save_stats_calls)
def test_extract_metadata(self):
patch("pad.plugins.pdf_info.PDFInfoPlugin._add_name").start()
patch("pad.plugins.pdf_info.PDFInfoPlugin._update_counts").start()
patch("pad.plugins.pdf_info.PDFInfoPlugin._save_stats").start()
add_name_calls = []
update_counts_calls = []
save_stats_calls = []
pdfs = {}
allpdfinfo = (
{"details": {"/Author": "Author1", "/Creator": "unittest",
"/Created": "2016-02-11", "/Title": "pdftest"}},
{"details": {"/Author": "Author2", "/Creator": "unittest",
"/Created": "2016-02-11", "/Title": "pdftest2"}},
{"details": {"/Author": "Author3", "/Creator": "unittest",
"/Created": "2016-02-11", "/Title": "pdftest3"}},
{"details": {"/Author": "Author4", "/Creator": "unittest",
"/Created": "2016-02-11", "/Title": "pdftest4"},
"images": ("image1", (100, 100))},
)
for i, pdf_info in enumerate(allpdfinfo):
name = "%d.pdf" % i
if "images" not in pdf_info:
pdf_object = new_pdf(details=pdf_info["details"], name=name)
else:
pdfc = PDFWithAttachments(
details=pdf_info["details"], name=name)
image = new_image_string(pdf_info["images"][1])
pdfc.addAttachment(pdf_info["images"][0], image)
pdf_object = {"data": pdfc.as_file(), "name": name}
pdfs.update({i: pdf_object})
add_name_calls.append(call(self.mock_msg, name))
update_counts_calls.append(call(self.mock_msg, incr=1))
save_stats_calls.append(call(self.mock_msg,
pdf_object["data"].read()))
self.mock_msg.msg = new_email(pdfs)
for part in self.mock_msg.msg.walk():
payload = part.get_payload(decode=True)
self.plugin.extract_metadata(self.mock_msg, payload, None, part)
self.plugin._add_name.assert_has_calls(add_name_calls)
self.plugin._update_counts.assert_has_calls(update_counts_calls)
self.plugin._save_stats.assert_has_calls(save_stats_calls)
