2019-04-07 00:25:14 +00:00
// Copyright 2019 The Gitea Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package integrations
import (
2020-01-29 01:01:06 +00:00
"context"
2019-04-07 00:25:14 +00:00
"encoding/json"
"fmt"
"os"
"runtime"
"strings"
2019-09-17 09:39:37 +00:00
"sync"
2019-04-07 00:25:14 +00:00
"testing"
2020-06-02 01:39:44 +00:00
"time"
2019-04-07 00:25:14 +00:00
"code.gitea.io/gitea/modules/log"
2020-01-29 01:01:06 +00:00
"code.gitea.io/gitea/modules/queue"
2019-04-07 00:25:14 +00:00
)
2020-06-02 01:39:44 +00:00
var (
prefix string
slowTest = 10 * time . Second
slowFlush = 5 * time . Second
)
2019-04-07 00:25:14 +00:00
// TestLogger is a logger which will write to the testing log
type TestLogger struct {
log . WriterLogger
}
var writerCloser = & testLoggerWriterCloser { }
type testLoggerWriterCloser struct {
2019-09-17 09:39:37 +00:00
sync . RWMutex
2019-11-25 23:21:37 +00:00
t [ ] * testing . TB
2019-04-07 00:25:14 +00:00
}
2019-09-17 09:39:37 +00:00
func ( w * testLoggerWriterCloser ) setT ( t * testing . TB ) {
w . Lock ( )
2019-11-25 23:21:37 +00:00
w . t = append ( w . t , t )
2019-09-17 09:39:37 +00:00
w . Unlock ( )
}
2019-04-07 00:25:14 +00:00
func ( w * testLoggerWriterCloser ) Write ( p [ ] byte ) ( int , error ) {
2019-09-17 09:39:37 +00:00
w . RLock ( )
2019-11-25 23:21:37 +00:00
var t * testing . TB
if len ( w . t ) > 0 {
t = w . t [ len ( w . t ) - 1 ]
}
2019-09-17 09:39:37 +00:00
w . RUnlock ( )
2019-11-25 23:21:37 +00:00
if t != nil && * t != nil {
2019-04-07 00:25:14 +00:00
if len ( p ) > 0 && p [ len ( p ) - 1 ] == '\n' {
p = p [ : len ( p ) - 1 ]
}
2019-04-11 11:49:49 +00:00
defer func ( ) {
err := recover ( )
if err == nil {
return
}
var errString string
errErr , ok := err . ( error )
if ok {
errString = errErr . Error ( )
} else {
errString , ok = err . ( string )
}
if ! ok {
panic ( err )
}
if ! strings . HasPrefix ( errString , "Log in goroutine after " ) {
panic ( err )
}
} ( )
2019-11-25 23:21:37 +00:00
( * t ) . Log ( string ( p ) )
2019-04-07 00:25:14 +00:00
return len ( p ) , nil
}
return len ( p ) , nil
}
func ( w * testLoggerWriterCloser ) Close ( ) error {
2019-11-25 23:21:37 +00:00
w . Lock ( )
if len ( w . t ) > 0 {
w . t = w . t [ : len ( w . t ) - 1 ]
}
w . Unlock ( )
2019-04-07 00:25:14 +00:00
return nil
}
// PrintCurrentTest prints the current test to os.Stdout
2019-11-25 23:21:37 +00:00
func PrintCurrentTest ( t testing . TB , skip ... int ) func ( ) {
2020-06-02 01:39:44 +00:00
start := time . Now ( )
2019-04-07 00:25:14 +00:00
actualSkip := 1
if len ( skip ) > 0 {
actualSkip = skip [ 0 ]
}
_ , filename , line , _ := runtime . Caller ( actualSkip )
if log . CanColorStdout {
2019-06-12 19:41:28 +00:00
fmt . Fprintf ( os . Stdout , "=== %s (%s:%d)\n" , fmt . Formatter ( log . NewColoredValue ( t . Name ( ) ) ) , strings . TrimPrefix ( filename , prefix ) , line )
2019-04-07 00:25:14 +00:00
} else {
fmt . Fprintf ( os . Stdout , "=== %s (%s:%d)\n" , t . Name ( ) , strings . TrimPrefix ( filename , prefix ) , line )
}
2019-09-17 09:39:37 +00:00
writerCloser . setT ( & t )
2019-11-25 23:21:37 +00:00
return func ( ) {
2020-06-02 01:39:44 +00:00
took := time . Since ( start )
if took > slowTest {
if log . CanColorStdout {
fmt . Fprintf ( os . Stdout , "+++ %s is a slow test (took %v)\n" , fmt . Formatter ( log . NewColoredValue ( t . Name ( ) , log . Bold , log . FgYellow ) ) , fmt . Formatter ( log . NewColoredValue ( took , log . Bold , log . FgYellow ) ) )
} else {
fmt . Fprintf ( os . Stdout , "+++ %s is a slow tets (took %v)\n" , t . Name ( ) , took )
}
}
timer := time . AfterFunc ( slowFlush , func ( ) {
if log . CanColorStdout {
fmt . Fprintf ( os . Stdout , "+++ %s ... still flushing after %v ...\n" , fmt . Formatter ( log . NewColoredValue ( t . Name ( ) , log . Bold , log . FgRed ) ) , slowFlush )
} else {
fmt . Fprintf ( os . Stdout , "+++ %s ... still flushing after %v ...\n" , t . Name ( ) , slowFlush )
}
} )
2020-02-11 09:34:17 +00:00
if err := queue . GetManager ( ) . FlushAll ( context . Background ( ) , - 1 ) ; err != nil {
2020-01-29 01:01:06 +00:00
t . Errorf ( "Flushing queues failed with error %v" , err )
}
2020-06-02 01:39:44 +00:00
timer . Stop ( )
flushTook := time . Since ( start ) - took
if flushTook > slowFlush {
if log . CanColorStdout {
fmt . Fprintf ( os . Stdout , "+++ %s had a slow clean-up flush (took %v)\n" , fmt . Formatter ( log . NewColoredValue ( t . Name ( ) , log . Bold , log . FgRed ) ) , fmt . Formatter ( log . NewColoredValue ( flushTook , log . Bold , log . FgRed ) ) )
} else {
fmt . Fprintf ( os . Stdout , "+++ %s had a slow clean-up flush (took %v)\n" , t . Name ( ) , flushTook )
}
}
2019-11-25 23:21:37 +00:00
_ = writerCloser . Close ( )
}
2019-04-07 00:25:14 +00:00
}
// Printf takes a format and args and prints the string to os.Stdout
func Printf ( format string , args ... interface { } ) {
if log . CanColorStdout {
for i := 0 ; i < len ( args ) ; i ++ {
args [ i ] = log . NewColoredValue ( args [ i ] )
}
}
fmt . Fprintf ( os . Stdout , "\t" + format , args ... )
}
// NewTestLogger creates a TestLogger as a log.LoggerProvider
func NewTestLogger ( ) log . LoggerProvider {
logger := & TestLogger { }
logger . Colorize = log . CanColorStdout
logger . Level = log . TRACE
return logger
}
// Init inits connection writer with json config.
// json config only need key "level".
func ( log * TestLogger ) Init ( config string ) error {
err := json . Unmarshal ( [ ] byte ( config ) , log )
if err != nil {
return err
}
log . NewWriterLogger ( writerCloser )
return nil
}
// Flush when log should be flushed
func ( log * TestLogger ) Flush ( ) {
}
2020-07-06 00:07:07 +00:00
//ReleaseReopen does nothing
func ( log * TestLogger ) ReleaseReopen ( ) error {
return nil
}
2019-04-07 00:25:14 +00:00
// GetName returns the default name for this implementation
func ( log * TestLogger ) GetName ( ) string {
return "test"
}
func init ( ) {
log . Register ( "test" , NewTestLogger )
_ , filename , _ , _ := runtime . Caller ( 0 )
prefix = strings . TrimSuffix ( filename , "integrations/testlogger.go" )
}