summaryrefslogtreecommitdiffstats
path: root/src/goqtestlib/util_test.go
blob: 73d035b64506a49c749d69e555624c22faecc92e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
// Copyright (C) 2016 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
package goqtestlib

import (
	"os"
	"testing"
)

func TestEnvironmentSet(t *testing.T) {
	testVariable := "myVariable"
	os.Setenv(testVariable, "initialValue")

	if os.Getenv(testVariable) != "initialValue" {
		t.Fatal("Invalid initial environment variable value")
	}

	func() {
		defer SetEnvironmentVariableAndRestoreOnExit(testVariable, "tempValue")()
		if os.Getenv(testVariable) != "tempValue" {
			t.Fatal("Environment variable should be set to the temporary value here")
		}
	}()

	if os.Getenv(testVariable) != "initialValue" {
		t.Fatal("Environment variable not restored")
	}
}