Fix concurrency integration test to work locally
This worked remotely but wound up trying to spawn multiple servers locally which doesn't work
This commit is contained in:
parent
43ed358f9a
commit
206797bda4
1 changed files with 9 additions and 1 deletions
|
@ -45,10 +45,18 @@ func TestMultiModelConcurrency(t *testing.T) {
|
||||||
wg.Add(len(req))
|
wg.Add(len(req))
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), time.Second*120)
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*120)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
|
|
||||||
|
client, _, cleanup := InitServerConnection(ctx, t)
|
||||||
|
defer cleanup()
|
||||||
|
|
||||||
|
for i := 0; i < len(req); i++ {
|
||||||
|
require.NoError(t, PullIfMissing(ctx, client, req[i].Model))
|
||||||
|
}
|
||||||
|
|
||||||
for i := 0; i < len(req); i++ {
|
for i := 0; i < len(req); i++ {
|
||||||
go func(i int) {
|
go func(i int) {
|
||||||
defer wg.Done()
|
defer wg.Done()
|
||||||
GenerateTestHelper(ctx, t, req[i], resp[i])
|
DoGenerate(ctx, t, client, req[i], resp[i], 30*time.Second, 10*time.Second)
|
||||||
}(i)
|
}(i)
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
Loading…
Reference in a new issue