From 08eb271a4ca257aaff5024e6bbd844a0182f0f78 Mon Sep 17 00:00:00 2001 From: kim <89579420+NyaaaWhatsUpDoc@users.noreply.github.com> Date: Sat, 7 May 2022 16:46:59 +0100 Subject: [performance] improved logrus output switching performance (#544) * improved logrus output switching performance Signed-off-by: kim * add license to test file Signed-off-by: kim --- internal/log/log_test.go | 69 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 69 insertions(+) create mode 100644 internal/log/log_test.go (limited to 'internal/log/log_test.go') diff --git a/internal/log/log_test.go b/internal/log/log_test.go new file mode 100644 index 000000000..01aae73ae --- /dev/null +++ b/internal/log/log_test.go @@ -0,0 +1,69 @@ +/* + GoToSocial + Copyright (C) 2021-2022 GoToSocial Authors admin@gotosocial.org + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU Affero General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU Affero General Public License for more details. + + You should have received a copy of the GNU Affero General Public License + along with this program. If not, see . +*/ + +package log_test + +import ( + "bytes" + "testing" + + "github.com/sirupsen/logrus" + "github.com/superseriousbusiness/gotosocial/internal/log" +) + +func TestOutputSplitFunc(t *testing.T) { + var outbuf, errbuf bytes.Buffer + + out := log.SplitErrOutputs(&outbuf, &errbuf) + + log := logrus.New() + log.SetOutput(out) + log.SetLevel(logrus.TraceLevel) + + for _, lvl := range logrus.AllLevels { + func() { + defer func() { recover() }() + log.Log(lvl, "hello world") + }() + + t.Logf("outbuf=%q errbuf=%q", outbuf.String(), errbuf.String()) + + switch lvl { + case logrus.PanicLevel: + if outbuf.Len() > 0 || errbuf.Len() == 0 { + t.Error("expected panic to log to OutputSplitter.Err") + } + case logrus.FatalLevel: + if outbuf.Len() > 0 || errbuf.Len() == 0 { + t.Error("expected fatal to log to OutputSplitter.Err") + } + case logrus.ErrorLevel: + if outbuf.Len() > 0 || errbuf.Len() == 0 { + t.Error("expected error to log to OutputSplitter.Err") + } + default: + if outbuf.Len() == 0 || errbuf.Len() > 0 { + t.Errorf("expected %s to log to OutputSplitter.Out", lvl) + } + } + + // Reset buffers + outbuf.Reset() + errbuf.Reset() + } +} -- cgit v1.2.3