github.com/ice-blockchain/go/src@v0.0.0-20240403114104-1564d284e521/net/http/filetransport_test.go (about)

     1  // Copyright 2011 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  package http
     6  
     7  import (
     8  	"io"
     9  	"os"
    10  	"path/filepath"
    11  	"testing"
    12  	"testing/fstest"
    13  )
    14  
    15  func checker(t *testing.T) func(string, error) {
    16  	return func(call string, err error) {
    17  		if err == nil {
    18  			return
    19  		}
    20  		t.Fatalf("%s: %v", call, err)
    21  	}
    22  }
    23  
    24  func TestFileTransport(t *testing.T) {
    25  	check := checker(t)
    26  
    27  	dname := t.TempDir()
    28  	fname := filepath.Join(dname, "foo.txt")
    29  	err := os.WriteFile(fname, []byte("Bar"), 0644)
    30  	check("WriteFile", err)
    31  	defer os.Remove(fname)
    32  
    33  	tr := &Transport{}
    34  	tr.RegisterProtocol("file", NewFileTransport(Dir(dname)))
    35  	c := &Client{Transport: tr}
    36  
    37  	fooURLs := []string{"file:///foo.txt", "file://../foo.txt"}
    38  	for _, urlstr := range fooURLs {
    39  		res, err := c.Get(urlstr)
    40  		check("Get "+urlstr, err)
    41  		if res.StatusCode != 200 {
    42  			t.Errorf("for %s, StatusCode = %d, want 200", urlstr, res.StatusCode)
    43  		}
    44  		if res.ContentLength != -1 {
    45  			t.Errorf("for %s, ContentLength = %d, want -1", urlstr, res.ContentLength)
    46  		}
    47  		if res.Body == nil {
    48  			t.Fatalf("for %s, nil Body", urlstr)
    49  		}
    50  		slurp, err := io.ReadAll(res.Body)
    51  		res.Body.Close()
    52  		check("ReadAll "+urlstr, err)
    53  		if string(slurp) != "Bar" {
    54  			t.Errorf("for %s, got content %q, want %q", urlstr, string(slurp), "Bar")
    55  		}
    56  	}
    57  
    58  	const badURL = "file://../no-exist.txt"
    59  	res, err := c.Get(badURL)
    60  	check("Get "+badURL, err)
    61  	if res.StatusCode != 404 {
    62  		t.Errorf("for %s, StatusCode = %d, want 404", badURL, res.StatusCode)
    63  	}
    64  	res.Body.Close()
    65  }
    66  
    67  func TestFileTransportFS(t *testing.T) {
    68  	check := checker(t)
    69  
    70  	fsys := fstest.MapFS{
    71  		"index.html": {Data: []byte("index.html says hello")},
    72  	}
    73  
    74  	tr := &Transport{}
    75  	tr.RegisterProtocol("file", NewFileTransportFS(fsys))
    76  	c := &Client{Transport: tr}
    77  
    78  	for fname, mfile := range fsys {
    79  		urlstr := "file:///" + fname
    80  		res, err := c.Get(urlstr)
    81  		check("Get "+urlstr, err)
    82  		if res.StatusCode != 200 {
    83  			t.Errorf("for %s, StatusCode = %d, want 200", urlstr, res.StatusCode)
    84  		}
    85  		if res.ContentLength != -1 {
    86  			t.Errorf("for %s, ContentLength = %d, want -1", urlstr, res.ContentLength)
    87  		}
    88  		if res.Body == nil {
    89  			t.Fatalf("for %s, nil Body", urlstr)
    90  		}
    91  		slurp, err := io.ReadAll(res.Body)
    92  		res.Body.Close()
    93  		check("ReadAll "+urlstr, err)
    94  		if string(slurp) != string(mfile.Data) {
    95  			t.Errorf("for %s, got content %q, want %q", urlstr, string(slurp), "Bar")
    96  		}
    97  	}
    98  
    99  	const badURL = "file://../no-exist.txt"
   100  	res, err := c.Get(badURL)
   101  	check("Get "+badURL, err)
   102  	if res.StatusCode != 404 {
   103  		t.Errorf("for %s, StatusCode = %d, want 404", badURL, res.StatusCode)
   104  	}
   105  	res.Body.Close()
   106  }