diff options
Diffstat (limited to 'pkg/data/put_test.go')
| -rw-r--r-- | pkg/data/put_test.go | 12 |
1 files changed, 4 insertions, 8 deletions
diff --git a/pkg/data/put_test.go b/pkg/data/put_test.go index a97187b..8188b21 100644 --- a/pkg/data/put_test.go +++ b/pkg/data/put_test.go @@ -41,12 +41,8 @@ func TestPut_Insert(t *testing.T) { db := tt.newDb(t) defer db.Close() - p, err := data.NewPut(ctx, db, tt.doc) - if err != nil { - t.Fatalf("could not construct receiver type: %v", err) - } - - gotErr := p.Insert() + p := data.NewPut(db, tt.doc) + gotErr := p.Insert(t.Context()) if !errors.Is(gotErr, tt.wantErr) { t.Fatalf("Unexpected error on Insert():, want %v got %v", tt.wantErr, gotErr) } else if gotErr != nil { @@ -105,12 +101,12 @@ func TestPutMany_Insert(t *testing.T) { for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { db := tt.newDb(t) - p, err := data.NewPutMany(db, tt.documents) + p, err := data.NewPutMany(t.Context(), db, tt.documents) if err != nil { t.Fatalf("could not construct receiver type: %v", err) } - gotErr := p.Insert(t.Context()) + gotErr := p.Insert() if !errors.Is(gotErr, tt.wantErr) { t.Fatalf("Recieved unexpected error, got %v want %v", gotErr, tt.wantErr) } else if err != nil { |