mirror of
https://github.com/golang/sys.git
synced 2026-02-08 11:46:04 +03:00
c7b8b68b14
These are useful for the same reason that the already existing GetSystemDirectory is. Change-Id: I3041ce6cbeb66a4f8a5960fbaf39381c8c9c80d6 Reviewed-on: https://go-review.googlesource.com/c/sys/+/191837 Run-TryBot: Jason A. Donenfeld <Jason@zx2c4.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Brad Fitzpatrick <bradfitz@golang.org>
82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
// 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 (
|
|
"strings"
|
|
"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")
|
|
}
|
|
}
|
|
|
|
func TestGetSystemDirectory(t *testing.T) {
|
|
d, err := windows.GetSystemDirectory()
|
|
if err != nil {
|
|
t.Fatalf("Failed to get system directory: %s", err)
|
|
}
|
|
if !strings.HasSuffix(strings.ToLower(d), "\\system32") {
|
|
t.Fatalf("System directory does not end in system32: %s", d)
|
|
}
|
|
}
|
|
|
|
func TestGetWindowsDirectory(t *testing.T) {
|
|
d1, err := windows.GetWindowsDirectory()
|
|
if err != nil {
|
|
t.Fatalf("Failed to get Windows directory: %s", err)
|
|
}
|
|
d2, err := windows.GetSystemWindowsDirectory()
|
|
if err != nil {
|
|
t.Fatalf("Failed to get system Windows directory: %s", err)
|
|
}
|
|
if !strings.HasSuffix(strings.ToLower(d1), `\windows`) {
|
|
t.Fatalf("Windows directory does not end in windows: %s", d1)
|
|
}
|
|
if !strings.HasSuffix(strings.ToLower(d2), `\windows`) {
|
|
t.Fatalf("System Windows directory does not end in windows: %s", d2)
|
|
}
|
|
}
|