From 96c6ee01f8d410f4ef7b831527c4537e2962075f Mon Sep 17 00:00:00 2001 From: Ghobad Palvaneh Date: Thu, 17 Aug 2023 17:38:26 +0330 Subject: [PATCH] write test --- logger_test.go | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/logger_test.go b/logger_test.go index 5f78708f..d42a7d9f 100644 --- a/logger_test.go +++ b/logger_test.go @@ -414,6 +414,26 @@ func TestLoggerWithConfigSkippingPaths(t *testing.T) { assert.Contains(t, buffer.String(), "") } +func TestLoggerWithConfigSkipper(t *testing.T) { + buffer := new(strings.Builder) + router := New() + router.Use(LoggerWithConfig(LoggerConfig{ + Output: buffer, + Skip: func(c *Context) bool { + return c.Writer.Status() == http.StatusNoContent + }, + })) + router.GET("/logged", func(c *Context) { c.Status(http.StatusOK) }) + router.GET("/skipped", func(c *Context) { c.Status(http.StatusNoContent) }) + + PerformRequest(router, "GET", "/logged") + assert.Contains(t, buffer.String(), "200") + + buffer.Reset() + PerformRequest(router, "GET", "/skipped") + assert.Contains(t, buffer.String(), "") +} + func TestDisableConsoleColor(t *testing.T) { New() assert.Equal(t, autoColor, consoleColorMode)