onedrive_test.go

package onedrive

import (
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"testing"

	"gotest.tools/assert"
)

var (
	mux      *http.ServeMux
	server   *httptest.Server
	oneDrive *OneDrive
)

func setup() {
	mux = http.NewServeMux()
	server = httptest.NewServer(mux)
	oneDrive = New(http.DefaultClient)
	oneDrive.BaseURL = server.URL
}

func teardown() {
	server.Close()
}

func fileWrapperHandler(file string, status int) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(status)
		b, err := ioutil.ReadFile(file)
		if err != nil {
			panic(err)
		}
		w.Write(b)
	}
}

func TestOneDrive_New(t *testing.T) {
	client := http.DefaultClient
	od := New(client)

	assert.Equal(t, od.Client, client)
	assert.Assert(t, od.Drives != nil)
	assert.Assert(t, od.Items != nil)
	assert.Assert(t, od.Sites != nil)
	assert.Assert(t, od.SearchService != nil)

	_, isSite := od.Sites.(siteService)
	assert.Assert(t, isSite)
}