def test_export_as_zip(self):
context = object()
request = object()
assembler = self.mocker.mock()
self.mock_adapter(assembler, interfaces.IPDFAssembler,
(Interface, Interface))
self.expect(assembler(ANY, ANY)).result(assembler)
self.expect(assembler.build_zip(
request=request)).result(request)
self.replay()
aspdf = ExportPDFView(context, request)
self.assertEqual(aspdf.export(output='zip'), request)
def test_export_with_unkown_output(self):
context = object()
request = object()
assembler = self.mocker.mock()
self.mock_adapter(assembler, interfaces.IPDFAssembler,
(Interface, Interface))
self.expect(assembler(ANY, ANY)).result(assembler)
self.replay()
aspdf = ExportPDFView(context, request)
with self.assertRaises(ValueError) as cm:
self.assertEqual(aspdf.export(output='foo'), request)
self.assertEqual(str(cm.exception),
'Unkown output "foo"')
