2015-11-12 14:55:17 -05:00
|
|
|
package container
|
2014-06-06 07:28:12 -04:00
|
|
|
|
|
|
|
import (
|
|
|
|
"sync/atomic"
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestStateRunStop(t *testing.T) {
|
|
|
|
s := NewState()
|
|
|
|
for i := 1; i < 3; i++ { // full lifecycle two times
|
2015-09-28 16:48:12 -04:00
|
|
|
s.Lock()
|
2016-03-18 14:50:19 -04:00
|
|
|
s.SetRunning(i+100, false)
|
2015-09-28 16:48:12 -04:00
|
|
|
s.Unlock()
|
2015-07-30 17:01:53 -04:00
|
|
|
|
2014-06-06 07:28:12 -04:00
|
|
|
if !s.IsRunning() {
|
|
|
|
t.Fatal("State not running")
|
|
|
|
}
|
|
|
|
if s.Pid != i+100 {
|
|
|
|
t.Fatalf("Pid %v, expected %v", s.Pid, i+100)
|
|
|
|
}
|
|
|
|
if s.ExitCode != 0 {
|
|
|
|
t.Fatalf("ExitCode %v, expected 0", s.ExitCode)
|
|
|
|
}
|
|
|
|
|
|
|
|
stopped := make(chan struct{})
|
|
|
|
var exit int64
|
|
|
|
go func() {
|
|
|
|
exitCode, _ := s.WaitStop(-1 * time.Second)
|
|
|
|
atomic.StoreInt64(&exit, int64(exitCode))
|
|
|
|
close(stopped)
|
|
|
|
}()
|
2016-03-18 14:50:19 -04:00
|
|
|
s.SetStoppedLocking(&ExitStatus{ExitCode: i})
|
2014-06-06 07:28:12 -04:00
|
|
|
if s.IsRunning() {
|
|
|
|
t.Fatal("State is running")
|
|
|
|
}
|
|
|
|
if s.ExitCode != i {
|
|
|
|
t.Fatalf("ExitCode %v, expected %v", s.ExitCode, i)
|
|
|
|
}
|
|
|
|
if s.Pid != 0 {
|
|
|
|
t.Fatalf("Pid %v, expected 0", s.Pid)
|
|
|
|
}
|
|
|
|
select {
|
|
|
|
case <-time.After(100 * time.Millisecond):
|
|
|
|
t.Fatal("Stop callback doesn't fire in 100 milliseconds")
|
|
|
|
case <-stopped:
|
|
|
|
t.Log("Stop callback fired")
|
|
|
|
}
|
|
|
|
exitCode := int(atomic.LoadInt64(&exit))
|
|
|
|
if exitCode != i {
|
|
|
|
t.Fatalf("ExitCode %v, expected %v", exitCode, i)
|
|
|
|
}
|
|
|
|
if exitCode, err := s.WaitStop(-1 * time.Second); err != nil || exitCode != i {
|
2014-08-13 03:37:30 -04:00
|
|
|
t.Fatalf("WaitStop returned exitCode: %v, err: %v, expected exitCode: %v, err: %v", exitCode, err, i, nil)
|
2014-06-06 07:28:12 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestStateTimeoutWait(t *testing.T) {
|
|
|
|
s := NewState()
|
2016-04-25 23:50:12 -04:00
|
|
|
stopped := make(chan struct{})
|
2014-06-06 07:28:12 -04:00
|
|
|
go func() {
|
2016-04-25 23:50:12 -04:00
|
|
|
s.WaitStop(100 * time.Millisecond)
|
|
|
|
close(stopped)
|
2014-06-06 07:28:12 -04:00
|
|
|
}()
|
|
|
|
select {
|
|
|
|
case <-time.After(200 * time.Millisecond):
|
2016-04-25 23:50:12 -04:00
|
|
|
t.Fatal("Stop callback doesn't fire in 100 milliseconds")
|
|
|
|
case <-stopped:
|
|
|
|
t.Log("Stop callback fired")
|
2014-06-06 07:28:12 -04:00
|
|
|
}
|
2015-07-30 17:01:53 -04:00
|
|
|
|
2016-04-25 23:50:12 -04:00
|
|
|
s.SetStoppedLocking(&ExitStatus{ExitCode: 1})
|
2015-07-30 17:01:53 -04:00
|
|
|
|
2016-04-25 23:50:12 -04:00
|
|
|
stopped = make(chan struct{})
|
2014-06-06 07:28:12 -04:00
|
|
|
go func() {
|
2016-04-25 23:50:12 -04:00
|
|
|
s.WaitStop(100 * time.Millisecond)
|
2014-06-06 07:28:12 -04:00
|
|
|
close(stopped)
|
|
|
|
}()
|
|
|
|
select {
|
|
|
|
case <-time.After(200 * time.Millisecond):
|
2016-04-25 23:50:12 -04:00
|
|
|
t.Fatal("Stop callback doesn't fire in 100 milliseconds")
|
2014-06-06 07:28:12 -04:00
|
|
|
case <-stopped:
|
2016-04-25 23:50:12 -04:00
|
|
|
t.Log("Stop callback fired")
|
2014-06-06 07:28:12 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|