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/ioutil"
        "testing"
)

func TestWriter(t *testing.T) {
        fileContents := []byte("my file contents")

        var b bytes.Buffer
        w := NewWriter(&b)
        {
                part, err := w.CreateFormFile("myfile", "my-file.txt")
                if err != nil {
                        t.Fatalf("CreateFormFile: %v", err)
                }
                part.Write(fileContents)
                err = w.WriteField("key", "val")
                if err != nil {
                        t.Fatalf("WriteField: %v", err)
                }
                part.Write([]byte("val"))
                err = w.Close()
                if err != nil {
                        t.Fatalf("Close: %v", err)
                }
                s := b.String()
                if len(s) == 0 {
                        t.Fatal("String: unexpected empty result")
                }
                if s[0] == '\r' || s[0] == '\n' {
                        t.Fatal("String: unexpected newline")
                }
        }

        r := NewReader(&b, w.Boundary())

        part, err := r.NextPart()
        if err != nil {
                t.Fatalf("part 1: %v", err)
        }
        if g, e := part.FormName(), "myfile"; g != e {
                t.Errorf("part 1: want form name %q, got %q", e, g)
        }
        slurp, err := ioutil.ReadAll(part)
        if err != nil {
                t.Fatalf("part 1: ReadAll: %v", err)
        }
        if e, g := string(fileContents), string(slurp); e != g {
                t.Errorf("part 1: want contents %q, got %q", e, g)
        }

        part, err = r.NextPart()
        if err != nil {
                t.Fatalf("part 2: %v", err)
        }
        if g, e := part.FormName(), "key"; g != e {
                t.Errorf("part 2: want form name %q, got %q", e, g)
        }
        slurp, err = ioutil.ReadAll(part)
        if err != nil {
                t.Fatalf("part 2: ReadAll: %v", err)
        }
        if e, g := "val", string(slurp); e != g {
                t.Errorf("part 2: want contents %q, got %q", e, g)
        }

        part, err = r.NextPart()
        if part != nil || err == nil {
                t.Fatalf("expected end of parts; got %v, %v", part, err)
        }
}