test(pubsublite): fix flaky TestAssignerHandlePartitionFailure (#3671)
Checks only the final error to ensure the test is deterministic.
Fixes https://github.com/googleapis/google-cloud-go/issues/3667.
diff --git a/pubsublite/internal/wire/assigner_test.go b/pubsublite/internal/wire/assigner_test.go
index 4e3c15b..4f33a3e 100644
--- a/pubsublite/internal/wire/assigner_test.go
+++ b/pubsublite/internal/wire/assigner_test.go
@@ -199,13 +199,10 @@
wantErr := errors.New("subscriber shutting down")
asn.SetReceiveError(wantErr)
- if gotErr := asn.StartError(); gotErr != nil {
- t.Errorf("Start() got err: (%v)", gotErr)
+ if gotErr := asn.FinalError(); !test.ErrorEqual(gotErr, wantErr) {
+ t.Errorf("Final err: (%v), want: (%v)", gotErr, wantErr)
}
if got, want := asn.NextPartitions(), []int{1, 2}; !testutil.Equal(got, want) {
t.Errorf("Partition assignments: got %v, want %v", got, want)
}
- if gotErr := asn.FinalError(); !test.ErrorEqual(gotErr, wantErr) {
- t.Errorf("Final err: (%v), want: (%v)", gotErr, wantErr)
- }
}