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:
Daniel Hiltgen 2024-05-23 13:12:14 -07:00
parent 43ed358f9a
commit 206797bda4

View file

@ -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()