From 09cd2f2f80920830cc3fd3636a6b9fc93b10f743 Mon Sep 17 00:00:00 2001 From: JP Appel Date: Fri, 18 Jul 2025 15:28:13 -0400 Subject: Add methods for updating existing document entries Additionally optimize various database queries --- pkg/data/put_test.go | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) (limited to 'pkg/data/put_test.go') 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 { -- cgit v1.2.3