Function TestOneShot

pkg/acquisition/modules/docker/docker_test.go:556–609

(t *testing.T)

}

func TestOneShot(t *testing.T) {
	ctx := t.Context()

	log.Info("Test 'TestOneShot'")

	tests := []struct {
		dsn            string
		expectedErr    string
		expectedOutput string
		expectedLines  int
		logType        string
	}{
		{
			dsn:            "docker://non_exist_docker",
			expectedErr:    "no container found named: non_exist_docker, can't run one shot acquisition",
			expectedOutput: "",
			expectedLines:  0,
			logType:        "test",
		},
		{
			dsn:            "docker://" + testContainerName,
			expectedErr:    "",
			expectedOutput: "",
			expectedLines:  3,
			logType:        "test",
		},
	}

	for _, ts := range tests {
		t.Run(ts.dsn, func(t *testing.T) {
			subLogger := log.WithField("type", ModuleName)

			dockerClient := &Source{}
			labels := make(map[string]string)
			labels["type"] = ts.logType

			err := dockerClient.ConfigureByDSN(ctx, ts.dsn, labels, subLogger, "")
			require.NoError(t, err)

			dockerClient.Client = &mockDockerCli{}
			out := make(chan pipeline.Event, 100)

			tomb := tomb.Tomb{}

			err = dockerClient.OneShotAcquisition(ctx, out, &tomb)
			cstest.AssertErrorContains(t, err, ts.expectedErr)

			// else we do the check before actualLines is incremented ...
			if ts.expectedLines != 0 {
				assert.Len(t, out, ts.expectedLines)
			}
		})
	}
}

func TestParseLabels(t *testing.T) {
	tests := []struct {