瀏覽代碼

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