mirror of
https://github.com/opentofu/opentofu.git
synced 2025-01-19 21:22:57 -06:00
4c5b866dea
This is part of a general effort to move all of Terraform's non-library package surface under internal in order to reinforce that these are for internal use within Terraform only. If you were previously importing packages under this prefix into an external codebase, you could pin to an earlier release tag as an interim solution until you've make a plan to achieve the same functionality some other way.
119 lines
2.5 KiB
Go
119 lines
2.5 KiB
Go
package response
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
)
|
|
|
|
func prettyJSON(o interface{}) (string, error) {
|
|
bytes, err := json.MarshalIndent(o, "", "\t")
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(bytes), nil
|
|
}
|
|
|
|
func TestNewPaginationMeta(t *testing.T) {
|
|
type args struct {
|
|
offset int
|
|
limit int
|
|
hasMore bool
|
|
currentURL string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantJSON string
|
|
}{
|
|
{
|
|
name: "first page",
|
|
args: args{0, 10, true, "http://foo.com/v1/bar"},
|
|
wantJSON: `{
|
|
"limit": 10,
|
|
"current_offset": 0,
|
|
"next_offset": 10,
|
|
"next_url": "http://foo.com/v1/bar?offset=10"
|
|
}`,
|
|
},
|
|
{
|
|
name: "second page",
|
|
args: args{10, 10, true, "http://foo.com/v1/bar"},
|
|
wantJSON: `{
|
|
"limit": 10,
|
|
"current_offset": 10,
|
|
"next_offset": 20,
|
|
"prev_offset": 0,
|
|
"next_url": "http://foo.com/v1/bar?offset=20",
|
|
"prev_url": "http://foo.com/v1/bar"
|
|
}`,
|
|
},
|
|
{
|
|
name: "last page",
|
|
args: args{40, 10, false, "http://foo.com/v1/bar"},
|
|
wantJSON: `{
|
|
"limit": 10,
|
|
"current_offset": 40,
|
|
"prev_offset": 30,
|
|
"prev_url": "http://foo.com/v1/bar?offset=30"
|
|
}`,
|
|
},
|
|
{
|
|
name: "misaligned start ending exactly on boundary",
|
|
args: args{32, 10, false, "http://foo.com/v1/bar"},
|
|
wantJSON: `{
|
|
"limit": 10,
|
|
"current_offset": 32,
|
|
"prev_offset": 22,
|
|
"prev_url": "http://foo.com/v1/bar?offset=22"
|
|
}`,
|
|
},
|
|
{
|
|
name: "misaligned start partially through first page",
|
|
args: args{5, 12, true, "http://foo.com/v1/bar"},
|
|
wantJSON: `{
|
|
"limit": 12,
|
|
"current_offset": 5,
|
|
"next_offset": 17,
|
|
"prev_offset": 0,
|
|
"next_url": "http://foo.com/v1/bar?offset=17",
|
|
"prev_url": "http://foo.com/v1/bar"
|
|
}`,
|
|
},
|
|
{
|
|
name: "no current URL",
|
|
args: args{10, 10, true, ""},
|
|
wantJSON: `{
|
|
"limit": 10,
|
|
"current_offset": 10,
|
|
"next_offset": 20,
|
|
"prev_offset": 0
|
|
}`,
|
|
},
|
|
{
|
|
name: "#58 regression test",
|
|
args: args{1, 3, true, ""},
|
|
wantJSON: `{
|
|
"limit": 3,
|
|
"current_offset": 1,
|
|
"next_offset": 4,
|
|
"prev_offset": 0
|
|
}`,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := NewPaginationMeta(tt.args.offset, tt.args.limit, tt.args.hasMore,
|
|
tt.args.currentURL)
|
|
gotJSON, err := prettyJSON(got)
|
|
if err != nil {
|
|
t.Fatalf("failed to marshal PaginationMeta to JSON: %s", err)
|
|
}
|
|
if gotJSON != tt.wantJSON {
|
|
// prettyJSON makes debugging easier due to the annoying pointer-to-ints, but it
|
|
// also implicitly tests JSON marshalling as we can see if it's omitting fields etc.
|
|
t.Fatalf("NewPaginationMeta() =\n%s\n want:\n%s\n", gotJSON, tt.wantJSON)
|
|
}
|
|
})
|
|
}
|
|
}
|