From 2cdb9c3f1deaf43b6ccb7442b3b3551e760026c9 Mon Sep 17 00:00:00 2001 From: Sumeet Kumar Rai Date: Wed, 13 Dec 2023 20:34:13 +0530 Subject: [PATCH] recordPlugin method mock method call not needed now --- agent/agent_test.go | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/agent/agent_test.go b/agent/agent_test.go index 36d76af5..42e0163d 100644 --- a/agent/agent_test.go +++ b/agent/agent_test.go @@ -483,7 +483,7 @@ func TestAgentRun(t *testing.T) { monitor := newMockMonitor() monitor.On("RecordRun", mockCtx, mock.AnythingOfType("agent.Run")).Once() - monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")) + monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")).Maybe() defer monitor.AssertExpectations(t) r := agent.NewAgent(agent.Config{ @@ -533,7 +533,7 @@ func TestAgentRun(t *testing.T) { monitor := newMockMonitor() monitor.On("RecordRun", mockCtx, mock.AnythingOfType("agent.Run")).Once() - monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")) + monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")).Maybe() defer monitor.AssertExpectations(t) r := agent.NewAgent(agent.Config{ @@ -585,7 +585,7 @@ func TestAgentRun(t *testing.T) { monitor := newMockMonitor() monitor.On("RecordRun", mockCtx, mock.AnythingOfType("agent.Run")).Once() - monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")) + monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")).Maybe() defer monitor.AssertExpectations(t) r := agent.NewAgent(agent.Config{ @@ -649,7 +649,7 @@ func TestAgentRun(t *testing.T) { monitor := newMockMonitor() monitor.On("RecordRun", mockCtx, mock.AnythingOfType("agent.Run")).Once() - monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")) + monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")).Maybe() defer monitor.AssertExpectations(t) r := agent.NewAgent(agent.Config{ @@ -706,7 +706,7 @@ func TestAgentRun(t *testing.T) { monitor := newMockMonitor() monitor.On("RecordRun", mockCtx, mock.AnythingOfType("agent.Run")).Once() - monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")) + monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")).Maybe() defer monitor.AssertExpectations(t) r := agent.NewAgent(agent.Config{ @@ -761,7 +761,7 @@ func TestAgentRun(t *testing.T) { monitor := newMockMonitor() monitor.On("RecordRun", mockCtx, mock.AnythingOfType("agent.Run")).Once() - monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")) + monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")).Maybe() defer monitor.AssertExpectations(t) r := agent.NewAgent(agent.Config{ @@ -815,7 +815,7 @@ func TestAgentRun(t *testing.T) { monitor := newMockMonitor() monitor.On("RecordRun", mockCtx, mock.AnythingOfType("agent.Run")).Once() - monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")) + monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")).Maybe() monitor.On("RecordSinkRetryCount", mockCtx, mock.AnythingOfType("agent.PluginInfo")) defer monitor.AssertExpectations(t) @@ -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) @@ -1057,7 +1057,7 @@ func TestAgentRunMultiple(t *testing.T) { monitor := newMockMonitor() monitor.On("RecordRun", mockCtx, mock.AnythingOfType("agent.Run")) - monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")) + monitor.On("RecordPlugin", mockCtx, mock.AnythingOfType("agent.PluginInfo")).Maybe() defer monitor.AssertExpectations(t) r := agent.NewAgent(agent.Config{