fix tests

Author: rajrohanyadav

cmd/nri-flex/nri-flex_test.go

@@ -58,8 +58,9 @@ func TestConfigDir(t *testing.T) {
 	r := fintegration.GetTestRuntime()
 	fintegration.RunFlex(r)
 	expectedSamples := []string{
-		`{"event_type":"flexStatusSample","flex.IntegrationVersion":"Unknown-SNAPSHOT","flex.counter.ConfigsProcessed":3,` +
-			`"flex.counter.EventCount":3,"flex.counter.EventDropCount":0,"flex.counter.MessageSample":2,"flex.counter.commandJsonOutSample":1}`,
+		`{"event_type":"flexStatusSample","flex.IntegrationVersion":"Unknown-SNAPSHOT","flex.counter.ConfigsProcessed":6,` +
+			`"flex.counter.EventCount":6,"flex.counter.EventDropCount":0,"flex.counter.MessageSample":2,` +
+			`"flex.counter.MessageLegacySample":2,"flex.counter.commandJsonOutSample":2}`,
 		`{"error":"true","event_type":"MessageSample","message":"bye","value":20.9}`,
 		`{"error":"false","event_type":"MessageSample","message":"hello","value":100}`,
 		`{"error":"true","event_type":"MessageLegacySample","integration_name": "com.newrelic.nri-flex",` +