summaryrefslogtreecommitdiffstats
path: root/vendor/golang.org/x/sys/windows/syscall_test.go
blob: d7009e44a5715d900b3f8dadd16d6a4af82647c7 (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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
// Copyright 2013 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

// +build windows

package windows_test

import (
	"syscall"
	"testing"

	"golang.org/x/sys/windows"
)

func testSetGetenv(t *testing.T, key, value string) {
	err := windows.Setenv(key, value)
	if err != nil {
		t.Fatalf("Setenv failed to set %q: %v", value, err)
	}
	newvalue, found := windows.Getenv(key)
	if !found {
		t.Fatalf("Getenv failed to find %v variable (want value %q)", key, value)
	}
	if newvalue != value {
		t.Fatalf("Getenv(%v) = %q; want %q", key, newvalue, value)
	}
}

func TestEnv(t *testing.T) {
	testSetGetenv(t, "TESTENV", "AVALUE")
	// make sure TESTENV gets set to "", not deleted
	testSetGetenv(t, "TESTENV", "")
}

func TestGetProcAddressByOrdinal(t *testing.T) {
	// Attempt calling shlwapi.dll:IsOS, resolving it by ordinal, as
	// suggested in
	// https://msdn.microsoft.com/en-us/library/windows/desktop/bb773795.aspx
	h, err := windows.LoadLibrary("shlwapi.dll")
	if err != nil {
		t.Fatalf("Failed to load shlwapi.dll: %s", err)
	}
	procIsOS, err := windows.GetProcAddressByOrdinal(h, 437)
	if err != nil {
		t.Fatalf("Could not find shlwapi.dll:IsOS by ordinal: %s", err)
	}
	const OS_NT = 1
	r, _, _ := syscall.Syscall(procIsOS, 1, OS_NT, 0, 0)
	if r == 0 {
		t.Error("shlwapi.dll:IsOS(OS_NT) returned 0, expected non-zero value")
	}
}