Add additional doc data functionality for update_document

ProzorroUKR · May 21, 2020 · 3a40be1 · 3a40be1
1 parent 0ab343d
commit 3a40be1
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/openprocurement_client/clients.py b/openprocurement_client/clients.py
@@ -371,7 +371,7 @@ def patch_document(self, resource_item_id, document_data, document_id,
 
     @verify_file
     def update_document(self, file_, resource_item_id, document_id, subitem_name=DOCUMENTS, doc_type=None, use_ds_client=True,
-                        doc_registration=True, depth_path=None, access_token=None):
+                        doc_registration=True, depth_path=None, access_token=None, additional_doc_data=None):
         headers = None
         if access_token:
             headers = {'X-Access-Token': access_token}
@@ -385,7 +385,7 @@ def update_document(self, file_, resource_item_id, document_id, subitem_name=DOC
             url = '{}/{}/{}/{}'.format(self.prefix_path, resource_item_id, subitem_name, document_id)
         return self._upload_resource_file(url,
                                           file_=file_, headers=headers, method='PUT', use_ds_client=use_ds_client,
-                                          doc_registration=doc_registration, doc_type=doc_type)
+                                          doc_registration=doc_registration, doc_type=doc_type, additional_doc_data=additional_doc_data)
 
     ###########################################################################
     #                          UPLOAD CLIENT METHODS

diff --git a/openprocurement_client/tests/tests_resources.py b/openprocurement_client/tests/tests_resources.py
@@ -922,6 +922,21 @@ def test_update_tender_document(self):
         )
         self.assertEqual(doc.data.title, file_.name)
 
+    def test_update_tender_document_with_additional_doc_data(self):
+        setup_routing(self.app, routes=["tender_document_update"])
+        file_ = StringIO()
+        file_.name = 'test_document.txt'
+        file_.write("test upload tender document text data")
+        file_.seek(0)
+        doc_data = {'documentOf': 'documents', 'relatedItem': '1'},
+        doc = self.client.update_document(
+            file_, self.limited_tender,
+            TEST_TENDER_KEYS_LIMITED.document_id,
+            doc_type='tenderNotice',
+            additional_doc_data=doc_data
+        )
+        self.assertEqual(doc.data.title, file_.name)
+
     def test_update_bid_document(self):
         setup_routing(self.app, routes=["tender_subpage_document_update"])
         file_ = StringIO()