traffic-replay: add extra check

Make sure --average-groups-per-user is not more than --number-of-users

Signed-off-by: Joe Guo <joeg@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
Reviewed-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>

diff --git a/script/traffic_replay b/script/traffic_replay
index 764462e5be3c10d99c404aae31d2bea996de7865..f414015a87210b7495a8d96c41ac9a408ea3fbe6 100755 (executable)
--- a/script/traffic_replay
+++ b/script/traffic_replay
@@ -195,6 +195,12 @@ def main():
                    "--number-of-groups"))
         sys.exit(1)
 
+    if opts.number_of_groups and opts.average_groups_per_user:
+        if opts.number_of_groups < opts.average_groups_per_user:
+            print_err(("--average-groups-per-user can not be more than "
+                       "--number-of-groups"))
+            sys.exit(1)
+
     if not opts.number_of_groups and opts.group_memberships:
         print_err("--group-memberships requires --number-of-groups")
         sys.exit(1)