@@ -13,21 +13,28 @@ def test_hec__send_trivially(mock_send):
13
13
hec .sendEvent ({'test' : 'test-tacular' })
14
14
assert json .loads (mock_send .call_args .args [0 ].dat )['test' ] == 'test-tacular'
15
15
16
- @mock .patch .object (HEC , '_send' ) # just in case, not actually used
16
+ @mock .patch .object (HEC , '_send' )
17
17
def test_queue_things_with_compression (mock_send , __opts__ , __salt__ ):
18
18
hec = HEC ('token' , 'index' , 'server' ,
19
19
disk_queue = TEST_DQ_DIR , disk_queue_size = 1000 ,
20
20
disk_queue_compression = 9 )
21
21
22
+ hec .queue .clear ()
23
+
22
24
results_of_side_effect = list ()
23
- def side_effect (x ):
24
- results_of_side_effect .append (x )
25
+ def side_effect (x , ** kw ):
26
+ if 'xxx-mytest-xxx' in x :
27
+ results_of_side_effect .append (x )
25
28
mock_send .side_effect = side_effect
26
29
27
30
gz = list ()
28
- for i in range (100 ):
29
- dat = {f'event{ i } ' : f'test{ i } ' }
31
+ for i in range (1 ):
32
+ dat = {f'event{ i } ' : f'-xxx-mytest-xxx- test{ i } ' }
30
33
hec .queueEvent (dat )
31
34
gz .append ( json .dumps (dat ) )
32
35
hec .flushQueue ()
33
- assert ' ' .join (results_of_side_effect ) == ' ' .join (gz )
36
+
37
+ cat_rez = ' ' .join (results_of_side_effect )
38
+ cat_gz = ' ' .join (gz )
39
+
40
+ assert cat_rez == cat_gz
0 commit comments