123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212 |
- // Copyright 2017 Google Inc. All rights reserved.
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- package logger
- import (
- "bytes"
- "fmt"
- "io"
- "io/ioutil"
- "os"
- "os/exec"
- "path/filepath"
- "reflect"
- "sort"
- "syscall"
- "testing"
- )
- func TestCreateFileWithRotation(t *testing.T) {
- dir, err := ioutil.TempDir("", "test-rotation")
- if err != nil {
- t.Fatalf("Failed to get TempDir: %v", err)
- }
- defer os.RemoveAll(dir)
- file := filepath.Join(dir, "build.log")
- writeFile := func(name string, data string) {
- f, err := CreateFileWithRotation(name, 3)
- if err != nil {
- t.Fatalf("Failed to create file: %v", err)
- }
- if n, err := io.WriteString(f, data); err == nil && n < len(data) {
- t.Fatalf("Short write")
- } else if err != nil {
- t.Fatalf("Failed to write: %v", err)
- }
- if err := f.Close(); err != nil {
- t.Fatalf("Failed to close: %v", err)
- }
- }
- writeFile(file, "a")
- writeFile(file, "b")
- writeFile(file, "c")
- writeFile(file, "d")
- writeFile(file, "e")
- d, err := os.Open(dir)
- if err != nil {
- t.Fatalf("Failed to open dir: %v", err)
- }
- names, err := d.Readdirnames(0)
- if err != nil {
- t.Fatalf("Failed to read dir: %v", err)
- }
- sort.Strings(names)
- expected := []string{".lock_build.log", "build.1.log", "build.2.log", "build.3.log", "build.log"}
- if !reflect.DeepEqual(names, expected) {
- t.Errorf("File list does not match.")
- t.Errorf(" got: %v", names)
- t.Errorf("expected: %v", expected)
- t.FailNow()
- }
- expectFileContents := func(name, expected string) {
- data, err := ioutil.ReadFile(filepath.Join(dir, name))
- if err != nil {
- t.Errorf("Error reading file: %v", err)
- return
- }
- str := string(data)
- if str != expected {
- t.Errorf("Contents of %v does not match.", name)
- t.Errorf(" got: %v", data)
- t.Errorf("expected: %v", expected)
- }
- }
- expectFileContents("build.log", "e")
- expectFileContents("build.1.log", "d")
- expectFileContents("build.2.log", "c")
- expectFileContents("build.3.log", "b")
- }
- func TestPanic(t *testing.T) {
- if os.Getenv("ACTUALLY_PANIC") == "1" {
- panicValue := "foo"
- log := New(&bytes.Buffer{})
- defer func() {
- p := recover()
- if p == panicValue {
- os.Exit(42)
- } else {
- fmt.Fprintf(os.Stderr, "Expected %q, got %v\n", panicValue, p)
- os.Exit(3)
- }
- }()
- defer log.Cleanup()
- log.Panic(panicValue)
- os.Exit(2)
- return
- }
- // Run this in an external process so that we don't pollute stderr
- cmd := exec.Command(os.Args[0], "-test.run=TestPanic")
- cmd.Env = append(os.Environ(), "ACTUALLY_PANIC=1")
- err := cmd.Run()
- if e, ok := err.(*exec.ExitError); ok && e.Sys().(syscall.WaitStatus).ExitStatus() == 42 {
- return
- }
- t.Errorf("Expected process to exit with status 42, got %v", err)
- }
- func TestFatal(t *testing.T) {
- if os.Getenv("ACTUALLY_FATAL") == "1" {
- log := New(&bytes.Buffer{})
- defer func() {
- // Shouldn't get here
- os.Exit(3)
- }()
- defer log.Cleanup()
- log.Fatal("Test")
- os.Exit(0)
- return
- }
- cmd := exec.Command(os.Args[0], "-test.run=TestFatal")
- cmd.Env = append(os.Environ(), "ACTUALLY_FATAL=1")
- err := cmd.Run()
- if e, ok := err.(*exec.ExitError); ok && e.Sys().(syscall.WaitStatus).ExitStatus() == 1 {
- return
- }
- t.Errorf("Expected process to exit with status 1, got %v", err)
- }
- func TestNonFatal(t *testing.T) {
- if os.Getenv("ACTUAL_TEST") == "1" {
- log := New(&bytes.Buffer{})
- defer log.Cleanup()
- log.Println("Test")
- return
- }
- cmd := exec.Command(os.Args[0], "-test.run=TestNonFatal")
- cmd.Env = append(os.Environ(), "ACTUAL_TEST=1")
- err := cmd.Run()
- if e, ok := err.(*exec.ExitError); ok || (ok && !e.Success()) {
- t.Errorf("Expected process to exit cleanly, got %v", err)
- }
- }
- func TestRecoverFatal(t *testing.T) {
- log := New(&bytes.Buffer{})
- defer func() {
- if p := recover(); p != nil {
- t.Errorf("Unexpected panic: %#v", p)
- }
- }()
- defer Recover(func(err error) {
- if err.Error() != "Test" {
- t.Errorf("Expected %q, but got %q", "Test", err.Error())
- }
- })
- log.Fatal("Test")
- t.Errorf("Should not get here")
- }
- func TestRecoverNonFatal(t *testing.T) {
- log := New(&bytes.Buffer{})
- defer func() {
- if p := recover(); p == nil {
- t.Errorf("Panic not thrown")
- } else if p != "Test" {
- t.Errorf("Expected %q, but got %#v", "Test", p)
- }
- }()
- defer Recover(func(err error) {
- t.Errorf("Recover function should not be called")
- })
- log.Panic("Test")
- t.Errorf("Should not get here")
- }
- func TestRuntimePanic(t *testing.T) {
- defer func() {
- if p := recover(); p == nil {
- t.Errorf("Panic not thrown")
- }
- }()
- defer Recover(func(err error) {
- t.Errorf("Recover function should not be called")
- })
- var i *int
- *i = 0
- t.Errorf("Should not get here")
- }
|