Subversion Repositories openrisc

// 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.

package http_test

import (
        "io/ioutil"
        "net/http"
        "os"
        "path/filepath"
        "testing"
)

func checker(t *testing.T) func(string, error) {
        return func(call string, err error) {
                if err == nil {
                        return
                }
                t.Fatalf("%s: %v", call, err)
        }
}

func TestFileTransport(t *testing.T) {
        check := checker(t)

        dname, err := ioutil.TempDir("", "")
        check("TempDir", err)
        fname := filepath.Join(dname, "foo.txt")
        err = ioutil.WriteFile(fname, []byte("Bar"), 0644)
        check("WriteFile", err)
        defer os.Remove(dname)
        defer os.Remove(fname)

        tr := &http.Transport{}
        tr.RegisterProtocol("file", http.NewFileTransport(http.Dir(dname)))
        c := &http.Client{Transport: tr}

        fooURLs := []string{"file:///foo.txt", "file://../foo.txt"}
        for _, urlstr := range fooURLs {
                res, err := c.Get(urlstr)
                check("Get "+urlstr, err)
                if res.StatusCode != 200 {
                        t.Errorf("for %s, StatusCode = %d, want 200", urlstr, res.StatusCode)
                }
                if res.ContentLength != -1 {
                        t.Errorf("for %s, ContentLength = %d, want -1", urlstr, res.ContentLength)
                }
                if res.Body == nil {
                        t.Fatalf("for %s, nil Body", urlstr)
                }
                slurp, err := ioutil.ReadAll(res.Body)
                check("ReadAll "+urlstr, err)
                if string(slurp) != "Bar" {
                        t.Errorf("for %s, got content %q, want %q", urlstr, string(slurp), "Bar")
                }
        }

        const badURL = "file://../no-exist.txt"
        res, err := c.Get(badURL)
        check("Get "+badURL, err)
        if res.StatusCode != 404 {
                t.Errorf("for %s, StatusCode = %d, want 404", badURL, res.StatusCode)
        }
}