// Copyright (c) HashiCorp, Inc. // SPDX-License-Identifier: BUSL-1.1 package static import ( "context" "testing" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" "github.com/hashicorp/boundary/internal/db" "github.com/hashicorp/boundary/internal/errors" "github.com/hashicorp/boundary/internal/kms" ) func TestRepository_New(t *testing.T) { ctx := context.Background() conn, _ := db.TestSetup(t, "postgres") rw := db.New(conn) wrapper := db.TestWrapper(t) kmsCache := kms.TestKms(t, conn, wrapper) type args struct { r db.Reader w db.Writer kms *kms.Kms opts []Option } tests := []struct { name string args args want *Repository wantIsErr errors.Code }{ { name: "valid", args: args{ r: rw, w: rw, kms: kmsCache, }, want: &Repository{ reader: rw, writer: rw, kms: kmsCache, defaultLimit: db.DefaultLimit, }, }, { name: "valid-with-limit", args: args{ r: rw, w: rw, kms: kmsCache, opts: []Option{WithLimit(5)}, }, want: &Repository{ reader: rw, writer: rw, kms: kmsCache, defaultLimit: 5, }, }, { name: "nil-reader", args: args{ r: nil, w: rw, kms: kmsCache, }, want: nil, wantIsErr: errors.InvalidParameter, }, { name: "nil-writer", args: args{ r: rw, w: nil, kms: kmsCache, }, want: nil, wantIsErr: errors.InvalidParameter, }, { name: "nil-kms", args: args{ r: rw, w: rw, kms: nil, }, want: nil, wantIsErr: errors.InvalidParameter, }, { name: "all-nils", args: args{ r: nil, w: nil, kms: nil, }, want: nil, wantIsErr: errors.InvalidParameter, }, } for _, tt := range tests { tt := tt t.Run(tt.name, func(t *testing.T) { assert, require := assert.New(t), require.New(t) got, err := NewRepository(ctx, tt.args.r, tt.args.w, tt.args.kms, tt.args.opts...) if tt.wantIsErr != 0 { assert.Truef(errors.Match(errors.T(tt.wantIsErr), err), "want err: %q got: %q", tt.wantIsErr, err) assert.Nil(got) return } assert.NoError(err) require.NotNil(got) assert.Equal(tt.want, got) }) } }