def test_extract_and_add_attachments(self):
file_field = self._create_test_file_field()
extract_and_add_attachments(file_field, self.document)
attachments = IAttachmentStorage(self.document)
self.assertEquals(len(attachments.values()), 1)
res = attachments.get(file_field.filename)
self.assertEquals(res.id, file_field.filename)
def test_extract_and_add_attachments(self):
file_field = self._create_test_file_field()
extract_and_add_attachments(file_field, self.document)
attachments = IAttachmentStorage(self.document)
self.assertEquals(len(attachments.values()), 1)
res = attachments.get(file_field.filename)
self.assertEquals(res.id, file_field.filename)
def test_remove(self):
""" """
question = createObject('slc.underflow.question')
attachments = IAttachmentStorage(question)
self.assertEqual(len(attachments.keys()), 0)
self.assertEqual(len(attachments.values()), 0)
for fname in ['data1.dat', 'data2.dat']:
attachments.add(file.ATFile(fname))
self.assertEqual(len(attachments.keys()), 2)
self.assertEqual(len(attachments.values()), 2)
self.assertRaises(KeyError, attachments.remove, 'data3.dat')
attachments.remove('data1.dat')
self.assertEqual(len(attachments.keys()), 1)
self.assertTrue('data2.dat' in attachments.keys())
attachments.remove('data2.dat')
self.assertEqual(len(attachments.keys()), 0)
def test_remove(self):
""" """
question = createObject('Document')
alsoProvides(question, IAttachmentStoragable)
attachments = IAttachmentStorage(question)
self.assertEqual(len(attachments.keys()), 0)
self.assertEqual(len(attachments.values()), 0)
for fname in ['data1.dat', 'data2.dat']:
attachments.add(file.ATFile(fname))
self.assertEqual(len(attachments.keys()), 2)
self.assertEqual(len(attachments.values()), 2)
self.assertRaises(KeyError, attachments.remove, 'data3.dat')
attachments.remove('data1.dat')
self.assertEqual(len(attachments.keys()), 1)
self.assertTrue('data2.dat' in attachments.keys())
attachments.remove('data2.dat')
self.assertEqual(len(attachments.keys()), 0)
def test_add(self):
""" """
comment1 = createObject('plone.Comment')
attachments = IAttachmentStorage(comment1)
self.assertEqual(len(attachments.keys()), 0)
self.assertEqual(len(attachments.values()), 0)
f = file.ATFile('data.dat')
attachments.add(f)
self.assertEqual([k for k in attachments.keys()], [f.getId()])
self.assertEqual([v for v in attachments.values()], [f])
# DuplicateIDError is thrown when an object with the same id is
# added again.
self.assertRaises(DuplicateIDError, attachments.add, f)
i = image.ATImage('image.jpg')
attachments.add(i)
self.assertEqual(len(attachments.keys()), 2)
self.assertEqual(len(attachments.values()), 2)
self.assertTrue(i.getId()in attachments.keys())
self.assertTrue(i in attachments.values())
def test_add(self):
""" """
doc1 = createObject('Document')
alsoProvides(doc1, IAttachmentStoragable)
attachments = IAttachmentStorage(doc1)
self.assertEqual(len(attachments.keys()), 0)
self.assertEqual(len(attachments.values()), 0)
f = file.ATFile('data.dat')
attachments.add(f)
self.assertEqual([k for k in attachments.keys()], [f.getId()])
self.assertEqual([v for v in attachments.values()], [f])
# DuplicateIDError is thrown when an object with the same id is
# added again.
self.assertRaises(DuplicateIDError, attachments.add, f)
i = image.ATImage('image.jpg')
attachments.add(i)
self.assertEqual(len(attachments.keys()), 2)
self.assertEqual(len(attachments.values()), 2)
self.assertTrue(i.getId() in attachments.keys())
self.assertTrue(i in attachments.values())
def test_extract_and_add_attachments_with_token(self):
token = "{0}-{1}".format(TEST_USER_ID,
datetime.utcnow().strftime('%Y%m%d%H%M%S%f'))
temp_attachment = self._create_test_temp_attachment(token)
temp_attachments = IAttachmentStorage(self.workspace)
temp_attachments.add(temp_attachment)
file_field = self._create_test_file_field()
extract_and_add_attachments(file_field, self.document, self.workspace,
token)
attachments = IAttachmentStorage(self.document)
self.assertEquals(len(attachments.values()), 1)
self.assertTrue(file_field.filename in attachments.keys())
res = attachments.get(file_field.filename)
self.assertEquals(res.id, file_field.filename)
self.assertTrue('/'.join(res.getPhysicalPath()).startswith('/'.join(
self.workspace.getPhysicalPath())))
def test_extract_and_add_attachments_with_token(self):
token = "{0}-{1}".format(
TEST_USER_ID,
datetime.utcnow().strftime('%Y%m%d%H%M%S%f'))
temp_attachment = self._create_test_temp_attachment(token)
temp_attachments = IAttachmentStorage(self.workspace)
temp_attachments.add(temp_attachment)
file_field = self._create_test_file_field()
extract_and_add_attachments(
file_field, self.document, self.workspace, token)
attachments = IAttachmentStorage(self.document)
self.assertEquals(len(attachments.values()), 1)
self.assertTrue(file_field.filename in attachments.keys())
res = attachments.get(file_field.filename)
self.assertEquals(res.id, file_field.filename)
self.assertTrue(
'/'.join(res.getPhysicalPath()).startswith(
'/'.join(self.workspace.getPhysicalPath())))
