130 lines
3.5 KiB
Go
130 lines
3.5 KiB
Go
package doormen
|
|
|
|
import (
|
|
"encoding/json"
|
|
"math"
|
|
"testing"
|
|
|
|
"github.com/didiercrunch/doorman/shared"
|
|
"gopkg.in/mgo.v2/bson"
|
|
)
|
|
|
|
func TestNewDoormanDefinition(t *testing.T) {
|
|
dmd := NewDoormanDefinition("some_name")
|
|
if dmd.Id == "" {
|
|
t.Error("need id at all time")
|
|
}
|
|
if dmd.Name != "some_name" {
|
|
t.Error("bad name")
|
|
}
|
|
}
|
|
|
|
func TestValidate(t *testing.T) {
|
|
dmd := &DoormanDefinition{Name: ""}
|
|
if err := dmd.Validate(); err.Error() != "name cannot be empty" {
|
|
t.Error(err)
|
|
}
|
|
dmd = &DoormanDefinition{Name: "foo"}
|
|
if err := dmd.Validate(); err.Error() != "need at least 2 values" {
|
|
t.Error(err)
|
|
}
|
|
dmd = &DoormanDefinition{Name: "foo", Values: []*DoormanValue{nil}}
|
|
if err := dmd.Validate(); err.Error() != "need at least 2 values" {
|
|
t.Error(err)
|
|
}
|
|
|
|
dmd = &DoormanDefinition{Name: "foo", Values: []*DoormanValue{&DoormanValue{"foo", 0}, &DoormanValue{"foo", 0}}}
|
|
if err := dmd.Validate(); err.Error() != "doorman value names must be unique within a doorman" {
|
|
t.Error(err)
|
|
}
|
|
|
|
dmd = &DoormanDefinition{Name: "foo", Values: []*DoormanValue{&DoormanValue{"foo", 0}, &DoormanValue{"", 0}}}
|
|
if err := dmd.Validate(); err.Error() != "doorman value cannot be empty" {
|
|
t.Error(err)
|
|
}
|
|
|
|
dmd = &DoormanDefinition{Name: "foo", Values: []*DoormanValue{&DoormanValue{"foo", -0.25}, &DoormanValue{"bar", 0.5}}}
|
|
if err := dmd.Validate(); err.Error() != "doorman value -0.25 is out of range" {
|
|
t.Error(err)
|
|
}
|
|
|
|
dmd = &DoormanDefinition{Name: "foo", Values: []*DoormanValue{&DoormanValue{"foo", 2.5}, &DoormanValue{"bar", 0.5}}}
|
|
if err := dmd.Validate(); err.Error() != "doorman value 2.5 is out of range" {
|
|
t.Error(err)
|
|
}
|
|
|
|
dmd = &DoormanDefinition{Name: "foo", Values: []*DoormanValue{&DoormanValue{"foo", 0.75}, &DoormanValue{"bar", 0.5}}}
|
|
if err := dmd.Validate(); err.Error() != "the sum of the probability must be 1.0" {
|
|
t.Error(err)
|
|
}
|
|
|
|
dmd = &DoormanDefinition{Name: "foo", Values: []*DoormanValue{&DoormanValue{"foo", 0.2}, &DoormanValue{"bar", 0.2}}}
|
|
if err := dmd.Validate(); err.Error() != "the sum of the probability must be 1.0" {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestAsJson(t *testing.T) {
|
|
id := &DoormanId{}
|
|
if `{}` != id.AsJson() {
|
|
t.Fail()
|
|
}
|
|
id.Name = "bob"
|
|
if `{}` != id.AsJson() {
|
|
t.Fail()
|
|
}
|
|
|
|
id.Id = bson.ObjectIdHex("54af464af6fb5e20c4000003")
|
|
if `{"54af464af6fb5e20c4000003":"bob"}` != id.AsJson() {
|
|
t.Error(id.AsJson())
|
|
}
|
|
}
|
|
|
|
func TestAsDoormanUpdatePayload(t *testing.T) {
|
|
wd := &DoormanDefinition{}
|
|
wd.Id = bson.NewObjectId()
|
|
wd.Values = []*DoormanValue{&DoormanValue{"C", 0.25}, &DoormanValue{"T1", 0.75}}
|
|
data, err := wd.AsDoormanUpdatePayload()
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
wu := &shared.DoormanUpdater{}
|
|
if err := json.Unmarshal(data, wu); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if wd.Id.Hex() != wu.Id {
|
|
t.Error("bad id", wu.Id)
|
|
}
|
|
if len(wu.Probabilities) != 2 {
|
|
t.Error("bad probability length")
|
|
} else if math.Abs(wu.Probabilities[0]-0.25) > epsilon {
|
|
t.Error("bad first probability")
|
|
} else if math.Abs(wu.Probabilities[1]-0.75) > epsilon {
|
|
t.Error("bad second probability")
|
|
} else if wu.Timestamp == 0 {
|
|
t.Error("no time specify")
|
|
} else if wu.Timestamp < 1421537442 {
|
|
t.Error("timestamp looks weird")
|
|
}
|
|
}
|
|
|
|
func TestQuickNewDoormanDefinition(t *testing.T) {
|
|
dmd := QuickNewDoormanDefinition("foo", 0.75, 0.25)
|
|
if dmd.Name != "foo" {
|
|
t.Error()
|
|
}
|
|
|
|
if len(dmd.Values) != 2 {
|
|
t.Error()
|
|
}
|
|
|
|
if dmd.Values[0].Name != "T0" || dmd.Values[0].Probability != 0.75 {
|
|
t.Error()
|
|
}
|
|
|
|
if dmd.Values[1].Name != "T1" || dmd.Values[1].Probability != 0.25 {
|
|
t.Error()
|
|
}
|
|
}
|