Merge branch 'add_test_with_real_swarmkv' into 'rel'v3.1.39

Add test with real swarmkv

See merge request tango/shaping-engine!101

1 files changed, 13 insertions, 2 deletions

diff --git a/shaping/test/stub.cpp b/shaping/test/stub.cpp
index 4da686e..24b3ac3 100644
--- a/shaping/test/stub.cpp
+++ b/shaping/test/stub.cpp
@@ -28,7 +28,7 @@ struct stub_matched_rules {
 
 struct stub_pkt_queue tx_queue;
 struct stub_matched_rules matched_rules;
-extern struct shaping_profile pf_array[MAX_STUB_PROFILE_NUM];
+struct shaping_profile pf_array[MAX_STUB_PROFILE_NUM];
 
 void stub_set_profile_type(int profile_id, enum shaping_profile_type type)
 {
@@ -36,6 +36,12 @@ void stub_set_profile_type(int profile_id, enum shaping_profile_type type)
     return;
 }
 
+void stub_set_profile_limit_direction(int profile_id, enum shaping_profile_limit_direction limit_direction)
+{
+    pf_array[profile_id].limit_direction = limit_direction;
+    return;
+}
+
 void stub_set_matched_shaping_rules(int rule_num, long long *rule_id, const int *priority, const int *profile_num, int profile_id[][MAX_REF_PROFILE])
 {
     struct shaping_rule *rules;
@@ -70,6 +76,12 @@ void stub_set_shaping_rule_dscp_value(int rule_id, int dscp_value)
     return;
 }
 
+void stub_set_shaping_rule_fair_factor(int rule_id, int fair_factor)
+{
+    matched_rules.rules[rule_id].fair_factor = fair_factor;
+    return;
+}
+
 void stub_clear_matched_shaping_rules()
 {
     memset(&matched_rules, 0, sizeof(struct stub_matched_rules));
@@ -104,7 +116,6 @@ void stub_init()
     TAILQ_INIT(&tx_queue);
     memset(&matched_rules, 0, sizeof(struct stub_matched_rules));
     
-    init_dummy_swarmkv();
     return;
 }