Add pagination example to Client4 (#25544)
* Add pagination example to Client4 * Break as soon less the expected items are returned * Print users as directly after fetching them
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
ebdc964161
Коммит
645d3a3b35
@@ -1,17 +1,21 @@
|
|||||||
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
||||||
// See LICENSE.txt for license information.
|
// See LICENSE.txt for license information.
|
||||||
|
|
||||||
package model
|
package model_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httptest"
|
"net/http/httptest"
|
||||||
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/mattermost/mattermost/server/public/model"
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -22,21 +26,21 @@ func TestClient4TrimTrailingSlash(t *testing.T) {
|
|||||||
|
|
||||||
for _, s := range slashes {
|
for _, s := range slashes {
|
||||||
testURL := baseURL + strings.Repeat("/", s)
|
testURL := baseURL + strings.Repeat("/", s)
|
||||||
client := NewAPIv4Client(testURL)
|
client := model.NewAPIv4Client(testURL)
|
||||||
assert.Equal(t, baseURL, client.URL)
|
assert.Equal(t, baseURL, client.URL)
|
||||||
assert.Equal(t, baseURL+APIURLSuffix, client.APIURL)
|
assert.Equal(t, baseURL+model.APIURLSuffix, client.APIURL)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// https://github.com/mattermost/mattermost/server/v8/channels/issues/8205
|
// https://github.com/mattermost/mattermost/server/v8/channels/issues/8205
|
||||||
func TestClient4CreatePost(t *testing.T) {
|
func TestClient4CreatePost(t *testing.T) {
|
||||||
post := &Post{
|
post := &model.Post{
|
||||||
Props: map[string]any{
|
Props: map[string]any{
|
||||||
"attachments": []*SlackAttachment{
|
"attachments": []*model.SlackAttachment{
|
||||||
{
|
{
|
||||||
Actions: []*PostAction{
|
Actions: []*model.PostAction{
|
||||||
{
|
{
|
||||||
Integration: &PostActionIntegration{
|
Integration: &model.PostActionIntegration{
|
||||||
Context: map[string]any{
|
Context: map[string]any{
|
||||||
"foo": "bar",
|
"foo": "bar",
|
||||||
},
|
},
|
||||||
@@ -51,15 +55,15 @@ func TestClient4CreatePost(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
var post Post
|
var post model.Post
|
||||||
err := json.NewDecoder(r.Body).Decode(&post)
|
err := json.NewDecoder(r.Body).Decode(&post)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
attachments := post.Attachments()
|
attachments := post.Attachments()
|
||||||
assert.Equal(t, []*SlackAttachment{
|
assert.Equal(t, []*model.SlackAttachment{
|
||||||
{
|
{
|
||||||
Actions: []*PostAction{
|
Actions: []*model.PostAction{
|
||||||
{
|
{
|
||||||
Integration: &PostActionIntegration{
|
Integration: &model.PostActionIntegration{
|
||||||
Context: map[string]any{
|
Context: map[string]any{
|
||||||
"foo": "bar",
|
"foo": "bar",
|
||||||
},
|
},
|
||||||
@@ -74,32 +78,32 @@ func TestClient4CreatePost(t *testing.T) {
|
|||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
}))
|
}))
|
||||||
|
|
||||||
client := NewAPIv4Client(server.URL)
|
client := model.NewAPIv4Client(server.URL)
|
||||||
_, resp, err := client.CreatePost(context.Background(), post)
|
_, resp, err := client.CreatePost(context.Background(), post)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestClient4SetToken(t *testing.T) {
|
func TestClient4SetToken(t *testing.T) {
|
||||||
expected := NewId()
|
expected := model.NewId()
|
||||||
|
|
||||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
authHeader := r.Header.Get(HeaderAuth)
|
authHeader := r.Header.Get(model.HeaderAuth)
|
||||||
|
|
||||||
token := strings.Split(authHeader, HeaderBearer)
|
token := strings.Split(authHeader, model.HeaderBearer)
|
||||||
|
|
||||||
if len(token) < 2 {
|
if len(token) < 2 {
|
||||||
t.Errorf("wrong authorization header format, got %s, expected: %s %s", authHeader, HeaderBearer, expected)
|
t.Errorf("wrong authorization header format, got %s, expected: %s %s", authHeader, model.HeaderBearer, expected)
|
||||||
}
|
}
|
||||||
|
|
||||||
assert.Equal(t, expected, strings.TrimSpace(token[1]))
|
assert.Equal(t, expected, strings.TrimSpace(token[1]))
|
||||||
|
|
||||||
var user User
|
var user model.User
|
||||||
err := json.NewEncoder(w).Encode(&user)
|
err := json.NewEncoder(w).Encode(&user)
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
}))
|
}))
|
||||||
|
|
||||||
client := NewAPIv4Client(server.URL)
|
client := model.NewAPIv4Client(server.URL)
|
||||||
client.SetToken(expected)
|
client.SetToken(expected)
|
||||||
|
|
||||||
_, resp, err := client.GetMe(context.Background(), "")
|
_, resp, err := client.GetMe(context.Background(), "")
|
||||||
@@ -113,7 +117,7 @@ func TestClient4RequestCancellation(t *testing.T) {
|
|||||||
t.Fatal("request should not hit the server")
|
t.Fatal("request should not hit the server")
|
||||||
}))
|
}))
|
||||||
|
|
||||||
client := NewAPIv4Client(server.URL)
|
client := model.NewAPIv4Client(server.URL)
|
||||||
|
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
|
||||||
@@ -131,7 +135,7 @@ func TestClient4RequestCancellation(t *testing.T) {
|
|||||||
t.Fatal("request should not hit the server")
|
t.Fatal("request should not hit the server")
|
||||||
}))
|
}))
|
||||||
|
|
||||||
client := NewAPIv4Client(server.URL)
|
client := model.NewAPIv4Client(server.URL)
|
||||||
|
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
|
||||||
@@ -149,3 +153,28 @@ func TestClient4RequestCancellation(t *testing.T) {
|
|||||||
<-done
|
<-done
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func ExampleClient4_GetUsers() {
|
||||||
|
client := model.NewAPIv4Client("http://localhost:8065")
|
||||||
|
client.SetToken(os.Getenv("MM_TOKEN"))
|
||||||
|
|
||||||
|
const perPage = 100
|
||||||
|
var page int
|
||||||
|
for {
|
||||||
|
users, _, err := client.GetUsers(context.TODO(), page, perPage, "")
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("error fetching users: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, u := range users {
|
||||||
|
fmt.Printf("%s\n", u.Username)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(users) < perPage {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
page++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Ссылка в новой задаче
Block a user