code review 4635042: src/pkg/http/cgi: Handler.Dir for specify working direc...

src/pkg/http/cgi/host_test.go

 // Copyright 2011 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.
 
 // Tests for package cgi
 
 package cgi
 
 import (
         "bufio"
         "exec"
         "fmt"
         "http"
         "http/httptest"
         "os"
         "path/filepath"
         "strings"
         "testing"
+        "runtime"
 )
 
 func newRequest(httpreq string) *http.Request {
         buf := bufio.NewReader(strings.NewReader(httpreq))
         req, err := http.ReadRequest(buf)
         if err != nil {
                 panic("cgi: bogus http request in test: " + httpreq)
         }
         req.RemoteAddr = "1.2.3.4"
         return req
@@ skipping 267 matching lines @@
                 Root:                "/test.cgi",
                 PathLocationHandler: baseHandler,
         }
         expectedMap := map[string]string{
                 "basepath":   "/foo",
                 "remoteaddr": "1.2.3.4",
         }
         runCgiTest(t, h, "GET /test.cgi?loc=/foo HTTP/1.0\nHost: example.com\n\n", expectedMap)
 }
 
-func TestDir(t *testing.T) {
-        if skipTest(t) {
-                return
-        }
+func TestDirUnix(t *testing.T) {
+        if runtime.GOOS == "windows" {
+                return
+        }
+
         cwd, _ := os.Getwd()
         h := &Handler{
                 Path: "testdata/test.cgi",
                 Root: "/test.cgi",
                 Dir:  cwd,
         }
         expectedMap := map[string]string{
                 "cwd": cwd,
         }
         runCgiTest(t, h, "GET /test.cgi HTTP/1.0\nHost: example.com\n\n", expectedMap)
 
         cwd, _ = os.Getwd()
         cwd = filepath.Join(cwd, "testdata")
         h = &Handler{
                 Path: "testdata/test.cgi",
                 Root: "/test.cgi",
         }
         expectedMap = map[string]string{
                 "cwd": cwd,
         }
         runCgiTest(t, h, "GET /test.cgi HTTP/1.0\nHost: example.com\n\n", expectedMap)
-
-}
+}
+
+func TestDirWindows(t *testing.T) {
+        if runtime.GOOS != "windows" {
+                return
+        }
+
+        cgifile, _ := filepath.Abs("testdata/test.cgi")
+
+        var perl string
+        var err os.Error
+        perl, err = exec.LookPath("perl")
+        if err != nil {
+                return
+        }
+        perl, _ = filepath.Abs(perl)
+
+        cwd, _ := os.Getwd()
+        h := &Handler{
+                Path: perl,
+                Root: "/test.cgi",
+                Dir:  cwd,
+                Args: []string{cgifile},
+                Env:  []string{"SCRIPT_FILENAME=" + cgifile},
+        }
+        expectedMap := map[string]string{
+                "cwd": cwd,
+        }
+        runCgiTest(t, h, "GET /test.cgi HTTP/1.0\nHost: example.com\n\n", expectedMap)
+
+        // If not specify Dir on windows, working directory should be
+        // base directory of perl.
+        cwd, _ = filepath.Split(perl)
+        if cwd != "" && cwd[len(cwd)-1] == filepath.Separator {
+                cwd = cwd[:len(cwd)-1]
+        }
+        h = &Handler{
+                Path: perl,
+                Root: "/test.cgi",
+                Args: []string{cgifile},
+                Env:  []string{"SCRIPT_FILENAME=" + cgifile},
+        }
+        expectedMap = map[string]string{
+                "cwd": cwd,
+        }
+        runCgiTest(t, h, "GET /test.cgi HTTP/1.0\nHost: example.com\n\n", expectedMap)
+}