test case additions

Author: jeevatkm

log_test.go

@@ -6,26 +6,38 @@ package log
 
 import (
 	"fmt"
+	"os"
+	"path/filepath"
 	"strings"
 	"testing"
 )
 
 func TestDefaultStandardLogger(t *testing.T) {
+	SetLevel(LevelInfo)
+	_ = SetPattern("%time:2006-01-02 15:04:05.000 %level:-5 %line %custom:- %message")
 	Trace("I shoudn't see this msg, because standard logger level is DEBUG")
 	Debug("I would like to see this message, debug is useful for dev")
 	Info("Yes, I would love to")
 	Warn("Yes, yes it's an warning")
 	Error("Yes, yes, yes - finally an error")
+	fmt.Println()
 
-	fmt.Printf("First round: %#v\n\n", Stats())
+	t.Logf("First round: %#v\n\n", Stats())
 
+	SetLevel(LevelDebug)
+	_ = SetPattern("%time:2006-01-02 15:04:05.000 %level:-5 %shortfile %line %custom:- %message")
 	Tracef("I shoudn't see this msg: %v", 4)
 	Debugf("I would like to see this message, debug is useful for dev: %v", 3)
 	Infof("Yes, I would love to: %v", 2)
 	Warnf("Yes, yes it's an warning: %v", 1)
 	Errorf("Yes, yes, yes - finally an error: %v", 0)
 
-	fmt.Printf("Second round: %#v\n\n", Stats())
+	t.Logf("Second round: %#v\n\n", Stats())
+
+	err := SetPattern("%level:-5 %shortfile %line %unknown")
+	if err == nil {
+		t.Error("Expected error got nil")
+	}
 }
 
 func TestNewCustomUTCConsoleReceiver(t *testing.T) {
@@ -38,7 +50,7 @@ receiver = "console"
 level = "debug"
 
 # if not suppiled then default pattern is used
-pattern = "%utctime:2006-01-02 15:04:05.000 %level:-5 %shortfile:-25 %custom:- %message"
+pattern = "%utctime:2006-01-02 15:04:05.000 %level:-5 %line %shortfile:-25 %custom:- %message"
  `
 	logger, err := New(config)
 	if err != nil {
@@ -53,9 +65,9 @@ pattern = "%utctime:2006-01-02 15:04:05.000 %level:-5 %shortfile:-25 %custom:- %
 	logger.Error("Yes, yes, yes - finally an error")
 
 	stats := logger.Stats()
-	fmt.Printf("First round: %#v\n\n", stats)
-	if stats.bytes != 417 {
-		t.Errorf("Expected: 417, got: %v", stats.bytes)
+	t.Logf("First round: %#v\n", stats)
+	if stats.bytes != 433 {
+		t.Errorf("Expected: 433, got: %v\n", stats.bytes)
 	}
 
 	logger.Tracef("I shoudn't see this msg, because standard logger level is DEBUG: %v", 4)
@@ -65,18 +77,18 @@ pattern = "%utctime:2006-01-02 15:04:05.000 %level:-5 %shortfile:-25 %custom:- %
 	logger.Errorf("Yes, yes, yes - finally an error: %v", 0)
 
 	stats = logger.Stats()
-	fmt.Printf("Second round: %#v\n\n", stats)
-	if stats.bytes != 846 {
-		t.Errorf("Expected: 846, got: %v", stats.bytes)
+	t.Logf("Second round: %#v\n", stats)
+	if stats.bytes != 878 {
+		t.Errorf("Expected: 878, got: %v\n", stats.bytes)
 	}
 
 	Tracef("I shoudn't see this msg: %v", 46583)
 	Debugf("I would like to see this message, debug is useful for dev: %v", 334545)
 
 	stats = logger.Stats()
-	fmt.Printf("Third round: %#v\n\n", stats)
-	if stats.bytes != 846 {
-		t.Errorf("Expected: 846, got: %v", stats.bytes)
+	t.Logf("Third round: %#v\n", stats)
+	if stats.bytes != 878 {
+		t.Errorf("Expected: 878, got: %v\n", stats.bytes)
 	}
 }
 
@@ -99,7 +111,7 @@ receiver = "CONSOLE"
 	logger.Error("Yes, yes, yes - finally an error")
 
 	stats := logger.Stats()
-	fmt.Printf("First round: %#v\n\n", stats)
+	t.Logf("First round: %#v\n", stats)
 	if stats.bytes != 313 {
 		t.Errorf("Expected: 313, got: %v", stats.bytes)
 	}
@@ -111,7 +123,7 @@ receiver = "CONSOLE"
 	logger.Errorf("Yes, yes, yes - finally an error: %v", 0)
 
 	stats = logger.Stats()
-	fmt.Printf("Second round: %#v\n\n", stats)
+	t.Logf("Second round: %#v\n", stats)
 	if stats.bytes != 638 {
 		t.Errorf("Expected: 638, got: %v", stats.bytes)
 	}
@@ -120,13 +132,71 @@ receiver = "CONSOLE"
 	Debugf("I would like to see this message, debug is useful for dev: %v", 334545)
 
 	stats = logger.Stats()
-	fmt.Printf("Third round: %#v\n\n", stats)
+	t.Logf("Third round: %#v\n", stats)
 	if stats.bytes != 638 {
 		t.Errorf("Expected: 638, got: %v", stats.bytes)
 	}
 }
 
+func TestNewCustomFileReceiverDailyRotation(t *testing.T) {
+	defer cleaupFiles("*.log")
+
+	fileLoggerConfig := `
+# file logger configuration
+# "FILE" uppercasse works too
+receiver = "file"
+
+# "debug" lowercase works too and if not supplied then defaults to DEBUG
+level = "info"
+
+# if not suppiled then default pattern is used
+pattern = "%utctime:2006-01-02 15:04:05.000 %level:-5 %longfile %line %custom:- %message"
+
+file = "daily-aah-filename.log"
+
+rotate {
+	mode = "daily"
+}
+ `
+
+	logger, err := New(fileLoggerConfig)
+	if err != nil {
+		t.Errorf("Unexpected error: %v", err)
+		t.FailNow()
+	}
+
+	for i := 0; i < 25; i++ {
+		logger.Trace("I shoudn't see this msg, because standard logger level is DEBUG")
+		logger.Debug("I would like to see this message, debug is useful for dev")
+		logger.Info("Yes, I would love to")
+		logger.Warn("Yes, yes it's an warning")
+		logger.Error("Yes, yes, yes - finally an error")
+	}
+
+	_ = logger.SetPattern("%time:2006-01-02 15:04:05.000 %level:-5 %longfile %line %custom:- %message")
+	for i := 0; i < 25; i++ {
+		logger.Tracef("I shoudn't see this msg, because standard logger level is DEBUG: %v", 4)
+		logger.Debugf("I would like to see this message, debug is useful for dev: %v", 3)
+		logger.Infof("Yes, I would love to: %v", 2)
+		logger.Warnf("Yes, yes it's an warning: %v", 1)
+		logger.Errorf("Yes, yes, yes - finally an error: %v", 0)
+	}
+
+	// Close scenario
+	logger.Close()
+	if !logger.Closed() {
+		t.Errorf("Expected 'true', got %v", logger.Closed())
+	}
+
+	logger.Info("This won't be written to file")
+
+	// once again
+	logger.Close()
+}
+
 func TestNewCustomFileReceiverLinesRotation(t *testing.T) {
+	defer cleaupFiles("*.log")
+
 	fileLoggerConfig := `
 # file logger configuration
 # "FILE" uppercasse works too
@@ -173,6 +243,8 @@ rotate {
 }
 
 func TestNewCustomFileReceiverSizeRotation(t *testing.T) {
+	defer cleaupFiles("*.log")
+
 	fileLoggerConfig := `
 # file logger configuration
 # "FILE" uppercasse works too
@@ -267,3 +339,52 @@ func TestNewMisc(t *testing.T) {
 		t.FailNow()
 	}
 }
+
+func TestLevelUnknown(t *testing.T) {
+	var level Level
+	if level.String() != "ERROR" {
+		t.Errorf("Expected level 'ERROR', got '%v'", level)
+	}
+
+	level = 9 // Unknown log level
+	if level.String() != "Unknown" {
+		t.Errorf("Expected level 'Unknown', got '%v'", level)
+	}
+}
+
+func TestStats(t *testing.T) {
+	stats := ReceiverStats{
+		lines: 200,
+		bytes: 764736,
+	}
+
+	if stats.Bytes() != 764736 {
+		t.Errorf("Expected '764736' bytes, got '%v' bytes", stats.Bytes())
+	}
+
+	if stats.Lines() != 200 {
+		t.Errorf("Expected '200' lines, got '%v' lines", stats.Lines())
+	}
+}
+
+func cleaupFiles(match string) {
+	pwd, _ := os.Getwd()
+
+	dir, err := os.Open(pwd)
+	if err != nil {
+		return
+	}
+
+	infos, err := dir.Readdir(-1)
+	if err != nil {
+		return
+	}
+
+	for _, info := range infos {
+		if !info.IsDir() {
+			if found, _ := filepath.Match(match, info.Name()); found {
+				_ = os.Remove(filepath.Join(pwd, info.Name()))
+			}
+		}
+	}
+}

stats.go

@@ -1,5 +1,5 @@
 // Copyright (c) 2016 Jeevanandam M (https://github.com/jeevatkm)
-// resty source code and usage is governed by a MIT style
+// go-aah/log source code and usage is governed by a MIT style
 // license that can be found in the LICENSE file.
 
 package log