17
17
from requests import exceptions as request_exception
18
18
19
19
from optimizely .helpers .enums import OdpRestApiConfig
20
+ from optimizely .odp .odp_event import OdpEvent , OdpEventEncoder
20
21
from optimizely .odp .zaius_rest_api_manager import ZaiusRestApiManager
21
22
from . import base
22
23
@@ -26,10 +27,9 @@ class ZaiusRestApiManagerTest(base.BaseTest):
26
27
user_value = "test-user-value"
27
28
api_key = "test-api-key"
28
29
api_host = "test-host"
29
-
30
30
events = [
31
- { "type" : "t1" , "action" : "a1" , "identifiers" : {"id-key-1" : "id-value-1" }, "data" : {"key-1" : "value1" }} ,
32
- { "type" : "t2" , "action" : "a2" , "identifiers" : {"id-key-2" : "id-value-2" }, "data" : {"key-2" : "value2" }},
31
+ OdpEvent ( 't1' , 'a1' , {"id-key-1" : "id-value-1" }, {"key-1" : "value1" }) ,
32
+ OdpEvent ( 't2' , 'a2' , {"id-key-2" : "id-value-2" }, {"key-2" : "value2" })
33
33
]
34
34
35
35
def test_send_odp_events__valid_request (self ):
@@ -42,7 +42,7 @@ def test_send_odp_events__valid_request(self):
42
42
request_headers = {'content-type' : 'application/json' , 'x-api-key' : self .api_key }
43
43
mock_request_post .assert_called_once_with (url = self .api_host + "/v3/events" ,
44
44
headers = request_headers ,
45
- data = json .dumps (self .events ),
45
+ data = json .dumps (self .events , cls = OdpEventEncoder ),
46
46
timeout = OdpRestApiConfig .REQUEST_TIMEOUT )
47
47
48
48
def test_send_odp_ovents_success (self ):
@@ -58,7 +58,8 @@ def test_send_odp_ovents_success(self):
58
58
self .assertFalse (should_retry )
59
59
60
60
def test_send_odp_events_invalid_json_no_retry (self ):
61
- events = {1 , 2 , 3 } # using a set to trigger JSON-not-serializable error
61
+ """Using a set to trigger JSON-not-serializable error."""
62
+ events = {1 , 2 , 3 }
62
63
63
64
with mock .patch ('requests.post' ) as mock_request_post , \
64
65
mock .patch ('optimizely.logger' ) as mock_logger :
0 commit comments