TheArchive/headscale
1
0
Fork 0
mirror of https://github.com/juanfont/headscale.git synced 2024-11-27 12:05:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Linting and updating tests

Browse Source
This commit is contained in:
Justin Angel 2022-02-21 10:09:23 -05:00
parent 7bf2a91dd0
commit daa75da277
3 changed files with 14 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
app_test.go
View File

@ -69,9 +69,8 @@ func (s *Suite) ResetDB(c *check.C) {
// Enusre an error is returned when an invalid auth mode // Enusre an error is returned when an invalid auth mode
// is supplied. // is supplied.
func (s *Suite) TestInvalidClientAuthMode(c *check.C) { func (s *Suite) TestInvalidClientAuthMode(c *check.C) {
app.cfg.TLSClientAuthMode = "invalid" _, isValid := LookupTLSClientAuthMode("invalid")
_, err := app.GetClientAuthMode() c.Assert(isValid, check.Equals, false)
c.Assert(err, check.NotNil)
} }
// Ensure that all client auth modes return a nil error. // Ensure that all client auth modes return a nil error.
@ -79,8 +78,7 @@ func (s *Suite) TestAuthModes(c *check.C) {
modes := []string{"disabled", "relaxed", "enforced"} modes := []string{"disabled", "relaxed", "enforced"}
for _, v := range modes { for _, v := range modes {
app.cfg.TLSClientAuthMode = v _, isValid := LookupTLSClientAuthMode(v)
_, err := app.GetClientAuthMode() c.Assert(isValid, check.Equals, true)
c.Assert(err, check.IsNil)
} }
} }

5
cmd/headscale/cli/utils.go
View File

@ -34,7 +34,6 @@ const (
) )
func LoadConfig(path string) error { func LoadConfig(path string) error {
viper.SetConfigName("config") viper.SetConfigName("config")
if path == "" { if path == "" {
viper.AddConfigPath("/etc/headscale/") viper.AddConfigPath("/etc/headscale/")
@ -295,7 +294,9 @@ func getHeadscaleConfig() headscale.Config {
Msgf("'ip_prefixes' not configured, falling back to default: %v", prefixes) Msgf("'ip_prefixes' not configured, falling back to default: %v", prefixes)
} }
tlsClientAuthMode, _ := headscale.LookupTLSClientAuthMode(viper.GetString("tls_client_auth_mode")) tlsClientAuthMode, _ := headscale.LookupTLSClientAuthMode(
viper.GetString("tls_client_auth_mode"),
)
return headscale.Config{ return headscale.Config{
ServerURL: viper.GetString("server_url"), ServerURL: viper.GetString("server_url"),