From 28401ac0b3bdd3c36126b9a135f0354e4ee7c596 Mon Sep 17 00:00:00 2001
From: Shawn <44221603+smarshall-spitzbart@users.noreply.github.com>
Date: Wed, 13 Sep 2023 15:14:48 -0700
Subject: [PATCH] Update steps_downtime.go

---
 tests/e2e/steps_downtime.go | 28 +++++++++++++++++++++++-----
 1 file changed, 23 insertions(+), 5 deletions(-)

diff --git a/tests/e2e/steps_downtime.go b/tests/e2e/steps_downtime.go
index 30e833c793..7820ea1a51 100644
--- a/tests/e2e/steps_downtime.go
+++ b/tests/e2e/steps_downtime.go
@@ -309,15 +309,33 @@ func stepsThrottledDowntime(consumerName string) []Step {
 				chainID("provi"): ChainState{
 					ValPowers: &map[validatorID]uint{
 						validatorID("alice"): 511,
-						validatorID("bob"):   0, // bob is jailed
+						validatorID("bob"):   0, // bob is jailed on provider
 						validatorID("carol"): 500,
 					},
 				},
 				chainID(consumerName): ChainState{
-					// VSC packet applying jailing is not yet relayed to consumer
 					ValPowers: &map[validatorID]uint{
 						validatorID("alice"): 511,
-						validatorID("bob"):   500,
+						// VSC packet applying jailing may or may not be relayed to consumer
+						// depending on hermes vs gorelayer behavior. Therefore skip asserting bob's consumer power
+						validatorID("carol"): 500,
+					},
+				},
+			},
+		},
+		{
+			// After relaying one more time, vsc packet applying jailing should be seen on consumer
+			action: relayPacketsAction{
+				chainA:  chainID("provi"),
+				chainB:  chainID(consumerName),
+				port:    "provider",
+				channel: 0,
+			},
+			state: State{
+				chainID(consumerName): ChainState{
+					ValPowers: &map[validatorID]uint{
+						validatorID("alice"): 511,
+						validatorID("bob"):   0,
 						validatorID("carol"): 500,
 					},
 				},
@@ -340,7 +358,7 @@ func stepsThrottledDowntime(consumerName string) []Step {
 				chainID(consumerName): ChainState{
 					ValPowers: &map[validatorID]uint{
 						validatorID("alice"): 511,
-						validatorID("bob"):   500, // VSC packet applying bob jailing is not yet relayed to consumer
+						validatorID("bob"):   0,
 						validatorID("carol"): 500,
 					},
 				},
@@ -364,7 +382,7 @@ func stepsThrottledDowntime(consumerName string) []Step {
 				chainID(consumerName): ChainState{
 					ValPowers: &map[validatorID]uint{
 						validatorID("alice"): 511,
-						validatorID("bob"):   0, // VSC packet applying bob jailing is also relayed and recv by consumer
+						validatorID("bob"):   0,
 						validatorID("carol"): 500,
 					},
 				},