71 lines
1.9 KiB
Go
71 lines
1.9 KiB
Go
package config_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/drakkan/sftpgo/api"
|
|
"github.com/drakkan/sftpgo/config"
|
|
"github.com/drakkan/sftpgo/dataprovider"
|
|
"github.com/drakkan/sftpgo/sftpd"
|
|
)
|
|
|
|
func TestLoadConfigTest(t *testing.T) {
|
|
configDir := ".."
|
|
confName := "sftpgo.conf"
|
|
configFilePath := filepath.Join(configDir, confName)
|
|
err := config.LoadConfig(configFilePath)
|
|
if err != nil {
|
|
t.Errorf("error loading config")
|
|
}
|
|
emptyHTTPDConf := api.HTTPDConf{}
|
|
if config.GetHTTPDConfig() == emptyHTTPDConf {
|
|
t.Errorf("error loading httpd conf")
|
|
}
|
|
emptyProviderConf := dataprovider.Config{}
|
|
if config.GetProviderConf() == emptyProviderConf {
|
|
t.Errorf("error loading provider conf")
|
|
}
|
|
emptySFTPDConf := sftpd.Configuration{}
|
|
if config.GetSFTPDConfig().BindPort == emptySFTPDConf.BindPort {
|
|
t.Errorf("error loading SFTPD conf")
|
|
}
|
|
confName = "sftpgo.conf.missing"
|
|
configFilePath = filepath.Join(configDir, confName)
|
|
err = config.LoadConfig(configFilePath)
|
|
if err == nil {
|
|
t.Errorf("loading a non existent config file must fail")
|
|
}
|
|
ioutil.WriteFile(configFilePath, []byte("{invalid json}"), 0666)
|
|
err = config.LoadConfig(configFilePath)
|
|
if err == nil {
|
|
t.Errorf("loading an invalid config file must fail")
|
|
}
|
|
os.Remove(configFilePath)
|
|
}
|
|
|
|
func TestEmptyBanner(t *testing.T) {
|
|
configDir := ".."
|
|
confName := "temp.conf"
|
|
configFilePath := filepath.Join(configDir, confName)
|
|
config.LoadConfig(configFilePath)
|
|
sftpdConf := config.GetSFTPDConfig()
|
|
sftpdConf.Banner = " "
|
|
c := make(map[string]sftpd.Configuration)
|
|
c["sftpd"] = sftpdConf
|
|
jsonConf, _ := json.Marshal(c)
|
|
err := ioutil.WriteFile(configFilePath, jsonConf, 0666)
|
|
if err != nil {
|
|
t.Errorf("error saving temporary configuration")
|
|
}
|
|
config.LoadConfig(configFilePath)
|
|
sftpdConf = config.GetSFTPDConfig()
|
|
if strings.TrimSpace(sftpdConf.Banner) == "" {
|
|
t.Errorf("SFTPD banner cannot be empty")
|
|
}
|
|
os.Remove(configFilePath)
|
|
}
|