API: Fix oui_v3

maxmouchet · Dec 22, 2019 · b39b63b · b39b63b
1 parent 4110028
commit b39b63b
Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/locomotive/api/oui_v3/client.py b/locomotive/api/oui_v3/client.py
@@ -127,4 +127,4 @@ def __to_segment(self, obj: dict) -> Segment:
         )
 
     def __to_proposal(self, obj: dict) -> Proposal:
-        return Proposal(flexibility_level=obj["flexibilityLevel"], price=obj["price"])
+        return Proposal(flexibility_level=obj["info"]["miInfo"]["proposalType"], price=obj["price"]['value'])
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -16,7 +16,7 @@ def test_helper():
 
 def test_search_with_origin_and_destination():
     runner = CliRunner()
-    result = runner.invoke(cli, ["search", "FRBES", "FRPAR"])
+    result = runner.invoke(cli, ["search", "FRPAR", "FRLYS"])
 
     assert "km) on" in result.output
     assert result.exit_code == 0
@@ -50,15 +50,15 @@ def test_search_with_date_provided():
 
 def test_search_with_class_provided():
     runner = CliRunner()
-    result = runner.invoke(cli, ["search", "FRBES", "FRPAR", "--class", "first"])
+    result = runner.invoke(cli, ["search", "FRPAR", "FRLYS", "--class", "first"])
 
     assert "km) on" in result.output
     assert result.exit_code == 0
 
 
 def test_search_with_formatter_provided():
     runner = CliRunner()
-    result = runner.invoke(cli, ["search", "FRBES", "FRPAR", "--format", "json"])
+    result = runner.invoke(cli, ["search", "FRPAR", "FRLYS", "--format", "json"])
 
     assert "km) on" in result.output
     assert result.exit_code == 0