be5c2d1569
These new configuration settings control whether pins are enqueued in the priority queue or not.
94 lines
1.8 KiB
Go
94 lines
1.8 KiB
Go
package stateless
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
var cfgJSON = []byte(`
|
|
{
|
|
"max_pin_queue_size": 4092,
|
|
"concurrent_pins": 2,
|
|
"priority_pin_max_age": "240h",
|
|
"priority_pin_max_retries": 4
|
|
}
|
|
`)
|
|
|
|
func TestLoadJSON(t *testing.T) {
|
|
cfg := &Config{}
|
|
err := cfg.LoadJSON(cfgJSON)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
j := &jsonConfig{}
|
|
|
|
json.Unmarshal(cfgJSON, j)
|
|
j.ConcurrentPins = 10
|
|
j.PriorityPinMaxAge = "216h"
|
|
j.PriorityPinMaxRetries = 2
|
|
tst, _ := json.Marshal(j)
|
|
err = cfg.LoadJSON(tst)
|
|
if err != nil {
|
|
t.Error("did not expect an error")
|
|
}
|
|
if cfg.ConcurrentPins != 10 {
|
|
t.Error("expected 10 concurrent pins")
|
|
}
|
|
if cfg.PriorityPinMaxAge != 9*24*time.Hour {
|
|
t.Error("expected 9 days max age")
|
|
}
|
|
if cfg.PriorityPinMaxRetries != 2 {
|
|
t.Error("expected 2 max retries")
|
|
}
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|
|
|
|
func TestDefault(t *testing.T) {
|
|
cfg := &Config{}
|
|
cfg.Default()
|
|
if cfg.Validate() != nil {
|
|
t.Fatal("error validating")
|
|
}
|
|
|
|
cfg.ConcurrentPins = -2
|
|
if cfg.Validate() == nil {
|
|
t.Fatal("expected error validating")
|
|
}
|
|
cfg.ConcurrentPins = 3
|
|
cfg.PriorityPinMaxRetries = -1
|
|
if cfg.Validate() == nil {
|
|
t.Fatal("expected error validating")
|
|
}
|
|
}
|
|
|
|
func TestApplyEnvVars(t *testing.T) {
|
|
os.Setenv("CLUSTER_STATELESS_CONCURRENTPINS", "22")
|
|
os.Setenv("CLUSTER_STATELESS_PRIORITYPINMAXAGE", "72h")
|
|
cfg := &Config{}
|
|
cfg.ApplyEnvVars()
|
|
|
|
if cfg.ConcurrentPins != 22 {
|
|
t.Fatal("failed to override concurrent_pins with env var")
|
|
}
|
|
|
|
if cfg.PriorityPinMaxAge != 3*24*time.Hour {
|
|
t.Fatal("failed to override priority_pin_max_age with env var")
|
|
}
|
|
}
|