Merge pull request #4318 from hove-io/expose_use_predicted_traffic_in_tyr

expose use_predicted_traffic in tyr

Author: xlqian

source/tyr/tests/integration/instance_test.py

@@ -180,6 +180,7 @@ def test_update_instances(create_instance):
         "language": "es-ES",
         "co2_emission_car_value": 4242.0,
         "co2_emission_car_unit": "UNITY",
+        "use_predicted_traffic": True,
     }
     resp = api_get('/v0/instances/{}'.format(create_instance))
     assert resp[0]['access_points'] is False
@@ -233,6 +234,7 @@ def test_update_instances(create_instance):
     assert resp['language'] == 'es-ES'
     assert resp['co2_emission_car_value'] == 4242.0
     assert resp['co2_emission_car_unit'] == 'UNITY'
+    assert resp['use_predicted_traffic'] is True
 
 
 def test_update_instances_is_free(create_instance):

source/tyr/tyr/fields.py

@@ -247,6 +247,7 @@ def format(self, value):
     'additional_parameters': fields.Raw,
     'co2_emission_car_value': fields.Raw,
     'co2_emission_car_unit': fields.Raw,
+    'use_predicted_traffic': fields.Boolean,
 }
 
 api_fields = {'id': fields.Raw, 'name': fields.Raw}

source/tyr/tyr/resources.py

@@ -997,6 +997,14 @@ def put(self, version=0, id=None, name=None):
             default=instance.co2_emission_car_unit,
         )
 
+        parser.add_argument(
+            'use_predicted_traffic',
+            type=inputs.boolean,
+            help='whether or not use predicted traffic for asgard',
+            location=('json', 'values'),
+            default=instance.use_predicted_traffic,
+        )
+
         args = parser.parse_args()
 
         try:
@@ -1098,6 +1106,7 @@ def map_args_to_instance(attr_name):
                         'additional_parameters',
                         'co2_emission_car_value',
                         'co2_emission_car_unit',
+                        'use_predicted_traffic',
                     ],
                 ),
                 maxlen=0,