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 multipart

import (

        "bytes"

        "io"

        "os"

        "regexp"

        "testing"

)

func TestReadForm(t *testing.T) {

        testBody := regexp.MustCompile("\n").ReplaceAllString(message, "\r\n")

        b := bytes.NewBufferString(testBody)

        r := NewReader(b, boundary)

        f, err := r.ReadForm(25)

        if err != nil {

                t.Fatal("ReadForm:", err)

        }

        defer f.RemoveAll()

        if g, e := f.Value["texta"][0], textaValue; g != e {

                t.Errorf("texta value = %q, want %q", g, e)

        }

        if g, e := f.Value["textb"][0], textbValue; g != e {

                t.Errorf("texta value = %q, want %q", g, e)

        }

        fd := testFile(t, f.File["filea"][0], "filea.txt", fileaContents)

        if _, ok := fd.(*os.File); ok {

                t.Error("file is *os.File, should not be")

        }

        fd.Close()

        fd = testFile(t, f.File["fileb"][0], "fileb.txt", filebContents)

        if _, ok := fd.(*os.File); !ok {

                t.Errorf("file has unexpected underlying type %T", fd)

        }

        fd.Close()

}

func testFile(t *testing.T, fh *FileHeader, efn, econtent string) File {

        if fh.Filename != efn {

                t.Errorf("filename = %q, want %q", fh.Filename, efn)

        }

        f, err := fh.Open()

        if err != nil {

                t.Fatal("opening file:", err)

        }

        b := new(bytes.Buffer)

        _, err = io.Copy(b, f)

        if err != nil {

                t.Fatal("copying contents:", err)

        }

        if g := b.String(); g != econtent {

                t.Errorf("contents = %q, want %q", g, econtent)

        }

        return f

}

const (

        fileaContents = "This is a test file."

        filebContents = "Another test file."

        textaValue    = "foo"

        textbValue    = "bar"

        boundary      = `MyBoundary`

)

const message = `

--MyBoundary

Content-Disposition: form-data; name="filea"; filename="filea.txt"

Content-Type: text/plain

` + fileaContents + `

--MyBoundary

Content-Disposition: form-data; name="fileb"; filename="fileb.txt"

Content-Type: text/plain

` + filebContents + `

--MyBoundary

Content-Disposition: form-data; name="texta"

` + textaValue + `

--MyBoundary

Content-Disposition: form-data; name="textb"

` + textbValue + `

--MyBoundary--

`