file_test.go

package golog

import (
	"io/ioutil"
	"strings"
	"sync"
	"testing"
	"time"
)

const file_prefix string = "golog_filetest"

func readLogFile(filename string) (loglines []string, err error) {
	file_data, err := ioutil.ReadFile(filename)
	if err != nil {
		return []string{}, err
	}

	log := string(file_data)
	log = strings.TrimSpace(log)
	return strings.Split(log, "\n"), nil
}

func TestDifferentPriorities(t *testing.T) {
	tmpfile, err := ioutil.TempFile("", file_prefix)
	if err != nil {
		t.Fatalf("Couldn't open tmp file: %s", err.Error())
	}

	proc := NewProcessorFromWriter(LOG_DEBUG, tmpfile, true)
	logger := NewLogger("file_test: ")
	logger.AddProcessor("file", proc)

	for _, p := range Priorities() {
		logger.Logf(p, "Hey, listen...")
	}

	d, _ := time.ParseDuration("100ms")
	time.Sleep(d)
	loglines, err := readLogFile(tmpfile.Name())
	if err != nil {
		t.Fatalf("Failed to read log file:  %s", err.Error())
	}

	if len(loglines) != len(Priorities()) {
		msg := "Unexpected number of log lines ouputed.  Expected %d, but was %d"
		t.Fatalf(msg, len(Priorities()), len(loglines))
	}

	for i, p := range Priorities() {
		expected := " " + p.ShortString() + ": file_test: Hey, listen..."
		if !strings.HasSuffix(loglines[i], expected) {
			t.Errorf("Unexpected log line.\nExpected: <date>%s\nBut was: %s", expected, loglines[i])
		}
	}
}

func TestConcurrentLogging(t *testing.T) {
	tmpfile, err := ioutil.TempFile("", "golog_conc_file_test")
	if err != nil {
		t.Fatalf("Couldn't create tmp file: %s", err.Error())
	}
	proc := NewProcessorFromWriter(LOG_DEBUG, tmpfile, true)
	logger := NewLogger("conc_test: ")
	logger.AddProcessor("file", proc)

	wg := &sync.WaitGroup{}
	num_routines := 5000
	for i := 0; i < num_routines; i++ {
		wg.Add(1)
		go func() {
			logger.Noticef("Help! I need somebody!")
			logger.Warningf("Help! Not just anybody!")
			logger.Errorf("Help! You know I need someone!")
			logger.Criticalf("Heeeeeeeeeeeeelp!")
			wg.Done()
		}()
	}
	wg.Wait()
	d, _ := time.ParseDuration("100ms")
	time.Sleep(d)

	loglines, err := readLogFile(tmpfile.Name())
	if err != nil {
		t.Fatalf("Error when reading tmp file:  %s", err.Error())
	}

	if len(loglines) != (num_routines * 4) {
		msg := "Unexpected number of log lines.  Expected %d, but was %d"
		t.Fatalf(msg, num_routines*4, len(loglines))
	}
}