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.

// Tests that involve both reading and writing.

package zip

import (

        "bytes"

        "fmt"

        "io"

        "reflect"

        "testing"

        "time"

)

type stringReaderAt string

func (s stringReaderAt) ReadAt(p []byte, off int64) (n int, err error) {

        if off >= int64(len(s)) {

                return 0, io.EOF

        }

        n = copy(p, s[off:])

        return

}

func TestOver65kFiles(t *testing.T) {

        if testing.Short() {

                t.Logf("slow test; skipping")

                return

        }

        buf := new(bytes.Buffer)

        w := NewWriter(buf)

        const nFiles = (1 << 16) + 42

        for i := 0; i < nFiles; i++ {

                _, err := w.Create(fmt.Sprintf("%d.dat", i))

                if err != nil {

                        t.Fatalf("creating file %d: %v", i, err)

                }

        }

        if err := w.Close(); err != nil {

                t.Fatalf("Writer.Close: %v", err)

        }

        rat := stringReaderAt(buf.String())

        zr, err := NewReader(rat, int64(len(rat)))

        if err != nil {

                t.Fatalf("NewReader: %v", err)

        }

        if got := len(zr.File); got != nFiles {

                t.Fatalf("File contains %d files, want %d", got, nFiles)

        }

        for i := 0; i < nFiles; i++ {

                want := fmt.Sprintf("%d.dat", i)

                if zr.File[i].Name != want {

                        t.Fatalf("File(%d) = %q, want %q", i, zr.File[i].Name, want)

                }

        }

}

func TestModTime(t *testing.T) {

        var testTime = time.Date(2009, time.November, 10, 23, 45, 58, 0, time.UTC)

        fh := new(FileHeader)

        fh.SetModTime(testTime)

        outTime := fh.ModTime()

        if !outTime.Equal(testTime) {

                t.Errorf("times don't match: got %s, want %s", outTime, testTime)

        }

}

func TestFileHeaderRoundTrip(t *testing.T) {

        fh := &FileHeader{

                Name:             "foo.txt",

                UncompressedSize: 987654321,

                ModifiedTime:     1234,

                ModifiedDate:     5678,

        }

        fi := fh.FileInfo()

        fh2, err := FileInfoHeader(fi)

        // Ignore these fields:

        fh2.CreatorVersion = 0

        fh2.ExternalAttrs = 0

        if !reflect.DeepEqual(fh, fh2) {

                t.Errorf("mismatch\n input=%#v\noutput=%#v\nerr=%v", fh, fh2, err)

        }

        if sysfh, ok := fi.Sys().(*FileHeader); !ok && sysfh != fh {

                t.Errorf("Sys didn't return original *FileHeader")

        }

}