friendica-exporter/internal/config/config_test.go

package config

import (
	"errors"
	"testing"
	"time"

	"friendica-exporter/internal/testutil"
	"github.com/google/go-cmp/cmp"
)

func testEnv(env map[string]string) func(string) string {
	return func(key string) string {
		return env[key]
	}
}

func TestConfig(t *testing.T) {
	defaults := defaultConfig()
	tt := []struct {
		desc       string
		args       []string
		env        map[string]string
		wantErr    error
		wantConfig Config
	}{
		{
			desc: "flags",
			args: []string{
				"test",
				"--addr",
				"127.0.0.1:9205",
				"--timeout",
				"30s",
				"--server",
				"http://localhost",
				"--stats-token",
				"/$&/bdgersert6756!",
			},
			env:     map[string]string{},
			wantErr: nil,
			wantConfig: Config{
				ListenAddr:    "127.0.0.1:9205",
				Timeout:       30 * time.Second,
				ServerURL:     "http://localhost",
				StatsToken:    "/$&/bdgersert6756!",
				TLSSkipVerify: false,
			},
		},
		{
			desc: "password from file",
			args: []string{
				"test",
				"--server",
				"http://localhost",
				"--stats-token",
				"@testdata/statsToken",
			},
			env:     map[string]string{},
			wantErr: nil,
			wantConfig: Config{
				ListenAddr:    defaults.ListenAddr,
				Timeout:       defaults.Timeout,
				ServerURL:     "http://localhost",
				StatsToken:    "234563twtwewr",
				TLSSkipVerify: false,
			},
		},
		{
			desc: "config from file",
			args: []string{
				"test",
				"--config-file",
				"testdata/all.yml",
			},
			env:     map[string]string{},
			wantErr: nil,
			wantConfig: Config{
				ListenAddr:    "127.0.0.10:9205",
				Timeout:       10 * time.Second,
				ServerURL:     "http://localhost",
				StatsToken:    "12w4352345zt§&/&)(&/",
				TLSSkipVerify: false,
			},
		},
		{
			desc: "config and token from file",
			args: []string{
				"test",
				"--config-file",
				"testdata/statsTokenFile.yml",
			},
			env:     map[string]string{},
			wantErr: nil,
			wantConfig: Config{
				ListenAddr:    "127.0.0.10:9205",
				Timeout:       10 * time.Second,
				ServerURL:     "http://localhost",
				StatsToken:    "234563twtwewr",
				TLSSkipVerify: false,
			},
		},
		{
			desc: "don't check tls certificates",
			args: []string{
				"test",
				"--tls-skip-verify",
				"true",
			},
			env:     map[string]string{},
			wantErr: nil,
			wantConfig: Config{
				ListenAddr:    ":9205",
				Timeout:       5 * time.Second,
				ServerURL:     "",
				StatsToken:    "",
				TLSSkipVerify: true,
			},
		},
		{
			desc: "env config",
			args: []string{
				"test",
			},
			env: map[string]string{
				envListenAddress: "127.0.0.11:9205",
				envTimeout:       "15s",
				envServerURL:     "http://localhost",
				envStatsToken:    "testpass/()/(g21!",
				envTLSSkipVerify: "true",
			},
			wantErr: nil,
			wantConfig: Config{
				ListenAddr:    "127.0.0.11:9205",
				Timeout:       15 * time.Second,
				ServerURL:     "http://localhost",
				StatsToken:    "testpass/()/(g21!",
				TLSSkipVerify: true,
			},
		},
		{
			desc: "minimal env",
			args: []string{
				"test",
			},
			env: map[string]string{
				envServerURL:  "http://localhost",
				envStatsToken: "testpass/()/(g21!",
			},
			wantErr: nil,
			wantConfig: Config{
				ListenAddr:    defaults.ListenAddr,
				Timeout:       defaults.Timeout,
				ServerURL:     "http://localhost",
				StatsToken:    "testpass/()/(g21!",
				TLSSkipVerify: false,
			},
		},
		{
			desc: "show help",
			args: []string{
				"test",
				"--help",
			},
			env:     map[string]string{},
			wantErr: nil,
			wantConfig: Config{
				RunMode: RunModeHelp,
			},
		},
		{
			desc: "show version",
			args: []string{
				"test",
				"--version",
			},
			env:     map[string]string{},
			wantErr: nil,
			wantConfig: Config{
				RunMode: RunModeVersion,
			},
		},
		{
			desc: "wrongflag",
			args: []string{
				"test",
				"--unknown",
			},
			env:     map[string]string{},
			wantErr: errors.New("error parsing flags: unknown flag: --unknown"),
		},
		{
			desc: "env wrong duration",
			args: []string{
				"test",
			},
			env: map[string]string{
				"FRIENDICA_TIMEOUT": "unknown",
			},
			wantErr: errors.New("error reading environment variables: time: invalid duration \"unknown\""),
		},
		{
			desc: "token from file error",
			args: []string{
				"test",
				"--server",
				"http://localhost",
				"--stats-token",
				"@testdata/notfound",
			},
			env:     map[string]string{},
			wantErr: errors.New("can not read stats token file: open testdata/notfound: no such file or directory"),
		},
		{
			desc: "config from file error",
			args: []string{
				"test",
				"--config-file",
				"testdata/notfound.yml",
			},
			env:     map[string]string{},
			wantErr: errors.New("error reading configuration file: open testdata/notfound.yml: no such file or directory"),
		},
		{
			desc: "fail parsing tlsSkipVerify env",
			args: []string{
				"test",
			},
			env: map[string]string{
				envTLSSkipVerify: "invalid",
			},
			wantErr: errors.New(`error reading environment variables: can not parse value for "FRIENDICA_TLS_SKIP_VERIFY": invalid`),
		},
	}

	for _, tc := range tt {
		tc := tc
		t.Run(tc.desc, func(t *testing.T) {
			t.Parallel()

			config, err := parseConfig(tc.args, testEnv(tc.env))

			if !testutil.EqualErrorMessage(err, tc.wantErr) {
				t.Errorf("got error %q, want %q", err, tc.wantErr)
			}

			if err != nil {
				return
			}

			if diff := cmp.Diff(config, tc.wantConfig); diff != "" {
				t.Errorf("config differs: -got +want\n%s", diff)
			}
		})
	}
}

func TestConfigValidate(t *testing.T) {
	tt := []struct {
		desc    string
		config  Config
		wantErr error
	}{
		{
			desc: "minimal",
			config: Config{
				ServerURL:  "https://example.com",
				StatsToken: "testtoken",
			},
			wantErr: nil,
		},
		{
			desc: "no url",
			config: Config{
				StatsToken: "testtoken",
			},
			wantErr: errValidateNoServerURL,
		},
		{
			desc: "auth help",
			config: Config{
				ServerURL: "https://example.com",
			},
			wantErr: errValidateNoStatsToken,
		},
	}

	for _, tc := range tt {
		tc := tc
		t.Run(tc.desc, func(t *testing.T) {
			t.Parallel()

			err := tc.config.Validate()

			if !testutil.EqualErrorMessage(err, tc.wantErr) {
				t.Errorf("got error %q, want %q", err, tc.wantErr)
			}
		})
	}
}