@@ -45,12 +45,28 @@ def assert_watermarks_committed(watermarks, dump):
45
45
assert_true (dump ['mid' ] == watermarks .mid , 'mid' , dump )
46
46
assert_true (dump ['low' ] == watermarks .low , 'low' , dump )
47
47
48
+ def assert_quota_goal_committed (qgoal , dump ):
49
+ metric_val = {
50
+ 'user_input' : 0 ,
51
+ 'some_mem_psi_us' : 1 ,
52
+ 'node_mem_used_bp' : 2 ,
53
+ 'node_mem_free_bp' : 3 ,
54
+ }
55
+ assert_true (dump ['metric' ] == metric_val [qgoal .metric ], 'metric' , dump )
56
+ assert_true (dump ['target_value' ] == qgoal .target_value , 'target_value' ,
57
+ dump )
58
+ if qgoal .metric == 'user_input' :
59
+ assert_true (dump ['current_value' ] == qgoal .current_value ,
60
+ 'current_value' , dump )
61
+ assert_true (dump ['nid' ] == qgoal .nid , 'nid' , dump )
62
+
48
63
def assert_quota_committed (quota , dump ):
49
64
assert_true (dump ['reset_interval' ] == quota .reset_interval_ms ,
50
65
'reset_interval' , dump )
51
66
assert_true (dump ['ms' ] == quota .ms , 'ms' , dump )
52
67
assert_true (dump ['sz' ] == quota .sz , 'sz' , dump )
53
- # TODO: assert goals are committed
68
+ for idx , qgoal in enumerate (quota .goals ):
69
+ assert_quota_goal_committed (qgoal , dump ['goals' ][idx ])
54
70
assert_true (dump ['weight_sz' ] == quota .weight_sz_permil , 'weight_sz' , dump )
55
71
assert_true (dump ['weight_nr_accesses' ] == quota .weight_nr_accesses_permil ,
56
72
'weight_nr_accesses' , dump )
0 commit comments