Function TestStreamPages_OnItemsErrorStopsPagination

internal/client/client_test.go:194–244

(t *testing.T)

}

func TestStreamPages_OnItemsErrorStopsPagination(t *testing.T) {
	apiCalls := 0
	rt := roundTripFunc(func(req *http.Request) (*http.Response, error) {
		apiCalls++
		if apiCalls == 1 {
			return jsonResponse(map[string]interface{}{
				"code": 0, "msg": "ok",
				"data": map[string]interface{}{
					"items":      []interface{}{map[string]interface{}{"id": "1"}},
					"has_more":   true,
					"page_token": "next",
				},
			}), nil
		}
		return jsonResponse(map[string]interface{}{
			"code": 0, "msg": "ok",
			"data": map[string]interface{}{
				"items":    []interface{}{map[string]interface{}{"id": "2"}},
				"has_more": false,
			},
		}), nil
	})

	ac, _ := newTestAPIClient(t, rt)
	sentinel := errors.New("stop streaming")
	var streamedItems []interface{}
	result, hasItems, err := ac.StreamPages(context.Background(), RawApiRequest{
		Method: "GET",
		URL:    "/open-apis/contact/v3/users",
		As:     "bot",
	}, func(items []interface{}) error {
		streamedItems = append(streamedItems, items...)
		return sentinel
	}, PaginationOptions{PageDelay: 0})

	if !errors.Is(err, sentinel) {
		t.Fatalf("err = %v, want sentinel", err)
	}
	if result != nil {
		t.Fatalf("result = %#v, want nil when callback stops pagination", result)
	}
	if hasItems {
		t.Fatal("hasItems = true, want false when callback stops before returning")
	}
	if apiCalls != 1 {
		t.Fatalf("apiCalls = %d, want early stop after first page", apiCalls)
	}
	if len(streamedItems) != 1 {
		t.Fatalf("streamedItems = %d, want first page only", len(streamedItems))
	}
}

func TestPaginateAll_PageLimitStopsPagination(t *testing.T) {
	apiCalls := 0