Browse Source

service: added attr 'location'

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

+ 4 - 0
finoex/__init__.py

@@ -362,6 +362,7 @@ class Service(Item):
 
     def __init__(self,
                  duration=None,
+                 location=None,
                  period=None,
                  state=None,
                  **kwargs
@@ -371,6 +372,9 @@ class Service(Item):
         if duration:
             assert isinstance(duration, ioex.datetimeex.Duration)
             self.duration = duration
+        if location:
+            assert isinstance(location, str)
+            self.location = location
         if period:
             assert isinstance(period, ioex.datetimeex.Period)
             self.period = period

+ 4 - 1
tests/test_yaml.py

@@ -37,9 +37,10 @@ def get_item_b():
 
 def get_service_a():
     return finoex.Service(
+        duration = ioex.datetimeex.Duration(years = 2),
+        location = 'vienna',
         name = u'service',
         price_brutto = finoex.Sum(1.0, u'EUR'),
-        duration = ioex.datetimeex.Duration(years = 2),
         state = 'success',
         )
 
@@ -252,6 +253,7 @@ website_url: http://campaign.com
     [get_service_a(), u"""!service
 duration: !duration
   years: 2
+location: vienna
 name: service
 price_brutto: !sum '1.0 EUR'
 state: success
@@ -408,6 +410,7 @@ website_url: http://campaign.com
     [get_service_a(), u"""!service
 duration: !duration
   years: 2
+location: vienna
 name: service
 price_brutto: !sum '1.0 EUR'
 state: success