func TestPortRange(t *testing.T) {
defer func() {
if r := recover(); r != nil {
t.Fatalf("Recovered from panic: %s", r)
return
}
}()
defer os.Remove("port_range.yml")
log.SetLogLevel(log.FATAL)
if err := ioutil.WriteFile("port_range.yml", []byte(
`
listen: '1:25565'
upstreams:
- hostname: server.local
upstream: localhost:65537
- hostname: server2.local
upstream: localhost:-1`), 0644); err != nil {
t.Fatal("Unable to write to port_range.yml")
return
}
SetConfig("port_range.yml")
confInit()
if len(config.Upstream) != 0 {
t.Fatalf("No valid upstreams provided, but %d upstreams found!", len(config.Upstream))
return
}
t.Log("Ok, no valid upstream.")
}
func TestUpstreamExtraField(t *testing.T) {
defer func() {
if r := recover(); r != nil {
t.Fatalf("Recovered from panic: %s", r)
return
}
}()
defer os.Remove("upstream_extra.yml")
log.SetLogLevel(log.FATAL)
if err := ioutil.WriteFile("upstream_extra.yml", []byte(
`
listen: ':25565'
upstreams:
-
hostname: server1.local
upstream: '-_-:invalidhost!:25568'
-
hostname: '*.local'
upstream: 127.0.0.1:25566
bungeecord: true
-
hostname: '*'
upstream: '-_-:anotherinvalid!!!'
-
hostname: '233'
upstream: '1.2.3.4'`), 0644); err != nil {
t.Fatal("Unable to write to upstream_extra.yml")
return
}
SetConfig("upstream_extra.yml")
confInit()
upstream, errr := GetUpstream("server1.local")
if errr != nil {
t.Errorf("%+v\n", config)
t.Errorf("No host found! Check GetUpstream!")
t.Fatalf("Msg: %s", errr.Text)
return
}
t.Log("Ok, upstream fetched")
bungee, err := upstream.GetExtra("bungeecord")
if err != nil {
t.Fatalf("Unable to get bungeecord info: %s", err)
return
}
be, ok := bungee.(bool)
if !ok {
t.Fatal("Unable to convert to bool!")
return
}
t.Log("Ok, type is bool")
if !be {
t.Fatalf("Value incorrect! true expected, false found")
} else {
t.Log("Ok, value is true.")
}
}
func TestInvalidHost(t *testing.T) {
defer func() {
if r := recover(); r != nil {
t.Fatalf("Recovered from panic: %s", r)
return
}
}()
defer os.Remove("invalid_host.yml")
log.SetLogLevel(log.FATAL)
if err := ioutil.WriteFile("invalid_host.yml", []byte(
`
listen: '[::]:25565'
upstreams:
-
hostname: server1.local
upstream: '-_-:invalidhost!:25568'
-
hostname: '*.local'
upstream: 127.0.0.1:25566
-
hostname: '*'
upstream: '-_-:anotherinvalid!!!'
-
hostname: '233'
upstream: '1.2.3.4'`), 0644); err != nil {
t.Fatal("Unable to write to invalid_host.yml")
return
}
SetConfig("invalid_host.yml")
confInit()
ulen := len(config.Upstream)
if len(config.Upstream) != 2 {
t.Errorf("There should be 2 valid upstreams, %d found", ulen)
} else {
t.Log("Ok. 2 valid upstreams")
}
if ulen > 0 {
if config.Upstream[0].Server != "127.0.0.1:25566" {
t.Errorf("Upstream 0 should be 127.0.0.1:25566, %s found", config.Upstream[0].Server)
} else {
t.Log("Ok. upstream 0 is 127.0.0.1:25566")
}
if ulen > 1 {
if config.Upstream[1].Server != "1.2.3.4:25565" {
t.Errorf("Upstream 1 should be 1.2.3.4:25565, %s found", config.Upstream[1].Server)
} else {
t.Log("Ok. upstream 1 is 1.2.3.4:25565")
}
}
}
}
func TestConfigExtraField(t *testing.T) {
defer func() {
if r := recover(); r != nil {
t.Fatalf("Recovered from panic: %s", r)
return
}
}()
defer os.Remove("conf_extra.yml")
log.SetLogLevel(log.FATAL)
if err := ioutil.WriteFile("conf_extra.yml", []byte(
`
listen: '1:25565'
upstreams:
- hostname: server.local
upstream: localhost:25566
- hostname: server2.local
upstream: localhost:25567
fail2ban:
interval: 300
max_conn: 6`), 0644); err != nil {
t.Fatal("Unable to write to upstream_extra.yml")
return
}
SetConfig("conf_extra.yml")
confInit()
val, err := GetExtraConf("fail2ban.max_conn")
if err != nil {
t.Fatalf("No fail2ban.max_conn found: %s", err)
return
}
t.Log("Ok, key exists")
lim, ok := val.(int)
if !ok {
t.Fatal("Unable to convert to int!")
return
}
t.Log("Ok, type is int")
if lim != 6 {
t.Fatalf("Value incorrect! Expect 6, %d found.", lim)
} else {
t.Log("Ok, value is 6.")
}
}
func TestEmptyErrorMsg(t *testing.T) {
defer func() {
if r := recover(); r != nil {
t.Fatalf("Recovered from panic: %s", r)
return
}
}()
defer os.Remove("empty_msg.yml")
log.SetLogLevel(log.FATAL)
if err := ioutil.WriteFile("empty_msg.yml", []byte(
`
listen: '[::]:23333'
upstreams:
- hostname: server.local
upstream: test.local:23345`), 0644); err != nil {
t.Fatal("Unable to write to invalid_host.yml")
return
}
SetConfig("empty_msg.yml")
confInit()
t.Log("Everything fine :)")
}
func confInit() {
content, err := ioutil.ReadFile(config_file)
if err != nil {
log.Fatalf("unable to load config %s: %s", config_file, err.Error())
}
err = yaml.Unmarshal(content, &config)
if err != nil {
log.Fatalf("error when parsing config file %s: %s", config_file, err.Error())
}
validateConfig()
if config.Log.Target != "" && config.Log.Target != "-" {
config.Log.Target, _ = filepath.Abs(config.Log.Target)
log.Info("log path: " + config.Log.Target)
}
log.Stop()
if config.Daemonize {
Daemonize()
}
log.Start()
if config.Log.Level != "" {
level := log.ToLevel(config.Log.Level)
if level == log.INVALID {
log.Errorf("Invalid log level %s", config.Log.Level)
} else {
log.SetLogLevel(level)
}
}
if config.Log.Target != "" && config.Log.Target != "-" {
err := log.Open(config.Log.Target)
if err != nil {
log.Fatalf("Unable to open log %s: %s", config.Log.Target, err.Error())
}
}
log.Info("config loaded.")
log.Info("server listen on: " + config.Listen_addr)
log.Infof("%d upstream server(s) found", len(config.Upstream))
}