65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
|
package disttest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"sync"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
"gitlab.com/slon/shad-go/distbuild/pkg/build"
|
||
|
|
)
|
||
|
|
|
||
|
|
var threeWorkerConfig = &Config{WorkerCount: 3}
|
||
|
|
|
||
|
|
func TestArtifactTransferBetweenWorkers(t *testing.T) {
|
||
|
|
env, cancel := newEnv(t, threeWorkerConfig)
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
baseJob := build.Job{
|
||
|
|
ID: build.ID{'a'},
|
||
|
|
Name: "write",
|
||
|
|
Cmds: []build.Cmd{
|
||
|
|
{CatTemplate: "OK", CatOutput: "{{.OutputDir}}/out.txt"},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
var wg sync.WaitGroup
|
||
|
|
wg.Add(3)
|
||
|
|
|
||
|
|
startTime := time.Now()
|
||
|
|
|
||
|
|
for i := 0; i < 3; i++ {
|
||
|
|
depJobID := build.ID{'b', byte(i)}
|
||
|
|
depJob := build.Job{
|
||
|
|
ID: depJobID,
|
||
|
|
Name: "cat",
|
||
|
|
Cmds: []build.Cmd{
|
||
|
|
{Exec: []string{"cat", fmt.Sprintf("{{index .Deps %q}}/out.txt", build.ID{'a'})}},
|
||
|
|
{Exec: []string{"sleep", "1"}, Environ: os.Environ()}, // DepTimeout is 100ms.
|
||
|
|
},
|
||
|
|
Deps: []build.ID{{'a'}},
|
||
|
|
}
|
||
|
|
|
||
|
|
graph := build.Graph{Jobs: []build.Job{baseJob, depJob}}
|
||
|
|
go func() {
|
||
|
|
defer wg.Done()
|
||
|
|
|
||
|
|
recorder := NewRecorder()
|
||
|
|
if !assert.NoError(t, env.Client.Build(env.Ctx, graph, recorder)) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Len(t, recorder.Jobs, 2)
|
||
|
|
assert.Equal(t, &JobResult{Stdout: "OK", Code: new(int)}, recorder.Jobs[depJobID])
|
||
|
|
}()
|
||
|
|
}
|
||
|
|
|
||
|
|
wg.Wait()
|
||
|
|
|
||
|
|
testDuration := time.Since(startTime)
|
||
|
|
assert.True(t, testDuration < time.Second*2, "test duration should be less than 2 seconds")
|
||
|
|
}
|