Function TestClassroomService_ListClassroomsIter

github/github-iterators_test.go:4050–4120

(t *testing.T)

}

func TestClassroomService_ListClassroomsIter(t *testing.T) {
	t.Parallel()
	client, mux, _ := setup(t)
	var callNum int
	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		callNum++
		switch callNum {
		case 1:
			w.Header().Set("Link", `<https://api.github.com/?page=1>; rel="next"`)
			fmt.Fprint(w, `[{},{},{}]`)
		case 2:
			fmt.Fprint(w, `[{},{},{},{}]`)
		case 3:
			fmt.Fprint(w, `[{},{}]`)
		case 4:
			w.WriteHeader(http.StatusNotFound)
		case 5:
			fmt.Fprint(w, `[{},{}]`)
		}
	})

	iter := client.Classroom.ListClassroomsIter(t.Context(), nil)
	var gotItems int
	for _, err := range iter {
		gotItems++
		if err != nil {
			t.Errorf("Unexpected error: %v", err)
		}
	}
	if want := 7; gotItems != want {
		t.Errorf("client.Classroom.ListClassroomsIter call 1 got %v items; want %v", gotItems, want)
	}

	opts := &ListOptions{}
	iter = client.Classroom.ListClassroomsIter(t.Context(), opts)
	gotItems = 0
	for _, err := range iter {
		gotItems++
		if err != nil {
			t.Errorf("Unexpected error: %v", err)
		}
	}
	if want := 2; gotItems != want {
		t.Errorf("client.Classroom.ListClassroomsIter call 2 got %v items; want %v", gotItems, want)
	}

	iter = client.Classroom.ListClassroomsIter(t.Context(), nil)
	gotItems = 0
	for _, err := range iter {
		gotItems++
		if err == nil {
			t.Error("expected error; got nil")
		}
	}
	if gotItems != 1 {
		t.Errorf("client.Classroom.ListClassroomsIter call 3 got %v items; want 1 (an error)", gotItems)
	}