139 lines
3.3 KiB
Go
139 lines
3.3 KiB
Go
package endpoints
|
||
|
||
import (
|
||
"git.dragse.it/anthrove/e621-to-graph/pkg/e621/model"
|
||
"github.com/jarcoal/httpmock"
|
||
"net/http"
|
||
"testing"
|
||
)
|
||
|
||
func TestGetNote(t *testing.T) {
|
||
httpmock.Activate()
|
||
defer httpmock.DeactivateAndReset()
|
||
|
||
noteResponse := model.Note{
|
||
ID: 385777,
|
||
CreatedAt: "2023-10-19T06:42:20.171-04:00",
|
||
UpdatedAt: "2023-10-19T06:42:20.171-04:00",
|
||
CreatorID: 1472475,
|
||
X: 277,
|
||
Y: 841,
|
||
Width: 101,
|
||
Height: 176,
|
||
Version: 2,
|
||
IsActive: true,
|
||
PostID: 2624329,
|
||
Body: "Well, isn’t it just right to use them?",
|
||
CreatorName: "ponypussypounder",
|
||
}
|
||
|
||
jsonResponser, err := httpmock.NewJsonResponder(200, noteResponse)
|
||
if err != nil {
|
||
t.Error(err)
|
||
return
|
||
}
|
||
httpmock.RegisterResponder("GET", "https://e621.net/notes/36957.json", jsonResponser)
|
||
|
||
requestContext := model.RequestContext{
|
||
Host: "https://e621.net",
|
||
UserAgent: "Go-e621-SDK (@username)",
|
||
Username: "memo",
|
||
APIKey: "123456",
|
||
}
|
||
|
||
client := http.Client{}
|
||
pool, err := GetNote(client, requestContext, "36957")
|
||
if err != nil {
|
||
t.Error(err)
|
||
return
|
||
}
|
||
|
||
if pool.ID == noteResponse.ID && pool.Body == noteResponse.Body {
|
||
return
|
||
}
|
||
|
||
t.Errorf("Respons did not match mock data:\nOriginal: %v\nMock: %v", pool, noteResponse)
|
||
|
||
}
|
||
|
||
func TestGetNotes(t *testing.T) {
|
||
httpmock.Activate()
|
||
defer httpmock.DeactivateAndReset()
|
||
|
||
noteResponse := []model.Note{
|
||
{
|
||
ID: 385777,
|
||
CreatedAt: "2023-10-19T06:42:20.171-04:00",
|
||
UpdatedAt: "2023-10-19T06:42:20.171-04:00",
|
||
CreatorID: 1472475,
|
||
X: 277,
|
||
Y: 841,
|
||
Width: 101,
|
||
Height: 176,
|
||
Version: 2,
|
||
IsActive: true,
|
||
PostID: 2624329,
|
||
Body: "Well, isn’t it just right to use them?",
|
||
CreatorName: "ponypussypounder",
|
||
},
|
||
{
|
||
ID: 56782,
|
||
CreatedAt: "2023-10-19T06:42:20.171-04:00",
|
||
UpdatedAt: "2023-10-19T06:42:20.171-04:00",
|
||
CreatorID: 1472475,
|
||
X: 277,
|
||
Y: 841,
|
||
Width: 101,
|
||
Height: 176,
|
||
Version: 2,
|
||
IsActive: true,
|
||
PostID: 2624329,
|
||
Body: "Well, isn’t it just right to use them?",
|
||
CreatorName: "ponypussypounder",
|
||
},
|
||
{
|
||
ID: 43847234,
|
||
CreatedAt: "2023-10-19T06:42:20.171-04:00",
|
||
UpdatedAt: "2023-10-19T06:42:20.171-04:00",
|
||
CreatorID: 1472475,
|
||
X: 277,
|
||
Y: 841,
|
||
Width: 101,
|
||
Height: 176,
|
||
Version: 2,
|
||
IsActive: true,
|
||
PostID: 2624329,
|
||
Body: "Well, isn’t it just right to use them?",
|
||
CreatorName: "ponypussypounder",
|
||
},
|
||
}
|
||
|
||
jsonResponser, err := httpmock.NewJsonResponder(200, noteResponse)
|
||
if err != nil {
|
||
t.Error(err)
|
||
return
|
||
}
|
||
httpmock.RegisterResponder("GET", "https://e621.net/notes.json", jsonResponser)
|
||
|
||
requestContext := model.RequestContext{
|
||
Host: "https://e621.net",
|
||
UserAgent: "Go-e621-SDK (@username)",
|
||
Username: "memo",
|
||
APIKey: "123456",
|
||
}
|
||
|
||
client := http.Client{}
|
||
pools, err := GetNotes(client, requestContext, map[string]string{})
|
||
if err != nil {
|
||
t.Error(err)
|
||
return
|
||
}
|
||
|
||
if len(pools) == 3 && pools[0].ID == noteResponse[0].ID && pools[1].Body == noteResponse[1].Body && pools[2].UpdatedAt == noteResponse[2].UpdatedAt {
|
||
return
|
||
}
|
||
|
||
t.Errorf("Respons did not match mock data:\nOriginal: %v\nMock: %v", pools, noteResponse)
|
||
|
||
}
|