api_test.go

package smugmug

import (
	"testing"
)

type albumMockHandler struct {
	called int
}

func (c *albumMockHandler) get(url string, obj interface{}) error {
	defer func() { c.called++ }()
	a := obj.(*albumsResponse)
	a.Response.Album = []album{
		{},
		{},
	}
	if c.called == 0 {
		a.Response.Pages.NextPage = "something"
		return nil
	}
	a.Response.Pages.NextPage = ""
	return nil
}

func TestGetAlbums(t *testing.T) {
	w := &Worker{
		req: &albumMockHandler{},
	}
	albums, err := w.albums("someurl")
	if err != nil {
		t.Errorf("Unexpected error %v", err)
	}
	if w.req.(*albumMockHandler).called != 2 {
		t.Errorf("Called, want 2, got %d", w.req.(*albumMockHandler).called)
	}
	if len(albums) != 4 {
		t.Errorf("Want 4, got %d", len(albums))
	}
}

type albumImages struct {
	called int
}

func (c *albumImages) get(url string, obj interface{}) error {
	defer func() { c.called++ }()
	a := obj.(*albumImagesResponse)
	a.Response.AlbumImage = []albumImage{
		{FileName: "value"},
		{FileName: "value"},
	}
	if c.called == 0 {
		a.Response.Pages.NextPage = "something"
		return nil
	}
	a.Response.Pages.NextPage = ""
	return nil
}

func TestGetAlbumImages(t *testing.T) {
	tmpl, _ := buildFilenameTemplate("")
	w := &Worker{
		req:          &albumImages{},
		filenameTmpl: tmpl,
	}
	albums, err := w.albumImages("someurl", "myAlbumPath")
	if err != nil {
		t.Errorf("Unexpected error %v", err)
	}
	if w.req.(*albumImages).called != 2 {
		t.Errorf("Called, want 2, got %d", w.req.(*albumImages).called)
	}
	if len(albums) != 4 {
		t.Errorf("Want 4, got %d", len(albums))
	}
}