Remake document content checking

Add `get_document_by_id` and `get_file_basename` functions, because they
are needed in `Отримати документ` keyword.
Remove `get_file_contents` function call.
Remake `Отримати документ` keyword: now it returns only name of
downloaded by broker file in lieu of file content and path. Moreover,
`${url}` arguments is replaced with `${doc_id}` so as to give to broker
better data about document to download.

@@ -79,12 +79,10 @@ Resource           resource.robot
   Звірити поле тендера  ${username}  ${TENDER['TENDER_UAID']}  ${USERS.users['${tender_owner}'].initial_data}  ${field}
   Звірити поле тендера  ${username}  ${TENDER['TENDER_UAID']}  ${USERS.users['${tender_owner}'].initial_data}  ${field}
 
 
 
 
-Звірити відображення вмісту документації до тендера для користувача ${username}
-  ${file_content_loaded}  ${file_name_loaded}=  Run as  ${viewer}  Отримати документ  ${TENDER['TENDER_UAID']}  ${USERS.users['${username}'].tender_data.data.documents[0].url}
-  ${file_name_uploaded}=  Set variable  ${USERS.users['${tender_owner}'].documents.filepath}
-  ${document_content_uploaded}=  get_file_contents  ${file_name_uploaded}
-  Порівняти об'єкти  ${file_content_loaded}  ${document_content_uploaded}
-  Порівняти об'єкти  ${file_name_loaded}  ${file_name_uploaded}
+Звірити відображення вмісту документа ${doc_id} з ${left} для користувача ${username}
+  ${file_name}=  Run as  ${username}  Отримати документ  ${TENDER['TENDER_UAID']}  ${doc_id}
+  ${right}=  Get File  ${OUTPUT_DIR}${/}${file_name}
+  Порівняти об'єкти  ${left}  ${right}
 
 
 
 
 Звірити відображення дати ${date} тендера для усіх користувачів
 Звірити відображення дати ${date} тендера для усіх користувачів
@@ -262,9 +260,8 @@ Resource           resource.robot
 Звірити відображення вмісту документації до всіх лотів для користувача ${username}
 Звірити відображення вмісту документації до всіх лотів для користувача ${username}
   ${number_of_lots}=  Get Length  ${USERS.users['${tender_owner}'].initial_data.data.lots}
   ${number_of_lots}=  Get Length  ${USERS.users['${tender_owner}'].initial_data.data.lots}
   :FOR  ${lot_index}  IN RANGE  ${number_of_lots}
   :FOR  ${lot_index}  IN RANGE  ${number_of_lots}
-  \  ${lot_index}=  Convert to integer  ${lot_index}
-  \  ${doc_index}=  get_document_index_by_id  ${USERS.users['${username}'].tender_data.data.documents}  ${USERS.users['${tender_owner}'].lots_documents[${lot_index}]}
-  \  Звірити відображення вмісту ${doc_index} документа до ${lot_index} лоту для користувача ${username}
+  \  ${doc_content}=  Get File  ${USERS.users['${tender_owner}'].lots_documents[${lot_index}].filepath}
+  \  Звірити відображення вмісту документа ${USERS.users['${tender_owner}'].lots_documents[${lot_index}].doc_id} з ${doc_content} для користувача ${username}
 
 
 
 
 Звірити відображення поля ${field} у новоствореному лоті для усіх користувачів
 Звірити відображення поля ${field} у новоствореному лоті для усіх користувачів

@@ -50,11 +50,11 @@ Library  openprocurement_client_helper.py
 
 
 
 
 Отримати документ
 Отримати документ
-  [Arguments]  ${username}  ${tender_uaid}  ${url}
+  [Arguments]  ${username}  ${tender_uaid}  ${doc_id}
   ${tender}=  openprocurement_client.Пошук тендера по ідентифікатору  ${username}  ${tender_uaid}
   ${tender}=  openprocurement_client.Пошук тендера по ідентифікатору  ${username}  ${tender_uaid}
