From 40ef9afa9077e68b1e711a44e4d8eac6b4a8fd92 Mon Sep 17 00:00:00 2001
From: Ivan Tyagov <ivan@nexedi.com>
Date: Wed, 24 Feb 2010 08:17:55 +0000
Subject: [PATCH] Refactor test to use new script for contribute.

git-svn-id: https://svn.erp5.org/repos/public/erp5/trunk@33079 20353a03-c40f-0410-a6d1-a30d3c3de9de
---
 product/ERP5OOo/tests/testDms.py | 64 +++++++++++++++++---------------
 1 file changed, 35 insertions(+), 29 deletions(-)

diff --git a/product/ERP5OOo/tests/testDms.py b/product/ERP5OOo/tests/testDms.py
index ac901aebae..cae699fb98 100644
--- a/product/ERP5OOo/tests/testDms.py
+++ b/product/ERP5OOo/tests/testDms.py
@@ -842,52 +842,58 @@ class TestDocument(ERP5TypeTestCase, ZopeTestCase.Functional):
                       document.Base_showFoundText())
 
   def test_Base_createNewFile(self):
-    context = self.portal.person_module.newContent(portal_type='Person')
-    ret = context.Base_createNewFile(portal_type=None,
-                               title=None,
-                               reference=None,
-                               short_title=None,
-                               language=None,
-                               version=None,
-                               description=None,
-                               file=makeFileUpload('TEST-en-002.odt'))
-    self.assertTrue(ret.endswith(
-      '?portal_status_message=Text%20created%20successfully.'), ret)
-    transaction.commit()
-    self.tic()
-    document_list = context.getFollowUpRelatedValueList()
+    """
+      Test contributing a file and attaching it to context.
+    """
+    person = self.portal.person_module.newContent(portal_type='Person')
+    contributed_document = person.Base_contribute(
+                                     portal_type=None,
+                                     title=None,
+                                     reference=None,
+                                     short_title=None,
+                                     language=None,
+                                     version=None,
+                                     description=None,
+                                     attach_document_to_context=True,
+                                     file=makeFileUpload('TEST-en-002.odt'))
+    self.assertEquals('Text', contributed_document.getPortalType())
+    self.stepTic()
+    document_list = person.getFollowUpRelatedValueList()
     self.assertEquals(1, len(document_list))
     document = document_list[0]
     self.assertEquals('converted', document.getExternalProcessingState())
     self.assertEquals('Text', document.getPortalType())
     self.assertEquals('title', document.getTitle())
+    self.assertEquals(contributed_document, document)
 
   def test_Base_createNewFile_empty(self):
-    context = self.portal.person_module.newContent(portal_type='Person')
+     """
+      Test contributing an empty file and attaching it to context.
+    """
+    person = self.portal.person_module.newContent(portal_type='Person')
     empty_file_upload = ZPublisher.HTTPRequest.FileUpload(FieldStorage(
                             fp=StringIO.StringIO(),
                             environ=dict(REQUEST_METHOD='PUT'),
                             headers={"content-disposition":
                               "attachment; filename=empty;"}))
 
-    ret = context.Base_createNewFile(portal_type=None,
-                               title=None,
-                               reference=None,
-                               short_title=None,
-                               language=None,
-                               version=None,
-                               description=None,
-                               file=empty_file_upload)
-
-    self.assertTrue(ret.endswith(
-      '?portal_status_message=File%20created%20successfully.'), ret)
-    transaction.commit()
-    self.tic()
-    document_list = context.getFollowUpRelatedValueList()
+    contributed_document = person.Base_contribute(
+                                    portal_type=None,
+                                    title=None,
+                                    reference=None,
+                                    short_title=None,
+                                    language=None,
+                                    version=None,
+                                    description=None,
+                                    attach_document_to_context=True,
+                                    file=empty_file_upload)
+    self.stepTic()
+    document_list = person.getFollowUpRelatedValueList()
     self.assertEquals(1, len(document_list))
     document = document_list[0]
     self.assertEquals('empty', document.getExternalProcessingState())
     self.assertEquals('File', document.getPortalType())
+    self.assertEquals(contributed_document, document)
 
   def test_HTML_to_ODT_conversion_keep_enconding(self):
     """This test perform an PDF conversion of HTML content
-- 
2.30.9