diff --git a/pkg/driver/driver.go b/pkg/driver/driver.go index 7bd74be0..069af05e 100644 --- a/pkg/driver/driver.go +++ b/pkg/driver/driver.go @@ -22,8 +22,10 @@ func Lookup(name string) (Driver, error) { return &DockerDriver{}, nil case "debug": return &DebugDriver{}, nil - default: + case "command": return &CommandDriver{Name: name}, nil + default: + return nil, fmt.Errorf("unsupported driver: %s", name) } } diff --git a/pkg/driver/driver_test.go b/pkg/driver/driver_test.go index 3c3400f1..0392b609 100644 --- a/pkg/driver/driver_test.go +++ b/pkg/driver/driver_test.go @@ -10,9 +10,16 @@ import ( var _ Driver = &DockerDriver{} var _ Driver = &DebugDriver{} -func TestLookup_ExternalDriver(t *testing.T) { +func TestLookup_UnsupportedDriver(t *testing.T) { d, err := Lookup("no_such_driver") + assert.Nil(t, d) + assert.Error(t, err) +} + +func TestLookup_CommandDriver(t *testing.T) { + d, err := Lookup("command") + assert.NoError(t, err) assert.IsType(t, d, &CommandDriver{}) }