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))
|
||||
ctx, cancel := context.WithTimeout(context.Background(), time.Second*120)
|
||||
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++ {
|
||||
go func(i int) {
|
||||
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)
|
||||
}
|
||||
wg.Wait()
|
||||
|
|
Loading…
Reference in a new issue