57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGenerateExampleConfig(t *testing.T) {
|
||
|
|
testOutput := bytes.Buffer{}
|
||
|
|
logrus.SetOutput(&testOutput)
|
||
|
|
|
||
|
|
config, err := generateExampleConfig("test.yaml")
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("unexpected error %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if config.SerialAddress != "/dev/ttyUSB0" {
|
||
|
|
t.Errorf("got %s instead of expected serial address /dev/ttyUSB0", config.SerialAddress)
|
||
|
|
}
|
||
|
|
|
||
|
|
if config.BaudRate != 115200 {
|
||
|
|
t.Errorf("got %d instead of expected baud rate 115200", config.BaudRate)
|
||
|
|
}
|
||
|
|
|
||
|
|
logOutput := testOutput.String()
|
||
|
|
if !strings.Contains(logOutput, "test.yaml") {
|
||
|
|
t.Error("logged instructions do not contain the file name test.yaml")
|
||
|
|
}
|
||
|
|
|
||
|
|
if !strings.Contains(logOutput, "---\\n") {
|
||
|
|
t.Errorf("logged instructions do not contain a yaml prefix: \n%s", logOutput)
|
||
|
|
}
|
||
|
|
|
||
|
|
yamlKeys := []string{"serial_address", "serial_baudrate", "paranoid", "debug", "gnupg_bin", "openssl_bin", "mysql_dsn"}
|
||
|
|
for _, item := range yamlKeys {
|
||
|
|
if !strings.Contains(logOutput, item) {
|
||
|
|
t.Errorf("%s missing in example log output", item)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestReadConfigNonExistingFile(t *testing.T) {
|
||
|
|
config, err := readConfig("nonExistant.yaml")
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("unexpected error %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
exampleConfig, _ := generateExampleConfig("nonExistant.yaml")
|
||
|
|
if config != exampleConfig {
|
||
|
|
t.Errorf("expected example config but got %+v", config)
|
||
|
|
}
|
||
|
|
}
|