classifier_test.go

package elasticthought

import (
	"encoding/json"
	"fmt"
	"strings"
	"sync"
	"testing"

	"github.com/couchbaselabs/go.assert"
	"github.com/tleyden/fakehttp"
)

// starting port to use for fakehttp servers
var port = 4444

// since I'm not sure if NextPort() can be called by multiple threads,
// protect it with a mutex
var portMutex = &sync.Mutex{}

// the fakehttp library doesn't provide an easy way to shutdown an http
// server, so as a workaround, run each fake http server on a unique port
// for each test.
func NextPort() int {
	portMutex.Lock()
	defer portMutex.Unlock()
	port2use := port
	port += 1
	return port2use
}

func TestInsertClassifier(t *testing.T) {

	testServer := fakehttp.NewHTTPServerWithPort(NextPort())
	testServer.Start()

	// response when go-couch tries to see that the server is up
	testServer.Response(200, jsonHeaders(), `{"version": "fake"}`)

	// response when go-couch check is db exists
	testServer.Response(200, jsonHeaders(), `{"db_name": "db"}`)

	// insert succeeds
	testServer.Response(200, jsonHeaders(), `{"id": "classifier", "rev": "bar", "ok": true}`)

	configuration := NewDefaultConfiguration()
	configuration.DbUrl = fmt.Sprintf("%v/db", testServer.URL)

	classifier := NewClassifier(*configuration)
	classifier.SpecificationUrl = "http://s3.com/proto.txt"
	classifier.TrainingJobID = "123"

	err := classifier.Insert()

	assert.True(t, err == nil)
	assert.Equals(t, "classifier", classifier.Id)
	assert.Equals(t, "bar", classifier.Revision)

}

func TestSetSpecificationUrl(t *testing.T) {

	testServer := fakehttp.NewHTTPServerWithPort(NextPort())
	testServer.Start()

	// response when go-couch tries to see that the server is up
	testServer.Response(200, jsonHeaders(), `{"version": "fake"}`)

	// response when go-couch check is db exists
	testServer.Response(200, jsonHeaders(), `{"db_name": "db"}`)

	// update succeeds
	testServer.Response(200, jsonHeaders(), `{"id": "classifier", "rev": "bar"}`)

	configuration := NewDefaultConfiguration()
	configuration.DbUrl = fmt.Sprintf("%v/db", testServer.URL)

	classifier := NewClassifier(*configuration)
	classifier.Id = "classifier"
	classifier.Revision = "foo"

	err := classifier.SetSpecificationUrl("whatever")
	assert.True(t, err == nil)

	// make assertions about outgoing request
	for _, savedReq := range testServer.SavedRequests {

		path := savedReq.Request.URL.Path

		if strings.HasSuffix(path, "db/classifier") {
			var requestDictionary map[string]interface{}
			err := json.Unmarshal(savedReq.Data, &requestDictionary)
			assert.True(t, err == nil)
			assert.Equals(t, requestDictionary["_id"], "classifier")
			assert.Equals(t, requestDictionary["_rev"], "foo")
			assert.Equals(t, requestDictionary["specification-url"], "whatever")

		}

	}

}