Skip to content

Commit 6fe3ba6

Browse files
committed
fix error
1 parent d1e3bb8 commit 6fe3ba6

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

internal/verifier/change_stream.go

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,14 @@ const (
3939
metadataChangeStreamCollectionName = "changeStream"
4040
)
4141

42+
type UnknownEventError struct {
43+
Event *ParsedEvent
44+
}
45+
46+
func (uee UnknownEventError) Error() string {
47+
return fmt.Sprintf("Unknown event type: %#q", uee.Event.OpType)
48+
}
49+
4250
// HandleChangeStreamEvent performs the necessary work for change stream events that occur during
4351
// operation.
4452
func (verifier *Verifier) HandleChangeStreamEvent(ctx context.Context, changeEvent *ParsedEvent) error {
@@ -57,7 +65,7 @@ func (verifier *Verifier) HandleChangeStreamEvent(ctx context.Context, changeEve
5765
case "update":
5866
return verifier.InsertChangeEventRecheckDoc(ctx, changeEvent)
5967
default:
60-
return errors.New(`Not supporting: "` + changeEvent.OpType + `" events`)
68+
return UnknownEventError{Event: changeEvent}
6169
}
6270
}
6371

internal/verifier/migration_verifier_test.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -510,7 +510,9 @@ func (suite *MultiMetaVersionTestSuite) TestFailedVerificationTaskInsertions() {
510510
suite.Require().NoError(err)
511511
event.OpType = "flibbity"
512512
err = verifier.HandleChangeStreamEvent(ctx, &event)
513-
suite.Require().ErrorIs(err, fmt.Errorf(`Not supporting: "flibbity" events`))
513+
badEventErr := UnknownEventError{}
514+
suite.Require().ErrorAs(err, &badEventErr)
515+
suite.Assert().Equal("flibbity", badEventErr.Event.OpType)
514516

515517
verifier.generation++
516518
func() {

0 commit comments

Comments
 (0)