chore: improve code

jferrl · Apr 27, 2024 · dbfb250 · dbfb250
1 parent 6df2363
commit dbfb250
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/yo.go b/yo.go
@@ -24,9 +24,9 @@ type Yoable[Y any] interface {
 type Opt[T Yoable[T]] func(*Model[T])
 
 // WithSpannerClientOption returns an Opt that sets the spanner client.
-func WithSpannerClientOption[T Yoable[T]](c spanner.Client) Opt[T] {
+func WithSpannerClientOption[T Yoable[T]](c *spanner.Client) Opt[T] {
 	return func(m *Model[T]) {
-		m.Client = c
+		m.Client = *c
 	}
 }
 

diff --git a/yo_test.go b/yo_test.go
@@ -179,7 +179,7 @@ func TestModel_Apply(t *testing.T) {
 			defer cleanup()
 
 			m := NewModel(tt.fields.model,
-				WithSpannerClientOption[*user.User](*spannerClient),
+				WithSpannerClientOption[*user.User](spannerClient),
 			)
 
 			before, after := tt.mutType.Hooks()