瀏覽代碼

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 1 年之前
父節點
當前提交
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()