59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
|
package builder
|
|||
|
|
|||
|
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{
|
|||
|
Client: http.Client{},
|
|||
|
Host: "https://e621.net",
|
|||
|
UserAgent: "Go-e621-SDK (@username)",
|
|||
|
Username: "memo",
|
|||
|
APIKey: "123456",
|
|||
|
}
|
|||
|
|
|||
|
getNote := NewGetNoteBuilder(requestContext)
|
|||
|
pool, err := getNote.SetNoteID(36957).Execute()
|
|||
|
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)
|
|||
|
|
|||
|
}
|