fixing test in listenerClient_test

chrysom/acquire_test.go

@@ -0,0 +1,19 @@
+// SPDX-FileCopyrightText: 2024 Comcast Cable Communications Management, LLC
+// SPDX-License-Identifier: Apache-2.0
+package chrysom
+
+import (
+	"net/http"
+
+	"github.com/stretchr/testify/mock"
+)
+
+type MockAquirer struct {
+	mock.Mock
+}
+
+func (m *MockAquirer) AddAuth(req *http.Request) error {
+	args := m.Called(req)
+
+	return args.Error(0)
+}

chrysom/basicClient_test.go

@@ -30,16 +30,6 @@ var (
 	errFails        = errors.New("fails")
 )
 
-type MockAquirer struct {
-	mock.Mock
-}
-
-func (m *MockAquirer) AddAuth(req *http.Request) error {
-	args := m.Called(req)
-
-	return args.Error(0)
-}
-
 func TestValidateBasicConfig(t *testing.T) {
 	type testCase struct {
 		Description    string

chrysom/listenerClient_test.go

@@ -15,6 +15,7 @@ import (
 
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/require"
 )
 
@@ -43,7 +44,6 @@ func TestListenerStartStopPairsParallel(t *testing.T) {
 	assert.Nil(t, err)
 	defer close()
 
-	t.Run("ParallelGroup", func(t *testing.T) {
 		for i := 0; i < 20; i++ {
 			testNumber := i
 			t.Run(strconv.Itoa(testNumber), func(t *testing.T) {
@@ -62,7 +62,6 @@ func TestListenerStartStopPairsParallel(t *testing.T) {
 				fmt.Printf("%d: Done\n", testNumber)
 			})
 		}
-	})
 
 	require.Equal(stopped, client.observer.state)
 }
@@ -106,14 +105,15 @@ func newStartStopClient(includeListener bool) (*ListenerClient, func(), error) {
 	server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
 		rw.Write(getItemsValidPayload())
 	}))
-
+	acquirer := new(MockAquirer)
+	acquirer.On("AddAuth", mock.Anything).Return(nil)
 	config := ListenerConfig{
 		PullInterval: time.Millisecond * 200,
 	}
 	if includeListener {
 		config.Listener = mockListener
 	}
-	client, err := NewListenerClient(config, mockMeasures, &BasicClient{})
+	client, err := NewListenerClient(config, mockMeasures, &BasicClient{auth: acquirer})
 	if err != nil {
 		return nil, nil, err
 	}