6bb0d181 |
package container |
47065b90 |
import (
"sync/atomic"
"testing"
"time" |
f96e04ff |
"github.com/docker/docker/daemon/execdriver" |
47065b90 |
)
func TestStateRunStop(t *testing.T) {
s := NewState()
for i := 1; i < 3; i++ { // full lifecycle two times
started := make(chan struct{})
var pid int64
go func() { |
abd72d40 |
runPid, _ := s.waitRunning(-1 * time.Second) |
47065b90 |
atomic.StoreInt64(&pid, int64(runPid))
close(started)
}() |
eff253df |
s.Lock() |
6bb0d181 |
s.SetRunning(i + 100) |
eff253df |
s.Unlock() |
abd72d40 |
|
47065b90 |
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)
}
select {
case <-time.After(100 * time.Millisecond):
t.Fatal("Start callback doesn't fire in 100 milliseconds")
case <-started:
t.Log("Start callback fired")
}
runPid := int(atomic.LoadInt64(&pid))
if runPid != i+100 {
t.Fatalf("Pid %v, expected %v", runPid, i+100)
} |
abd72d40 |
if pid, err := s.waitRunning(-1 * time.Second); err != nil || pid != i+100 {
t.Fatalf("waitRunning returned pid: %v, err: %v, expected pid: %v, err: %v", pid, err, i+100, nil) |
47065b90 |
}
stopped := make(chan struct{})
var exit int64
go func() {
exitCode, _ := s.WaitStop(-1 * time.Second)
atomic.StoreInt64(&exit, int64(exitCode))
close(stopped)
}() |
6bb0d181 |
s.SetStoppedLocking(&execdriver.ExitStatus{ExitCode: i}) |
47065b90 |
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 { |
391c35c8 |
t.Fatalf("WaitStop returned exitCode: %v, err: %v, expected exitCode: %v, err: %v", exitCode, err, i, nil) |
47065b90 |
}
}
}
func TestStateTimeoutWait(t *testing.T) {
s := NewState()
started := make(chan struct{})
go func() { |
abd72d40 |
s.waitRunning(100 * time.Millisecond) |
47065b90 |
close(started)
}()
select {
case <-time.After(200 * time.Millisecond):
t.Fatal("Start callback doesn't fire in 100 milliseconds")
case <-started:
t.Log("Start callback fired")
} |
abd72d40 |
|
eff253df |
s.Lock() |
6bb0d181 |
s.SetRunning(49) |
eff253df |
s.Unlock() |
abd72d40 |
|
47065b90 |
stopped := make(chan struct{})
go func() { |
abd72d40 |
s.waitRunning(100 * time.Millisecond) |
47065b90 |
close(stopped)
}()
select {
case <-time.After(200 * time.Millisecond):
t.Fatal("Start callback doesn't fire in 100 milliseconds")
case <-stopped:
t.Log("Start callback fired")
}
} |