move timeout handling to state machine
1 file changed, 5 insertions(+), 2 deletions(-)
changed files
M state/state_test.go → state/state_test.go
@@ -2,6 +2,7 @@ package state import ( "bytes" + "context" "fmt" "log" "sync"@@ -48,7 +49,7 @@ return m.waitErr } -func (m *MockCommand) Stop() error { +func (m *MockCommand) Stop(_ context.Context) error { m.mu.Lock() defer m.mu.Unlock() m.stopCalled = true@@ -148,7 +149,9 @@ mockCmd := NewMockCommand() var buf bytes.Buffer logger := log.New(&buf, "", 0) - sm := New(mockCmd, logger) + sm := New(mockCmd, Options{ + Logger: logger, + }) return sm, mockCmd, &buf }