X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/a606686934b4028c29dca8e9dae6b2e188cee9c5..871743f6ab20d68dcede1673a9764730fda38308:/build_run_test.go?ds=inline diff --git a/build_run_test.go b/build_run_test.go index 64d0d86..fc8b4b7 100644 --- a/build_run_test.go +++ b/build_run_test.go @@ -82,6 +82,7 @@ func TestBuildRunStopsWhenTheQuotaGrows(t *testing.T) { require.Greater(t, len(records), int(buildQuotaLimit)) var report strings.Builder + require.NoError(t, buildWriteReport(&report, appended, stop, records, sizes)) require.Contains(t, report.String(), "stopped early: ") require.Contains(t, report.String(), "cached (80.0%)") @@ -111,11 +112,14 @@ func TestBuildRunStopsAtTheCallLimit(t *testing.T) { func TestBuildRunWaitsOnlyBeforeUserMessages(t *testing.T) { gate := make(chan struct{}) + var mutex sync.Mutex + running, most := 0, 0 server := newModelFake(t) - server.reply = func(w http.ResponseWriter, call ModelFakeCall) { + server.reply = func(w http.ResponseWriter, _ ModelFakeCall) { mutex.Lock() + running++ most = max(most, running) mutex.Unlock() @@ -123,6 +127,7 @@ func TestBuildRunWaitsOnlyBeforeUserMessages(t *testing.T) { <-gate mutex.Lock() + running-- mutex.Unlock()