@@ -26,7 +26,9 @@ func (k mockKVS) GetLongTerm(ctx context.Context, ephemeral view.Identity) (view
2626func (k mockKVS ) HaveSameBinding (ctx context.Context , this , that view.Identity ) (bool , error ) {
2727 return false , nil
2828}
29- func (k mockKVS ) PutBinding (ctx context.Context , ephemeral , longTerm view.Identity ) error { return nil }
29+ func (k mockKVS ) PutBindings (ctx context.Context , longTerm view.Identity , ephemeral ... view.Identity ) error {
30+ return nil
31+ }
3032
3133type mockExtractor struct {}
3234
@@ -58,7 +60,7 @@ func TestPKIResolveConcurrency(t *testing.T) {
5860func TestGetIdentity (t * testing.T ) {
5961 // setup
6062 bindingStore := & mock.BindingStore {}
61- bindingStore .PutBindingReturns (nil )
63+ bindingStore .PutBindingsReturns (nil )
6264
6365 service , err := endpoint .NewService (bindingStore )
6466 require .NoError (t , err )
@@ -73,7 +75,7 @@ func TestGetIdentity(t *testing.T) {
7375 require .NoError (t , err )
7476 resolvers := service .Resolvers ()
7577 assert .Len (t , resolvers , 1 )
76- assert .Equal (t , 0 , bindingStore .PutBindingCallCount ())
78+ assert .Equal (t , 0 , bindingStore .PutBindingsCallCount ())
7779
7880 _ , err = service .AddResolver (
7981 "alice" ,
@@ -85,7 +87,7 @@ func TestGetIdentity(t *testing.T) {
8587 require .NoError (t , err )
8688 resolvers = service .Resolvers ()
8789 assert .Len (t , resolvers , 1 )
88- assert .Equal (t , 1 , bindingStore .PutBindingCallCount ())
90+ assert .Equal (t , 1 , bindingStore .PutBindingsCallCount ())
8991
9092 err = service .AddPublicKeyExtractor (ext )
9193 require .NoError (t , err )
0 commit comments