summaryrefslogtreecommitdiff
path: root/worker_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'worker_test.go')
-rwxr-xr-xworker_test.go118
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())
-}