|
1
|
|
|
# -*- coding: utf-8 -*- |
|
2
|
|
|
|
|
3
|
|
|
import os |
|
4
|
|
|
import json |
|
5
|
|
|
|
|
6
|
|
|
# import pytest |
|
7
|
|
|
# import mock |
|
8
|
|
|
|
|
9
|
|
|
from route4me.sdk.self_test import MockerResourceWithNetworkClient |
|
10
|
|
|
from .optimizations import Optimizations |
|
11
|
|
|
import route4me.sdk.resources.optimizations as M |
|
12
|
|
|
|
|
13
|
|
|
from ..models import Optimization |
|
14
|
|
|
from ..models import AlgorithmTypeEnum |
|
15
|
|
|
from ..models import OptimizationStateEnum |
|
16
|
|
|
|
|
17
|
|
|
|
|
18
|
|
|
class TestOptimizations: |
|
19
|
|
|
def test_ctor(self): |
|
20
|
|
|
|
|
21
|
|
|
api_key = '11111111111111111111111111111111' |
|
22
|
|
|
|
|
23
|
|
|
ns = Optimizations(api_key=api_key) |
|
24
|
|
|
|
|
25
|
|
|
assert ns is not None |
|
26
|
|
|
|
|
27
|
|
|
|
|
28
|
|
|
class TestOptimizationApi(MockerResourceWithNetworkClient): |
|
29
|
|
|
|
|
30
|
|
|
resource_module = M |
|
31
|
|
|
|
|
32
|
|
|
def test_create(self): |
|
33
|
|
|
|
|
34
|
|
|
with open(os.path.join( |
|
35
|
|
|
# '..', '..', '..', |
|
36
|
|
|
'submodules', 'route4me-api-data-examples', 'Optimizations', |
|
37
|
|
|
'create_response.json' |
|
38
|
|
|
)) as f: |
|
39
|
|
|
sample_response_data = json.load(f) |
|
40
|
|
|
|
|
41
|
|
|
self.set_response(data=sample_response_data) |
|
42
|
|
|
|
|
43
|
|
|
o = Optimization() |
|
44
|
|
|
o.algorithm_type = AlgorithmTypeEnum.TSP |
|
45
|
|
|
|
|
46
|
|
|
r = Optimizations(api_key='test') |
|
47
|
|
|
res = r.create(o) |
|
48
|
|
|
|
|
49
|
|
|
print(self.mock_fluent_request_class.mock_calls) |
|
50
|
|
|
# call(), |
|
51
|
|
|
# call().method('POST'), |
|
52
|
|
|
# call().url('https://www.route4me.com//api.v4/optimization_problem.php'), |
|
53
|
|
|
# call().qs(None), |
|
54
|
|
|
# call().json({'links': {}, 'parameters': {'algorithm_type': 1}}), |
|
55
|
|
|
# call().user_agent('requests/2.18.3 (Linux 4.8.0-53-generic) Route4Me-Python-SDK/0.1.0 CPython/3.5.2'), |
|
56
|
|
|
# call().header('Route4Me-User-Agent', 'requests/2.18.3 (Linux 4.8.0-53-generic) ..'), |
|
57
|
|
|
# call().accept('application/json'), |
|
58
|
|
|
# call().header('Route4Me-Api-Key', 'test'), |
|
59
|
|
|
# call().qs({'format': 'json', 'api_key': 'test'}), |
|
60
|
|
|
# call().send(), |
|
61
|
|
|
# call().send().json() |
|
62
|
|
|
|
|
63
|
|
|
# assertions |
|
64
|
|
|
mock_freq = self.last_request() |
|
65
|
|
|
mock_freq.method.assert_called_with('POST') |
|
66
|
|
|
mock_freq.url.assert_called_with( |
|
67
|
|
|
'https://www.route4me.com/api.v4/optimization_problem.php' |
|
68
|
|
|
) |
|
69
|
|
|
mock_freq.json.assert_called_with(dict(o)) |
|
70
|
|
|
|
|
71
|
|
|
# assertions on response |
|
72
|
|
|
assert isinstance(res, Optimization) |
|
73
|
|
|
assert res.ID == '1EDB78F63556D99336E06A13A34CF139' |
|
74
|
|
|
assert res.name == 'Fri, 17 Jun 2016 08:21:59 +0000 UTC' |
|
75
|
|
|
assert res.algorithm_type == AlgorithmTypeEnum.TSP |
|
76
|
|
|
assert res.state == OptimizationStateEnum.MATRIX_PROCESSING |
|
77
|
|
|
|