test: fix test expectations

Signed-off-by: Manan Gupta <[email protected]>
vitessio · Nov 7, 2024 · e0cbfc7 · e0cbfc7
1 parent e8279bc
commit e0cbfc7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/go/vt/vtctl/grpcvtctldserver/server_test.go b/go/vt/vtctl/grpcvtctldserver/server_test.go
@@ -5570,7 +5570,7 @@ func TestGetTransactionInfo(t *testing.T) {
 					"80-": {
 						Dtid: "bb",
 						Participants: []*querypb.Target{
-							{Keyspace: ks, Shard: "-80"},
+							{Keyspace: ks, Shard: "-80", TabletType: topodatapb.TabletType_PRIMARY},
 						},
 					},
 				},
@@ -5587,7 +5587,8 @@ func TestGetTransactionInfo(t *testing.T) {
 				Metadata: &querypb.TransactionMetadata{
 					Dtid: "bb",
 					Participants: []*querypb.Target{
-						{Keyspace: ks, Shard: "-80"},
+						{Keyspace: ks, Shard: "-80", TabletType: topodatapb.TabletType_PRIMARY},
+						{Keyspace: ks, Shard: "80-", TabletType: topodatapb.TabletType_PRIMARY},
 					},
 				},
 				ShardStates: []*vtctldatapb.ShardTransactionState{