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())))