| // 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" |
| "reflect" |
| "strings" |
| "testing" |
| "time" |
| ) |
| |
| 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) |
| } |
| s := buf.String() |
| zr, err := NewReader(strings.NewReader(s), int64(len(s))) |
| 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") |
| } |
| } |