浏览代码

Fix concurrency integration test to work locally

This worked remotely but wound up trying to spawn multiple servers
locally which doesn't work
Daniel Hiltgen 11 月之前
父节点
当前提交
206797bda4
共有 1 个文件被更改,包括 9 次插入1 次删除
  1. 9 1
      integration/concurrency_test.go

+ 9 - 1
integration/concurrency_test.go

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