X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/087d725b5ab41ad34ffd24938482d45a8b99a080..HEAD:/model_fake_helper_test.go?ds=inline diff --git a/model_fake_helper_test.go b/model_fake_helper_test.go index 08c4732..9873fb6 100644 --- a/model_fake_helper_test.go +++ b/model_fake_helper_test.go @@ -54,6 +54,7 @@ func (f *ModelFake) serve(w http.ResponseWriter, r *http.Request) { raw, err := io.ReadAll(r.Body) if err != nil { http.Error(w, err.Error(), http.StatusBadRequest) + return } @@ -65,10 +66,12 @@ func (f *ModelFake) serve(w http.ResponseWriter, r *http.Request) { err = json.Unmarshal(raw, &call.body) if err != nil { http.Error(w, err.Error(), http.StatusBadRequest) + return } f.mutex.Lock() + call.number = len(f.calls) f.calls = append(f.calls, call) reply := f.reply @@ -103,6 +106,7 @@ func modelTestRequest() ModelRequest { // modelWriteEvents writes events as a server-sent event stream. func modelWriteEvents(w http.ResponseWriter, events ...any) { w.Header().Set("Content-Type", "text/event-stream") + for _, event := range events { data, err := json.Marshal(event) if err != nil { @@ -155,10 +159,13 @@ func modelReadUsage(t *testing.T, path string) []ModelUsageRecord { }() var records []ModelUsageRecord + scanner := bufio.NewScanner(file) for scanner.Scan() { var record ModelUsageRecord + require.NoError(t, json.Unmarshal(scanner.Bytes(), &record)) + records = append(records, record) }