-  ${token}=  Get Variable Value  ${USERS.users['${username}'].access_token}
-  ${contents}  ${filename}=  Call Method  ${USERS.users['${username}'].client}  get_file   ${tender}   ${url}   ${token}
-  [return]   ${contents}  ${filename}
+  ${document}=  get_document_by_id  ${tender.data}  ${doc_id}
+  ${filename}=  download_file_from_url  ${document.url}  ${OUTPUT_DIR}${/}${document.title}
+  [return]  ${filename}
 
 
 
 
 Отримати посилання на аукціон для глядача
 Отримати посилання на аукціон для глядача

@@ -3,6 +3,8 @@ from openprocurement_client.utils import get_tender_id_by_uaid
 from openprocurement_client.exceptions import IdNotFound
 from openprocurement_client.exceptions import IdNotFound
 from restkit.errors import RequestFailed
 from restkit.errors import RequestFailed
 from retrying import retry
 from retrying import retry
+import os
+import urllib
 
 
 
 
 def retry_if_request_failed(exception):
 def retry_if_request_failed(exception):
@@ -34,3 +36,25 @@ def get_complaint_internal_id(tender, complaintID):
     except AttributeError:
     except AttributeError:
         pass
         pass
     raise IdNotFound
     raise IdNotFound
+
+
+def get_document_by_id(data, doc_id):
+    for document in data.get('documents', []):
+        if doc_id in document.get('title', ''):
+            return document
+    for complaint in data.get('complaints', []):
+        for document in complaint.get('documents', []):
+            if doc_id in document.get('title', ''):
+                return document
+    for award in data.get('awards', []):
+        for document in award.get('documents', []):
+            if doc_id in document.get('title', ''):
+                return document
+    raise Exception('Document with id {} not found'.format(doc_id))
+
+
+def download_file_from_url(url, path_to_save_file):
+    f = open(path_to_save_file, 'wb')
+    f.write(urllib.urlopen(url).read())
+    f.close()
+    return os.path.basename(f.name)

@@ -457,9 +457,7 @@ ${item_meat}        ${True}
   ...      viewer
   ...      viewer
   ...      ${USERS.users['${viewer}'].broker}
   ...      ${USERS.users['${viewer}'].broker}
   ...      add_tender_doc  level2
   ...      add_tender_doc  level2
-  ${doc_index}=  get_document_index_by_id  ${USERS.users['${viewer}'].tender_data.data.documents}  ${USERS.users['${tender_owner}'].documents.filepath}
-  Отримати дані із тендера  ${tender_owner}  ${TENDER['TENDER_UAID']}  documents[${doc_index}].url
-  Звірити відображення вмісту документації до тендера для користувача ${viewer}
+  Звірити відображення вмісту документа ${USERS.users['${tender_owner}'].tender_document.doc_id} з ${USERS.users['${tender_owner}'].tender_document.doc_content} для користувача ${viewer}
 
 
 
 
 Відображення вмісту документації до всіх лотів
 Відображення вмісту документації до всіх лотів
@@ -467,7 +465,6 @@ ${item_meat}        ${True}
   ...      viewer
   ...      viewer
   ...      ${USERS.users['${viewer}'].broker}
   ...      ${USERS.users['${viewer}'].broker}
   ...      add_lot_doc  level2
   ...      add_lot_doc  level2
-  Отримати посилання на документацію до всіх лотів для користувача ${viewer}
   Звірити відображення вмісту документації до всіх лотів для користувача ${viewer}
   Звірити відображення вмісту документації до всіх лотів для користувача ${viewer}
 
 
 
 
@@ -862,6 +859,7 @@ ${item_meat}        ${True}
   [Teardown]  Оновити LAST_MODIFICATION_DATE
   [Teardown]  Оновити LAST_MODIFICATION_DATE
   Можливість створити вимогу про виправлення умов закупівлі із документацією
   Можливість створити вимогу про виправлення умов закупівлі із документацією
 
 
+
 Відображення опису вимоги про виправлення умов закупівлі
 Відображення опису вимоги про виправлення умов закупівлі
   [Tags]  ${USERS.users['${viewer}'].broker}: Відображення оскарження
   [Tags]  ${USERS.users['${viewer}'].broker}: Відображення оскарження
   ...  viewer
   ...  viewer