Add warp

1 files changed, 26 insertions, 6 deletions

diff --git a/scenarios/garage-warp b/scenarios/garage-warp
index 0f00580..6b7bbc1 100755
--- a/scenarios/garage-warp
+++ b/scenarios/garage-warp
@@ -1,13 +1,33 @@
 #!/usr/bin/env python3
 from fragments import garage, warp, shared
+import sys
 
+garage_version_flavor = {
+    "garage-v0.7": "v0.7.3",
+    "garage-v0.8": "89b8087ba81c508ba382aa6c9cb6bb3afa6a43c8"
+}
+garage_version = garage_version_flavor["garage-v0.7"]
 
+warp_bench_flavor = {
+    "warp-fast": "mixed --obj.size 5M --objects 200 --duration=1m",
+    "warp-default": "mixed"
+}
+warp_bench = warp_bench_flavor["warp-fast"]
 
-if shared.id() == 1:
-  garage.deploy_coord()
-  warp.on_garage()
-  garage.delete_key()
-  garage.destroy()
-else
 
+for flavor in sys.argv[1:]:
+    if flavor in garage_version_flavor:
+        garage_version = garage_version_flavor[flavor]
+    if flavor in warp_bench_flavor:
+        warp_bench = warp_bench_flavor[flavor]
 
+if shared.id() == 1:
+    garage.deploy_coord(version=garage_version)
+    warp.on_garage(params=warp_bench)
+    garage.delete_key()
+    garage.destroy()
+else:
+    garage.deploy_follow(version=garage_version)
+    garage.sync_on_key_down()
+    garage.destroy()
+shared.log("bye")