keywords and function for test case 'Можливість створити план закупівлі'

@@ -37,6 +37,25 @@ Resource           resource.robot
   \  Should Match Regexp   ${document_url}   ^https?:\/\/public.docs(?:-sandbox)?\.openprocurement\.org\/get\/([0-9A-Fa-f]{32})   msg=Not a Document Service Upload
   \  Should Match Regexp   ${document_url}   ^https?:\/\/public.docs(?:-sandbox)?\.openprocurement\.org\/get\/([0-9A-Fa-f]{32})   msg=Not a Document Service Upload
 
 
 
 
+Можливість створити план закупівлі
+  ${NUMBER_OF_LOTS}=  Convert To Integer  ${NUMBER_OF_LOTS}
+  ${NUMBER_OF_ITEMS}=  Convert To Integer  ${NUMBER_OF_ITEMS}
+  ${tender_parameters}=  Create Dictionary
+  ...      mode=${MODE}
+  ...      number_of_items=${NUMBER_OF_ITEMS}
+  ...      number_of_lots=${NUMBER_OF_LOTS}
+  ...      tender_meat=${${TENDER_MEAT}}
+  ...      lot_meat=${${LOT_MEAT}}
+  ...      item_meat=${${ITEM_MEAT}}
+  ${DIALOGUE_TYPE}=  Get Variable Value  ${DIALOGUE_TYPE}
+  Run keyword if  '${DIALOGUE_TYPE}' != '${None}'  Set to dictionary  ${tender_parameters}  dialogue_type=${DIALOGUE_TYPE}
+  ${tender_data}=  Підготувати дані для створення тендера  ${tender_parameters}
+  ${adapted_data}=  Адаптувати дані для оголошення тендера  ${tender_data}
+  ${TENDER_UAID}=  Run As  ${tender_owner}  Створити план  ${adapted_data}
+  Set To Dictionary  ${USERS.users['${tender_owner}']}  initial_data=${adapted_data}
+  Set To Dictionary  ${TENDER}  TENDER_UAID=${TENDER_UAID}
+
+
 Можливість знайти тендер по ідентифікатору для усіх користувачів
 Можливість знайти тендер по ідентифікатору для усіх користувачів
   :FOR  ${username}  IN  ${tender_owner}  ${provider}  ${provider1}  ${viewer}
   :FOR  ${username}  IN  ${tender_owner}  ${provider}  ${provider1}  ${viewer}
   \  Можливість знайти тендер по ідентифікатору для користувача ${username}
   \  Можливість знайти тендер по ідентифікатору для користувача ${username}

@@ -17,6 +17,19 @@ Library  openprocurement_client.utils
   [return]  ${tender_id}
   [return]  ${tender_id}
 
 
 
 
+Отримати internal id плану по UAid
+  [Arguments]  ${username}  ${tender_uaid}
+  Log  ${username}
+  Log  ${tender_uaid}
+  Log Many  ${USERS.users['${username}'].id_map}
+  ${status}=  Run Keyword And Return Status  Dictionary Should Contain Key  ${USERS.users['${username}'].id_map}  ${tender_uaid}
+  Run Keyword And Return If  ${status}  Get From Dictionary  ${USERS.users['${username}'].id_map}  ${tender_uaid}
+  Call Method  ${USERS.users['${username}'].client}  get_plans
+  ${tender_id}=  Wait Until Keyword Succeeds  5x  30 sec  get_plan_id_by_uaid  ${tender_uaid}  ${USERS.users['${username}'].client}
+  Set To Dictionary  ${USERS.users['${username}'].id_map}  ${tender_uaid}  ${tender_id}
+  [return]  ${tender_id}
+
+
 Підготувати клієнт для користувача
 Підготувати клієнт для користувача
   [Arguments]  ${username}
   [Arguments]  ${username}
   [Documentation]  Відкрити браузер, створити об’єкти api wrapper і
   [Documentation]  Відкрити браузер, створити об’єкти api wrapper і
