Browse Source

shipping: added attr 'tracking_number

Fabian Peter Hammerle 7 years ago
parent
commit
e9bb9b59c9
2 changed files with 7 additions and 0 deletions
  1. 4 0
      finoex/__init__.py
  2. 3 0
      tests/test_yaml.py

+ 4 - 0
finoex/__init__.py

@@ -489,9 +489,13 @@ class Shipping(Transportation):
     yaml_tag = u'!shipping'
 
     def __init__(self,
+                 tracking_number=None,
                  **kwargs
                  ):
         super(Shipping, self).__init__(**kwargs)
+        if tracking_number:
+            assert isinstance(tracking_number, str)
+            self.tracking_number = tracking_number
 
 
 class TaxiRide(Transportation):

+ 3 - 0
tests/test_yaml.py

@@ -151,6 +151,7 @@ def get_shipping():
     return finoex.Shipping(
             price_brutto = get_sum_a(),
             destination_point = u'home',
+            tracking_number = 'LX123456789CN',
             )
 
 def get_taxi_ride():
@@ -402,6 +403,7 @@ valid_until: 2016-07-14 18:50:04+00:00
     [get_shipping(), u"""!shipping
 destination_point: home
 price_brutto: !sum '1.23 EUR'
+tracking_number: LX123456789CN
 """],
     [get_taxi_ride(), u"""!taxi-ride
 arrival_time: 2016-05-02 18:10:00+01:05
@@ -544,6 +546,7 @@ estimated_arrival_time: !period
     [get_shipping(), u"""!shipping
 destination_point: home
 price_brutto: !sum 1.23 EUR
+tracking_number: LX123456789CN
 """],
     [get_order_a(), u"""!order
 customer_id: customer