chore: Add tests for runtime WithLogger opt

runtime/docker/opts_test.go

@@ -5,6 +5,7 @@
 package docker
 
 import (
+	"github.com/sirupsen/logrus"
 	"reflect"
 	"testing"
 )
@@ -72,3 +73,47 @@ func TestDocker_ClientOpt_WithHostVolumes(t *testing.T) {
 		}
 	}
 }
+
+func TestDocker_ClientOpt_WithLogger(t *testing.T) {
+	// setup tests
+	tests := []struct {
+		failure bool
+		logger  *logrus.Entry
+	}{
+		{
+			failure: false,
+			logger:  &logrus.Entry{},
+		},
+		{
+			failure: false,
+			logger:  nil,
+		},
+	}
+
+	// run tests
+	for _, test := range tests {
+		_service, err := New(
+			WithLogger(test.logger),
+		)
+
+		if test.failure {
+			if err == nil {
+				t.Errorf("WithLogger should have returned err")
+			}
+
+			continue
+		}
+
+		if err != nil {
+			t.Errorf("WithLogger returned err: %v", err)
+		}
+
+		if test.logger == nil && _service.Logger == nil {
+			t.Errorf("_engine.Logger should not be nil even if nil is passed to WithLogger")
+		}
+
+		if test.logger != nil && !reflect.DeepEqual(_service.Logger, test.logger) {
+			t.Errorf("WithLogger set %v, want %v", _service.Logger, test.logger)
+		}
+	}
+}

runtime/kubernetes/opts_test.go

@@ -5,6 +5,7 @@
 package kubernetes
 
 import (
+	"github.com/sirupsen/logrus"
 	"reflect"
 	"testing"
 )
@@ -161,3 +162,48 @@ func TestKubernetes_ClientOpt_WithPrivilegedImages(t *testing.T) {
 		}
 	}
 }
+
+func TestKubernetes_ClientOpt_WithLogger(t *testing.T) {
+	// setup tests
+	tests := []struct {
+		failure bool
+		logger  *logrus.Entry
+	}{
+		{
+			failure: false,
+			logger:  &logrus.Entry{},
+		},
+		{
+			failure: false,
+			logger:  nil,
+		},
+	}
+
+	// run tests
+	for _, test := range tests {
+		_engine, err := New(
+			WithConfigFile("testdata/config"),
+			WithLogger(test.logger),
+		)
+
+		if test.failure {
+			if err == nil {
+				t.Errorf("WithLogger should have returned err")
+			}
+
+			continue
+		}
+
+		if err != nil {
+			t.Errorf("WithLogger returned err: %v", err)
+		}
+
+		if test.logger == nil && _engine.Logger == nil {
+			t.Errorf("_engine.Logger should not be nil even if nil is passed to WithLogger")
+		}
+
+		if test.logger != nil && !reflect.DeepEqual(_engine.Logger, test.logger) {
+			t.Errorf("WithLogger set %v, want %v", _engine.Logger, test.logger)
+		}
+	}
+}