61 lines
1,005 B
Go
61 lines
1,005 B
Go
package balanced
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
var cfgJSON = []byte(`
|
|
{
|
|
"allocate_by": ["tag", "disk"]
|
|
}
|
|
`)
|
|
|
|
func TestLoadJSON(t *testing.T) {
|
|
cfg := &Config{}
|
|
err := cfg.LoadJSON(cfgJSON)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestToJSON(t *testing.T) {
|
|
cfg := &Config{}
|
|
cfg.LoadJSON(cfgJSON)
|
|
newjson, err := cfg.ToJSON()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
cfg = &Config{}
|
|
err = cfg.LoadJSON(newjson)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(cfg.AllocateBy) != 2 {
|
|
t.Error("configuration was lost in serialization/deserialization")
|
|
}
|
|
}
|
|
|
|
func TestDefault(t *testing.T) {
|
|
cfg := &Config{}
|
|
cfg.Default()
|
|
if cfg.Validate() != nil {
|
|
t.Fatal("error validating")
|
|
}
|
|
|
|
cfg.AllocateBy = nil
|
|
if cfg.Validate() == nil {
|
|
t.Fatal("expected error validating")
|
|
}
|
|
|
|
}
|
|
|
|
func TestApplyEnvVars(t *testing.T) {
|
|
os.Setenv("CLUSTER_BALANCED_ALLOCATEBY", "a,b,c")
|
|
cfg := &Config{}
|
|
cfg.ApplyEnvVars()
|
|
|
|
if len(cfg.AllocateBy) != 3 {
|
|
t.Fatal("failed to override allocate_by with env var")
|
|
}
|
|
}
|