浏览代码

harden integration tests

Daniel Hiltgen 9 月之前
父节点
当前提交
343aba9fca
共有 2 个文件被更改,包括 2 次插入2 次删除
  1. 1 1
      integration/concurrency_test.go
  2. 1 1
      integration/utils_test.go

+ 1 - 1
integration/concurrency_test.go

@@ -40,7 +40,7 @@ func TestMultiModelConcurrency(t *testing.T) {
 		}
 		}
 		resp = [2][]string{
 		resp = [2][]string{
 			[]string{"sunlight"},
 			[]string{"sunlight"},
-			[]string{"england", "english", "massachusetts", "pilgrims", "british"},
+			[]string{"england", "english", "massachusetts", "pilgrims", "british", "festival"},
 		}
 		}
 	)
 	)
 	var wg sync.WaitGroup
 	var wg sync.WaitGroup

+ 1 - 1
integration/utils_test.go

@@ -275,7 +275,7 @@ func DoGenerate(ctx context.Context, t *testing.T, client *api.Client, genReq ap
 				break
 				break
 			}
 			}
 		}
 		}
-		require.True(t, atLeastOne, "none of %v found in %s", anyResp, response)
+		require.True(t, atLeastOne, "%s: none of %v found in %s", genReq.Model, anyResp, response)
 		slog.Info("test pass", "model", genReq.Model, "prompt", genReq.Prompt, "contains", anyResp, "response", response)
 		slog.Info("test pass", "model", genReq.Model, "prompt", genReq.Prompt, "contains", anyResp, "response", response)
 	case <-ctx.Done():
 	case <-ctx.Done():
 		t.Error("outer test context done while waiting for generate")
 		t.Error("outer test context done while waiting for generate")