diff --git a/client/consul/consul_testing.go b/client/consul/consul_testing.go index a7997a3eb..f161b6f08 100644 --- a/client/consul/consul_testing.go +++ b/client/consul/consul_testing.go @@ -31,7 +31,7 @@ func NewMockConsulOp(op, allocID, task string) MockConsulOp { // MockConsulServiceClient implements the ConsulServiceAPI interface to record // and log task registration/deregistration. type MockConsulServiceClient struct { - Ops []MockConsulOp + ops []MockConsulOp mu sync.Mutex logger log.Logger @@ -85,3 +85,9 @@ func (m *MockConsulServiceClient) AllocRegistrations(allocID string) (*consul.Al return nil, nil } + +func (m *MockConsulServiceClient) GetOps() []MockConsulOp { + m.mu.Lock() + defer m.mu.Unlock() + return m.ops +}