diff options
Diffstat (limited to 'worker_test.go')
-rwxr-xr-x | worker_test.go | 118 |
1 files changed, 2 insertions, 116 deletions
diff --git a/worker_test.go b/worker_test.go index 78738064..e82d383e 100755 --- a/worker_test.go +++ b/worker_test.go @@ -15,7 +15,7 @@ func Test_GetState(t *testing.T) { w, err := NewPipeFactory().SpawnWorkerWithContext(ctx, cmd) go func() { - assert.NoError(t, w.Wait(ctx)) + assert.NoError(t, w.Wait()) assert.Equal(t, StateStopped, w.State().Value()) }() @@ -38,7 +38,7 @@ func Test_Kill(t *testing.T) { wg.Add(1) go func() { defer wg.Done() - assert.Error(t, w.Wait(ctx)) + assert.Error(t, w.Wait()) // TODO changed from stopped, discuss assert.Equal(t, StateErrored, w.State().Value()) }() @@ -64,117 +64,3 @@ func Test_OnStarted(t *testing.T) { assert.Equal(t, "can't attach to running process", err.Error()) } - -func TestErrBuffer_Write_Len(t *testing.T) { - buf := newErrBuffer(nil) - defer func() { - buf.Close() - }() - - _, err := buf.Write([]byte("hello")) - if err != nil { - t.Errorf("fail to write: error %v", err) - } - assert.Equal(t, 5, buf.Len()) - assert.Equal(t, "hello", buf.String()) -} - -func TestErrBuffer_Write_Event(t *testing.T) { - buf := newErrBuffer(nil) - defer func() { - buf.Close() - }() - - wg := &sync.WaitGroup{} - wg.Add(1) - buf.logCallback = func(log []byte) { - assert.Equal(t, []byte("hello\n"), log) - wg.Done() - } - buf.enable = true - - _, err := buf.Write([]byte("hello\n")) - if err != nil { - t.Errorf("fail to write: error %v", err) - } - - wg.Wait() - - // messages are read - assert.Equal(t, 0, buf.Len()) -} - -func TestErrBuffer_Write_Event_Separated(t *testing.T) { - buf := newErrBuffer(nil) - defer func() { - buf.Close() - }() - - wg := &sync.WaitGroup{} - wg.Add(1) - - buf.logCallback = func(log []byte) { - assert.Equal(t, []byte("hello\nending"), log) - wg.Done() - } - buf.enable = true - - _, err := buf.Write([]byte("hel")) - if err != nil { - t.Errorf("fail to write: error %v", err) - } - - _, err = buf.Write([]byte("lo\n")) - if err != nil { - t.Errorf("fail to write: error %v", err) - } - - _, err = buf.Write([]byte("ending")) - if err != nil { - t.Errorf("fail to write: error %v", err) - } - - wg.Wait() - assert.Equal(t, 0, buf.Len()) - assert.Equal(t, "", buf.String()) -} - -func TestErrBuffer_Write_Event_Separated_NoListener(t *testing.T) { - buf := newErrBuffer(nil) - defer func() { - buf.Close() - }() - - _, err := buf.Write([]byte("hel")) - if err != nil { - t.Errorf("fail to write: error %v", err) - } - - _, err = buf.Write([]byte("lo\n")) - if err != nil { - t.Errorf("fail to write: error %v", err) - } - - _, err = buf.Write([]byte("ending")) - if err != nil { - t.Errorf("fail to write: error %v", err) - } - - assert.Equal(t, 12, buf.Len()) - assert.Equal(t, "hello\nending", buf.String()) -} - -func TestErrBuffer_Write_Remaining(t *testing.T) { - buf := newErrBuffer(nil) - defer func() { - buf.Close() - }() - - _, err := buf.Write([]byte("hel")) - if err != nil { - t.Errorf("fail to write: error %v", err) - } - - assert.Equal(t, 3, buf.Len()) - assert.Equal(t, "hel", buf.String()) -} |