60c5339042
* Graceful: Create callbacks to with contexts * Graceful: Say when Gitea is completely finished * Graceful: Git and Process within HammerTime Force all git commands to terminate at HammerTime Force all process commands to terminate at HammerTime Move almost all git processes to run as git Commands * Graceful: Always Hammer after Shutdown * ProcessManager: Add cancel functionality * Fix tests * Make sure that process.Manager.Kill() cancels * Make threadsafe access to Processes and remove own unused Kill * Remove cmd from the process manager as it is no longer used * the default context is the correct context * get rid of double till
73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package process
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestManager_Add(t *testing.T) {
|
|
pm := Manager{processes: make(map[int64]*Process)}
|
|
|
|
pid := pm.Add("foo", nil)
|
|
assert.Equal(t, int64(1), pid, "expected to get pid 1 got %d", pid)
|
|
|
|
pid = pm.Add("bar", nil)
|
|
assert.Equal(t, int64(2), pid, "expected to get pid 2 got %d", pid)
|
|
}
|
|
|
|
func TestManager_Cancel(t *testing.T) {
|
|
pm := Manager{processes: make(map[int64]*Process)}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
pid := pm.Add("foo", cancel)
|
|
|
|
pm.Cancel(pid)
|
|
|
|
select {
|
|
case <-ctx.Done():
|
|
default:
|
|
assert.Fail(t, "Cancel should cancel the provided context")
|
|
}
|
|
}
|
|
|
|
func TestManager_Remove(t *testing.T) {
|
|
pm := Manager{processes: make(map[int64]*Process)}
|
|
|
|
pid1 := pm.Add("foo", nil)
|
|
assert.Equal(t, int64(1), pid1, "expected to get pid 1 got %d", pid1)
|
|
|
|
pid2 := pm.Add("bar", nil)
|
|
assert.Equal(t, int64(2), pid2, "expected to get pid 2 got %d", pid2)
|
|
|
|
pm.Remove(pid2)
|
|
|
|
_, exists := pm.processes[pid2]
|
|
assert.False(t, exists, "PID %d is in the list but shouldn't", pid2)
|
|
}
|
|
|
|
func TestExecTimeoutNever(t *testing.T) {
|
|
|
|
// TODO Investigate how to improve the time elapsed per round.
|
|
maxLoops := 10
|
|
for i := 1; i < maxLoops; i++ {
|
|
_, stderr, err := GetManager().ExecTimeout(5*time.Second, "ExecTimeout", "git", "--version")
|
|
if err != nil {
|
|
t.Fatalf("git --version: %v(%s)", err, stderr)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestExecTimeoutAlways(t *testing.T) {
|
|
|
|
maxLoops := 100
|
|
for i := 1; i < maxLoops; i++ {
|
|
_, stderr, err := GetManager().ExecTimeout(100*time.Microsecond, "ExecTimeout", "sleep", "5")
|
|
// TODO Simplify logging and errors to get precise error type. E.g. checking "if err != context.DeadlineExceeded".
|
|
if err == nil {
|
|
t.Fatalf("sleep 5 secs: %v(%s)", err, stderr)
|
|
}
|
|
}
|
|
}
|