Тесты

Author: lashnev

platron/request/request_builders/recurring_set_schedule_buider.py

@@ -1,5 +1,6 @@
 from platron.request.request_builders.request_builder import RequestBuilder
 from platron.sdk_exception import SdkException
+import re
 
 
 class RecurringSetScheduleBuilder(RequestBuilder):
@@ -38,6 +39,12 @@ def add_template(self, start_date, interval, period, max_periods=None):
         if max_periods != None:
             self.pg_template.update({'pg_max_periods': max_periods})
 
+    def after_xml_created(self, xml):
+        without_middle_tags = re.sub(r'</n\d+><n\d+>', '</pg_dates><pg_dates>', xml)
+        without_start_tags = re.sub(r'<n\d+>', '', without_middle_tags)
+        without_end_tags = re.sub(r'</n\d+>', '', without_start_tags)
+        return without_end_tags
+
     @staticmethod
     def __get_intervals():
         return {'day': True, 'week': True, 'month': True}

tests/integration/receipt_test.py

@@ -10,15 +10,20 @@
 class ReceiptTest(BaseIntegrationTest):
 
     def test_create_transaction_chain(self):
-        builder = InitPaymentBuilder('10.00', 'test')
+        builder = InitPaymentBuilder('50.00', 'test')
         client = PostClient(self.get_merchant_id(), self.get_secret_key())
         result = client.request(builder)
         root = fromstring(result)
         pg_payment_id = root.find('pg_payment_id').text
 
         builder = ReceiptBuilder('payment', pg_payment_id)
+        builder.add_additional_payment('credit', '20')
+
         receipt_item1 = Item('test', '10.00', '1')
+        receipt_item1.add_agent('commissionaire', 'test agent', '123456789012', '79050000000')
         receipt_item2 = Item('test2', '20.00', '1')
+        receipt_item2.add_agent('commissionaire', 'test agent', '123456789012', '79050000000')
+
         builder.add_item(receipt_item1)
         builder.add_item(receipt_item2)
 

tests/unit/request/request_builders/recurring_set_schedule_builder_test.py

@@ -7,19 +7,19 @@ class RecurringSetScheduleBuilderTest(unittest.TestCase):
 
     def test_get_params(self):
         builder = RecurringSetScheduleBuilder('12345', '100')
-        builder.add_template('2018-01-01', 'week', '10', '100')
-        builder.add_dates({'1': '2018-10-10', '2': '2019-10-10'})
+        builder.add_template('2018-01-01 00:00:00', 'week', '10', '100')
+        builder.add_dates({'1': '2018-10-10 00:00:00', '2': '2019-10-10 00:00:00'})
 
         params = builder.get_params()
         template = params.get('pg_template')
         dates = params.get('pg_dates')
 
         self.assertEquals('12345', params.get('pg_recurring_profile'))
-        self.assertEquals('2018-01-01', template.get('pg_start_date'))
+        self.assertEquals('2018-01-01 00:00:00', template.get('pg_start_date'))
         self.assertEquals('week', template.get('pg_interval'))
         self.assertEquals('10', template.get('pg_period'))
         self.assertEquals('100', template.get('pg_max_periods'))
-        self.assertEquals('2018-10-10', dates.get('1'));
+        self.assertEquals('2018-10-10 00:00:00', dates.get('1'));
 
         with self.assertRaises(SdkException):
-            builder.add_template('2018-01-01', 'wrong_interval', '10', '100')
+            builder.add_template('2018-01-01 00:00:00', 'wrong_interval', '10', '100')