upstream-mattermost/server/config/logconfigsrc_test.go

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

package config

import (
	"encoding/json"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

var (
	validJSON        = []byte(`{"file":{ "Type":"file"}}`)
	badJSON          = []byte(`{"file":{ Type="file"}}`)
	validEscapedJSON = []byte(`"{\"file\":{ \"Type\":\"file\"}}"`)
	badEscapedJSON   = []byte(`"{\"file\":{ Type:\"file\"}}"`)
)

func TestNewLogConfigSrc(t *testing.T) {
	store := NewTestMemoryStore()
	require.NotNil(t, store)
	err := store.SetFile("advancedlogging.conf", validJSON)
	require.NoError(t, err)

	tests := []struct {
		name        string
		dsn         json.RawMessage
		configStore *Store
		wantErr     bool
		wantType    LogConfigSrc
	}{
		{name: "empty dsn", dsn: []byte(""), configStore: store, wantErr: true, wantType: nil},
		{name: "garbage dsn", dsn: []byte("!@wfejwcevioj"), configStore: store, wantErr: true, wantType: nil},
		{name: "valid json dsn", dsn: validJSON, configStore: store, wantErr: false, wantType: &jsonSrc{}},
		{name: "invalid json dsn", dsn: badJSON, configStore: store, wantErr: true, wantType: nil},
		{name: "valid escaped json dsn", dsn: validEscapedJSON, configStore: store, wantErr: false, wantType: &jsonSrc{}},
		{name: "invalid escaped json dsn", dsn: badEscapedJSON, configStore: store, wantErr: true, wantType: nil},
		{name: "valid filespec dsn", dsn: []byte("advancedlogging.conf"), configStore: store, wantErr: false, wantType: &fileSrc{}},
		{name: "invalid filespec dsn", dsn: []byte("/nobody/here.conf"), configStore: store, wantErr: true, wantType: nil},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := NewLogConfigSrc(tt.dsn, tt.configStore)
			if tt.wantErr {
				assert.Error(t, err)
			} else {
				assert.NoError(t, err)
				assert.IsType(t, tt.wantType, got)
			}
		})
	}
}