@@ -32,7 +45,9 @@ Library  openprocurement_client.utils
 #  Uncomment this line if there is need to process files operations without DS.
 #  Uncomment this line if there is need to process files operations without DS.
 #  ${ds_api_wraper}=  set variable  ${None}
 #  ${ds_api_wraper}=  set variable  ${None}
   ${ds_api_wraper}=  prepare_ds_api_wrapper  ${DS_HOST_URL}  ${auth_ds}
   ${ds_api_wraper}=  prepare_ds_api_wrapper  ${DS_HOST_URL}  ${auth_ds}
-  ${api_wrapper}=  prepare_api_wrapper  ${USERS.users['${username}'].api_key}  ${RESOURCE}  ${API_HOST_URL}  ${API_VERSION}  ${ds_api_wraper}
+  ${api_wrapper}=  Run Keyword If  '${MODE}' == 'planning'
+  ...     prepare_plan_api_wrapper  ${USERS.users['${username}'].api_key}  ${API_HOST_URL}  ${API_VERSION}
+  ...                     ELSE  prepare_api_wrapper  ${USERS.users['${username}'].api_key}  ${RESOURCE}  ${API_HOST_URL}  ${API_VERSION}  ${ds_api_wraper}
   Set To Dictionary  ${USERS.users['${username}']}  client=${api_wrapper}
   Set To Dictionary  ${USERS.users['${username}']}  client=${api_wrapper}
   Set To Dictionary  ${USERS.users['${username}']}  access_token=${EMPTY}
   Set To Dictionary  ${USERS.users['${username}']}  access_token=${EMPTY}
   ${id_map}=  Create Dictionary
   ${id_map}=  Create Dictionary
@@ -117,6 +132,20 @@ Library  openprocurement_client.utils
   [return]  ${tender.data.tenderID}
   [return]  ${tender.data.tenderID}
 
 
 
 
+Створити план
+  [Arguments]  ${username}  ${tender_data}
+  ${tender}=  Call Method  ${USERS.users['${username}'].client}  create_plan  ${tender_data}
+  Log  ${tender}
+  ${access_token}=  Get Variable Value  ${tender.access.token}
+  ${tender}=  Call Method  ${USERS.users['${username}'].client}  patch_plan  ${tender}
+  Log  ${tender}
+  Log  ${\n}${API_HOST_URL}/api/${API_VERSION}/plans/${tender.data.id}${\n}  WARN
+  Set To Dictionary  ${USERS.users['${username}']}   access_token=${access_token}
+  Set To Dictionary  ${USERS.users['${username}']}   tender_data=${tender}
+  Log   ${USERS.users['${username}'].tender_data}
+  [return]  ${tender.data.planID}
+
+
 Пошук тендера по ідентифікатору
 Пошук тендера по ідентифікатору
   [Arguments]  ${username}  ${tender_uaid}  ${save_key}=tender_data
   [Arguments]  ${username}  ${tender_uaid}  ${save_key}=tender_data
   ${internalid}=  openprocurement_client.Отримати internal id по UAid  ${username}  ${tender_uaid}
   ${internalid}=  openprocurement_client.Отримати internal id по UAid  ${username}  ${tender_uaid}

 from openprocurement_client.client import Client, EDRClient
 from openprocurement_client.client import Client, EDRClient
 from openprocurement_client.document_service_client \
 from openprocurement_client.document_service_client \
     import DocumentServiceClient
     import DocumentServiceClient
+from openprocurement_client.plan import PlansClient
 from openprocurement_client.exceptions import IdNotFound
 from openprocurement_client.exceptions import IdNotFound
 from restkit.errors import RequestFailed, BadStatusLine
 from restkit.errors import RequestFailed, BadStatusLine
 from retrying import retry
 from retrying import retry
@@ -106,3 +107,13 @@ def download_file_from_url(url, path_to_save_file):
     f.write(urllib.urlopen(url).read())
     f.write(urllib.urlopen(url).read())
     f.close()
     f.close()
     return os.path.basename(f.name)
     return os.path.basename(f.name)
+
+
+class StableClient_plan(PlansClient):
+    @retry(stop_max_attempt_number=100, wait_random_min=500, wait_random_max=4000, retry_on_exception=retry_if_request_failed)
+    def request(self, *args, **kwargs):
+        return super(StableClient_plan, self).request(*args, **kwargs)
+
+
+def prepare_plan_api_wrapper(key, host_url, api_version):
+    return StableClient_plan(key, host_url, api_version)