From 5c5695394b064e10d0977953ad4bbf520c66cf02 Mon Sep 17 00:00:00 2001 From: Ravi Suhag Date: Thu, 31 Oct 2024 14:16:58 +0530 Subject: [PATCH] test: fix failing agent test --- agent/agent_test.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/agent/agent_test.go b/agent/agent_test.go index 149acba4..3f30a683 100644 --- a/agent/agent_test.go +++ b/agent/agent_test.go @@ -875,7 +875,7 @@ func TestAgentRun(t *testing.T) { monitor := newMockMonitor() monitor.On("RecordRun", utils.OfTypeContext(), mock.AnythingOfType("agent.Run")).Once() - monitor.On("RecordPlugin", utils.OfTypeContext(), mock.AnythingOfType("agent.PluginInfo")) + monitor.On("RecordPlugin", utils.OfTypeContext(), mock.AnythingOfType("agent.PluginInfo")).Maybe() monitor.On("RecordSinkRetryCount", utils.OfTypeContext(), mock.AnythingOfType("agent.PluginInfo")).Maybe() defer monitor.AssertExpectations(t)