libgo/go/mime/multipart/formdata_test.go

   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 multipart
   6
   7 import (
   8         "bytes"
   9         "io"
  10         "os"
  11         "regexp"
  12         "strings"
  13         "testing"
  14 )
  15
  16 func TestReadForm(t *testing.T) {
  17         testBody := regexp.MustCompile("\n").ReplaceAllString(message, "\r\n")
  18         b := strings.NewReader(testBody)
  19         r := NewReader(b, boundary)
  20         f, err := r.ReadForm(25)
  21         if err != nil {
  22                 t.Fatal("ReadForm:", err)
  23         }
  24         defer f.RemoveAll()
  25         if g, e := f.Value["texta"][0], textaValue; g != e {
  26                 t.Errorf("texta value = %q, want %q", g, e)
  27         }
  28         if g, e := f.Value["textb"][0], textbValue; g != e {
  29                 t.Errorf("texta value = %q, want %q", g, e)
  30         }
  31         fd := testFile(t, f.File["filea"][0], "filea.txt", fileaContents)
  32         if _, ok := fd.(*os.File); ok {
  33                 t.Error("file is *os.File, should not be")
  34         }
  35         fd.Close()
  36         fd = testFile(t, f.File["fileb"][0], "fileb.txt", filebContents)
  37         if _, ok := fd.(*os.File); !ok {
  38                 t.Errorf("file has unexpected underlying type %T", fd)
  39         }
  40         fd.Close()
  41 }
  42
  43 func testFile(t *testing.T, fh *FileHeader, efn, econtent string) File {
  44         if fh.Filename != efn {
  45                 t.Errorf("filename = %q, want %q", fh.Filename, efn)
  46         }
  47         f, err := fh.Open()
  48         if err != nil {
  49                 t.Fatal("opening file:", err)
  50         }
  51         b := new(bytes.Buffer)
  52         _, err = io.Copy(b, f)
  53         if err != nil {
  54                 t.Fatal("copying contents:", err)
  55         }
  56         if g := b.String(); g != econtent {
  57                 t.Errorf("contents = %q, want %q", g, econtent)
  58         }
  59         return f
  60 }
  61
  62 const (
  63         fileaContents = "This is a test file."
  64         filebContents = "Another test file."
  65         textaValue    = "foo"
  66         textbValue    = "bar"
  67         boundary      = `MyBoundary`
  68 )
  69
  70 const message = `
  71 --MyBoundary
  72 Content-Disposition: form-data; name="filea"; filename="filea.txt"
  73 Content-Type: text/plain
  74
  75 ` + fileaContents + `
  76 --MyBoundary
  77 Content-Disposition: form-data; name="fileb"; filename="fileb.txt"
  78 Content-Type: text/plain
  79
  80 ` + filebContents + `
  81 --MyBoundary
  82 Content-Disposition: form-data; name="texta"
  83
  84 ` + textaValue + `
  85 --MyBoundary
  86 Content-Disposition: form-data; name="textb"
  87
  88 ` + textbValue + `
  89 --MyBoundary--
  90 `
  91
  92 func TestReadForm_NoReadAfterEOF(t *testing.T) {
  93         maxMemory := int64(32) << 20
  94         boundary := `---------------------------8d345eef0d38dc9`
  95         body := `
  96 -----------------------------8d345eef0d38dc9
  97 Content-Disposition: form-data; name="version"
  98
  99 171
 100 -----------------------------8d345eef0d38dc9--`
 101
 102         mr := NewReader(&failOnReadAfterErrorReader{t: t, r: strings.NewReader(body)}, boundary)
 103
 104         f, err := mr.ReadForm(maxMemory)
 105         if err != nil {
 106                 t.Fatal(err)
 107         }
 108         t.Logf("Got: %#v", f)
 109 }
 110
 111 // failOnReadAfterErrorReader is an io.Reader wrapping r.
 112 // It fails t if any Read is called after a failing Read.
 113 type failOnReadAfterErrorReader struct {
 114         t      *testing.T
 115         r      io.Reader
 116         sawErr error
 117 }
 118
 119 func (r *failOnReadAfterErrorReader) Read(p []byte) (n int, err error) {
 120         if r.sawErr != nil {
 121                 r.t.Fatalf("unexpected Read on Reader after previous read saw error %v", r.sawErr)
 122         }
 123         n, err = r.r.Read(p)
 124         r.sawErr = err
 125         return
 126 }