mattermost/server/public/model/service_environment_test.go
Jesse Hallam 2d1848bf14
fix TestGetServiceEnvironment (#24506)
A late change to the service environment functionality had us defaulting to `dev` instead of `test` without the `production` build tag. Unfortunately, because these tests were unintentionally being skipped within the submodule, the failure wasn't caught at the time. A simple update addresses the unit tests.

Fixes: https://mattermost.atlassian.net/browse/MM-54197
2023-09-08 11:00:48 -03:00

47 lines
2 KiB
Go

package model_test
import (
"os"
"testing"
"github.com/mattermost/mattermost/server/public/model"
"github.com/stretchr/testify/require"
)
// TestGetServiceEnvironment verifies the semantics of the MM_SERVICEENVIRONMENT environment
// variable when explicitly configured as well as when left undefined or empty.
//
// To guard against accidental use of production keys (especially telemetry), all development and
// testing defaults to the test service environment, making it impossible to test the production
// semantics at the unit test level. Validating the default, enterprise service environment is left
// to smoketests before releasing.
func TestGetServiceEnvironment(t *testing.T) {
t.Run("no env defaults to dev (without production tag)", func(t *testing.T) {
require.Equal(t, model.ServiceEnvironmentDev, model.GetServiceEnvironment())
})
t.Run("empty string defaults to dev (without production tag)", func(t *testing.T) {
os.Setenv("MM_SERVICEENVIRONMENT", "")
defer os.Unsetenv("MM_SERVICEENVIRONMENT")
require.Equal(t, model.ServiceEnvironmentDev, model.GetServiceEnvironment())
})
t.Run("production", func(t *testing.T) {
os.Setenv("MM_SERVICEENVIRONMENT", "production")
defer os.Unsetenv("MM_SERVICEENVIRONMENT")
require.Equal(t, model.ServiceEnvironmentProduction, model.GetServiceEnvironment())
})
t.Run("test", func(t *testing.T) {
os.Setenv("MM_SERVICEENVIRONMENT", "test")
defer os.Unsetenv("MM_SERVICEENVIRONMENT")
require.Equal(t, model.ServiceEnvironmentTest, model.GetServiceEnvironment())
})
t.Run("dev", func(t *testing.T) {
os.Setenv("MM_SERVICEENVIRONMENT", "dev")
defer os.Unsetenv("MM_SERVICEENVIRONMENT")
require.Equal(t, model.ServiceEnvironmentDev, model.GetServiceEnvironment())
})
t.Run("whitespace and case insensitive", func(t *testing.T) {
os.Setenv("MM_SERVICEENVIRONMENT", " Test ")
defer os.Unsetenv("MM_SERVICEENVIRONMENT")
require.Equal(t, model.ServiceEnvironmentTest, model.GetServiceEnvironment())
})
}