libgo/go/io/ioutil/ioutil_test.go

   1 // Copyright 2009 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 ioutil_test
   6
   7 import (
   8         . "io/ioutil"
   9         "os"
  10         "testing"
  11 )
  12
  13 func checkSize(t *testing.T, path string, size int64) {
  14         dir, err := os.Stat(path)
  15         if err != nil {
  16                 t.Fatalf("Stat %q (looking for size %d): %s", path, size, err)
  17         }
  18         if dir.Size != size {
  19                 t.Errorf("Stat %q: size %d want %d", path, dir.Size, size)
  20         }
  21 }
  22
  23 func TestReadFile(t *testing.T) {
  24         filename := "rumpelstilzchen"
  25         contents, err := ReadFile(filename)
  26         if err == nil {
  27                 t.Fatalf("ReadFile %s: error expected, none found", filename)
  28         }
  29
  30         filename = "ioutil_test.go"
  31         contents, err = ReadFile(filename)
  32         if err != nil {
  33                 t.Fatalf("ReadFile %s: %v", filename, err)
  34         }
  35
  36         checkSize(t, filename, int64(len(contents)))
  37 }
  38
  39 func TestWriteFile(t *testing.T) {
  40         filename := "_test/rumpelstilzchen"
  41         data := "Programming today is a race between software engineers striving to " +
  42                 "build bigger and better idiot-proof programs, and the Universe trying " +
  43                 "to produce bigger and better idiots. So far, the Universe is winning."
  44
  45         if err := WriteFile(filename, []byte(data), 0644); err != nil {
  46                 t.Fatalf("WriteFile %s: %v", filename, err)
  47         }
  48
  49         contents, err := ReadFile(filename)
  50         if err != nil {
  51                 t.Fatalf("ReadFile %s: %v", filename, err)
  52         }
  53
  54         if string(contents) != data {
  55                 t.Fatalf("contents = %q\nexpected = %q", string(contents), data)
  56         }
  57
  58         // cleanup
  59         os.Remove(filename) // ignore error
  60 }
  61
  62
  63 func TestReadDir(t *testing.T) {
  64         dirname := "rumpelstilzchen"
  65         _, err := ReadDir(dirname)
  66         if err == nil {
  67                 t.Fatalf("ReadDir %s: error expected, none found", dirname)
  68         }
  69
  70         dirname = "."
  71         list, err := ReadDir(dirname)
  72         if err != nil {
  73                 t.Fatalf("ReadDir %s: %v", dirname, err)
  74         }
  75
  76         foundTest := false
  77         foundTestDir := false
  78         for _, dir := range list {
  79                 switch {
  80                 case dir.IsRegular() && dir.Name == "ioutil_test.go":
  81                         foundTest = true
  82                 case dir.IsDirectory() && dir.Name == "_test":
  83                         foundTestDir = true
  84                 }
  85         }
  86         if !foundTest {
  87                 t.Fatalf("ReadDir %s: test file not found", dirname)
  88         }
  89         if !foundTestDir {
  90                 t.Fatalf("ReadDir %s: _test directory not found", dirname)
  91         }
  92 }