chore: debug

rudderlabs · Oct 22, 2024 · 052423f · 052423f
1 parent 28fed5a
commit 052423f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/processor/processor_test.go b/processor/processor_test.go
@@ -4,6 +4,7 @@ import (
  "context"
  "encoding/json"
  "fmt"
+ dedupTypes "github.com/rudderlabs/rudder-server/services/dedup/types"
  "reflect"
  "slices"
  "sort"
@@ -2644,7 +2645,10 @@ var _ = Describe("Processor", Ordered, func() {
 
  callUnprocessed := c.mockGatewayJobsDB.EXPECT().GetUnprocessed(gomock.Any(), gomock.Any()).Return(jobsdb.JobsResult{Jobs: unprocessedJobsList}, nil).Times(1)
  c.MockDedup.EXPECT().Get(gomock.Any()).Return(true, int64(0), nil).After(callUnprocessed).AnyTimes()
- c.MockDedup.EXPECT().GetBatch(gomock.Any()).Return(nil, nil, nil).After(callUnprocessed).AnyTimes()
+ c.MockDedup.EXPECT().GetBatch(gomock.Any()).Return(map[dedupTypes.KeyValue]bool{
+ {Key: "message-some-id", Value: 6, JobID: 1010, WorkspaceID: ""}: true,
+ {Key: "message-some-id", Value: 6, JobID: 2010, WorkspaceID: ""}: true,
+ }, nil, nil).After(callUnprocessed).AnyTimes()
  c.MockDedup.EXPECT().Commit(gomock.Any()).Times(1)
 
  // We expect one transform call to destination A, after callUnprocessed.