1// Copyright 2013 The Go Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style 3// license that can be found in the LICENSE file. 4 5// +build windows 6 7package windows_test 8 9import ( 10 "strings" 11 "syscall" 12 "testing" 13 14 "golang.org/x/sys/windows" 15) 16 17func testSetGetenv(t *testing.T, key, value string) { 18 err := windows.Setenv(key, value) 19 if err != nil { 20 t.Fatalf("Setenv failed to set %q: %v", value, err) 21 } 22 newvalue, found := windows.Getenv(key) 23 if !found { 24 t.Fatalf("Getenv failed to find %v variable (want value %q)", key, value) 25 } 26 if newvalue != value { 27 t.Fatalf("Getenv(%v) = %q; want %q", key, newvalue, value) 28 } 29} 30 31func TestEnv(t *testing.T) { 32 testSetGetenv(t, "TESTENV", "AVALUE") 33 // make sure TESTENV gets set to "", not deleted 34 testSetGetenv(t, "TESTENV", "") 35} 36 37func TestGetProcAddressByOrdinal(t *testing.T) { 38 // Attempt calling shlwapi.dll:IsOS, resolving it by ordinal, as 39 // suggested in 40 // https://msdn.microsoft.com/en-us/library/windows/desktop/bb773795.aspx 41 h, err := windows.LoadLibrary("shlwapi.dll") 42 if err != nil { 43 t.Fatalf("Failed to load shlwapi.dll: %s", err) 44 } 45 procIsOS, err := windows.GetProcAddressByOrdinal(h, 437) 46 if err != nil { 47 t.Fatalf("Could not find shlwapi.dll:IsOS by ordinal: %s", err) 48 } 49 const OS_NT = 1 50 r, _, _ := syscall.Syscall(procIsOS, 1, OS_NT, 0, 0) 51 if r == 0 { 52 t.Error("shlwapi.dll:IsOS(OS_NT) returned 0, expected non-zero value") 53 } 54} 55 56func TestGetSystemDirectory(t *testing.T) { 57 d, err := windows.GetSystemDirectory() 58 if err != nil { 59 t.Fatalf("Failed to get system directory: %s", err) 60 } 61 if !strings.HasSuffix(strings.ToLower(d), "\\system32") { 62 t.Fatalf("System directory does not end in system32: %s", d) 63 } 64} 65