generate keys on startup for e2es (#858)

* generate keys on startup for e2es * typos
worldcoin · Dec 27, 2024 · 4d69c9d · 4d69c9d
1 parent 68d9654
commit 4d69c9d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/deploy/e2e/iris-mpc-0.yaml.tpl b/deploy/e2e/iris-mpc-0.yaml.tpl
@@ -40,7 +40,6 @@ iris-mpc-0:
     seccompProfile:
       type: RuntimeDefault
 
-
   resources:
     limits:
       cpu: 31
@@ -190,7 +189,7 @@ iris-mpc-0:
 
   initContainer:
     enabled: true
-    image: "amazon/aws-cli:2.17.62"
+    image: "ghcr.io/worldcoin/iris-mpc:146c2cae43dbeb586144d9d37d152a6b2bfacdd4" # no-cuda image
     name: "iris-mpc-0-copy-cuda-libs"
     env:
       - name: PARTY_ID
@@ -203,6 +202,10 @@ iris-mpc-0:
       name: "iris-mpc-0-init"
       init.sh: |
         #!/usr/bin/env bash
+        apt-update && apt install -y awscli
         cd /libs
         aws s3 cp s3://wf-smpcv2-stage-libs/libcublas.so.12.2.5.6 .
         aws s3 cp s3://wf-smpcv2-stage-libs/libcublasLt.so.12.2.5.6 .
+
+        key-manager --node-id 0 --env $ENV rotate --public-key-bucket-name wf-$ENV-stage-public-keys
+
diff --git a/deploy/e2e/iris-mpc-1.yaml.tpl b/deploy/e2e/iris-mpc-1.yaml.tpl
@@ -190,7 +190,7 @@ iris-mpc-1:
 
   initContainer:
     enabled: true
-    image: "amazon/aws-cli:2.17.62"
+    image: "ghcr.io/worldcoin/iris-mpc:146c2cae43dbeb586144d9d37d152a6b2bfacdd4" # no-cuda image
     name: "iris-mpc-1-copy-cuda-libs"
     env:
       - name: PARTY_ID
@@ -203,6 +203,10 @@ iris-mpc-1:
       name: "iris-mpc-1-init"
       init.sh: |
         #!/usr/bin/env bash
+        apt-update && apt install -y awscli
         cd /libs
         aws s3 cp s3://wf-smpcv2-stage-libs/libcublas.so.12.2.5.6 .
         aws s3 cp s3://wf-smpcv2-stage-libs/libcublasLt.so.12.2.5.6 .
+
+        key-manager --node-id 1 --env $ENV rotate --public-key-bucket-name wf-$ENV-stage-public-keys
+
diff --git a/deploy/e2e/iris-mpc-2.yaml.tpl b/deploy/e2e/iris-mpc-2.yaml.tpl
@@ -190,7 +190,7 @@ iris-mpc-2:
 
   initContainer:
     enabled: true
-    image: "amazon/aws-cli:2.17.62"
+    image: "ghcr.io/worldcoin/iris-mpc:146c2cae43dbeb586144d9d37d152a6b2bfacdd4" # no-cuda image
     name: "iris-mpc-2-copy-cuda-libs"
     env:
       - name: PARTY_ID
@@ -203,6 +203,10 @@ iris-mpc-2:
       name: "iris-mpc-2-init"
       init.sh: |
         #!/usr/bin/env bash
+        apt-update && apt install -y awscli
         cd /libs
         aws s3 cp s3://wf-smpcv2-stage-libs/libcublas.so.12.2.5.6 .
         aws s3 cp s3://wf-smpcv2-stage-libs/libcublasLt.so.12.2.5.6 .
+
+        key-manager --node-id 2 --env $ENV rotate --public-key-bucket-name wf-$ENV-stage-public-keys
+