Skip to content

Commit f6fce34

Browse files
committed
fix
1 parent 4121f95 commit f6fce34

File tree

11 files changed

+152
-187
lines changed

11 files changed

+152
-187
lines changed

.github/workflows/pull-db-tests.yml

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,10 +154,18 @@ jobs:
154154
runs-on: ubuntu-latest
155155
services:
156156
mysql:
157-
image: mysql:8.0
157+
# official mysql image doesn't support MYSQL_EXTRA_FLAGS
158+
image: bitnami/mysql:8.0
158159
env:
159-
MYSQL_ALLOW_EMPTY_PASSWORD: true
160+
ALLOW_EMPTY_PASSWORD: true
160161
MYSQL_DATABASE: testgitea
162+
# https://dev.mysql.com/doc/refman/8.0/en/optimizing-innodb-diskio.html
163+
MYSQL_EXTRA_FLAGS: >
164+
--disable-log-bin
165+
--innodb-buffer-pool-size=1G
166+
--innodb-doublewrite=off
167+
--innodb-flush-log-at-trx-commit=0
168+
--innodb-flush-method=nosync
161169
ports:
162170
- "3306:3306"
163171
elasticsearch:

models/migrations/base/tests.go

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@ import (
1515
"code.gitea.io/gitea/models/unittest"
1616
"code.gitea.io/gitea/modules/base"
1717
"code.gitea.io/gitea/modules/git"
18-
"code.gitea.io/gitea/modules/log"
1918
"code.gitea.io/gitea/modules/setting"
2019
"code.gitea.io/gitea/modules/testlogger"
2120

@@ -91,21 +90,19 @@ func PrepareTestEnv(t *testing.T, skip int, syncModels ...any) (*xorm.Engine, fu
9190
}
9291

9392
func MainTest(m *testing.M) {
94-
log.RegisterEventWriter("test", testlogger.NewTestLoggerWriter)
93+
testlogger.Init()
9594

9695
giteaRoot := base.SetupGiteaRoot()
9796
if giteaRoot == "" {
98-
fmt.Println("Environment variable $GITEA_ROOT not set")
99-
os.Exit(1)
97+
testlogger.Fatalf("Environment variable $GITEA_ROOT not set\n")
10098
}
10199
giteaBinary := "gitea"
102100
if runtime.GOOS == "windows" {
103101
giteaBinary += ".exe"
104102
}
105103
setting.AppPath = filepath.Join(giteaRoot, giteaBinary)
106104
if _, err := os.Stat(setting.AppPath); err != nil {
107-
fmt.Printf("Could not find gitea binary at %s\n", setting.AppPath)
108-
os.Exit(1)
105+
testlogger.Fatalf("Could not find gitea binary at %s\n", setting.AppPath)
109106
}
110107

111108
giteaConf := os.Getenv("GITEA_CONF")
@@ -122,17 +119,15 @@ func MainTest(m *testing.M) {
122119

123120
tmpDataPath, err := os.MkdirTemp("", "data")
124121
if err != nil {
125-
fmt.Printf("Unable to create temporary data path %v\n", err)
126-
os.Exit(1)
122+
testlogger.Fatalf("Unable to create temporary data path %v\n", err)
127123
}
128124

129125
setting.CustomPath = filepath.Join(setting.AppWorkPath, "custom")
130126
setting.AppDataPath = tmpDataPath
131127

132128
unittest.InitSettings()
133129
if err = git.InitFull(context.Background()); err != nil {
134-
fmt.Printf("Unable to InitFull: %v\n", err)
135-
os.Exit(1)
130+
testlogger.Fatalf("Unable to InitFull: %v\n", err)
136131
}
137132
setting.LoadDBSetting()
138133
setting.InitLoggersForTest()

modules/log/color.go

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,13 +86,19 @@ type ColoredValue struct {
8686
colors []ColorAttribute
8787
}
8888

89+
var _ fmt.Formatter = (*ColoredValue)(nil)
90+
8991
func (c *ColoredValue) Format(f fmt.State, verb rune) {
9092
_, _ = f.Write(ColorBytes(c.colors...))
9193
s := fmt.Sprintf(fmt.FormatString(f, verb), c.v)
9294
_, _ = f.Write([]byte(s))
9395
_, _ = f.Write(resetBytes)
9496
}
9597

98+
func (c *ColoredValue) Value() any {
99+
return c.v
100+
}
101+
96102
func NewColoredValue(v any, color ...ColorAttribute) *ColoredValue {
97103
return &ColoredValue{v: v, colors: color}
98104
}

modules/queue/workergroup.go

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -236,6 +236,9 @@ loop:
236236
emptyCounter := 0
237237
for {
238238
select {
239+
case <-q.ctxRun.Done():
240+
log.Debug("Queue %q is shutting down", q.GetName())
241+
return
239242
case data, dataOk := <-wg.popItemChan:
240243
if !dataOk {
241244
return
@@ -251,9 +254,6 @@ loop:
251254
log.Error("Failed to pop item from queue %q (doFlush): %v", q.GetName(), err)
252255
}
253256
return
254-
case <-q.ctxRun.Done():
255-
log.Debug("Queue %q is shutting down", q.GetName())
256-
return
257257
case <-time.After(20 * time.Millisecond):
258258
// There is no reliable way to make sure all queue items are consumed by the Flush, there always might be some items stored in some buffers/temp variables.
259259
// If we run Gitea in a cluster, we can even not guarantee all items are consumed in a deterministic instance.
@@ -331,6 +331,15 @@ func (q *WorkerPoolQueue[T]) doRun() {
331331
var batchDispatchC <-chan time.Time = infiniteTimerC
332332
for {
333333
select {
334+
case flush := <-q.flushChan:
335+
// before flushing, it needs to try to dispatch the batch to worker first, in case there is no worker running
336+
// after the flushing, there is at least one worker running, so "doFlush" could wait for workers to finish
337+
// since we are already in a "flush" operation, so the dispatching function shouldn't read the flush chan.
338+
q.doDispatchBatchToWorker(wg, skipFlushChan)
339+
q.doFlush(wg, flush)
340+
case <-q.ctxRun.Done():
341+
log.Debug("Queue %q is shutting down", q.GetName())
342+
return
334343
case data, dataOk := <-wg.popItemChan:
335344
if !dataOk {
336345
return
@@ -349,20 +358,11 @@ func (q *WorkerPoolQueue[T]) doRun() {
349358
case <-batchDispatchC:
350359
batchDispatchC = infiniteTimerC
351360
q.doDispatchBatchToWorker(wg, q.flushChan)
352-
case flush := <-q.flushChan:
353-
// before flushing, it needs to try to dispatch the batch to worker first, in case there is no worker running
354-
// after the flushing, there is at least one worker running, so "doFlush" could wait for workers to finish
355-
// since we are already in a "flush" operation, so the dispatching function shouldn't read the flush chan.
356-
q.doDispatchBatchToWorker(wg, skipFlushChan)
357-
q.doFlush(wg, flush)
358361
case err := <-wg.popItemErr:
359362
if !q.isCtxRunCanceled() {
360363
log.Error("Failed to pop item from queue %q (doRun): %v", q.GetName(), err)
361364
}
362365
return
363-
case <-q.ctxRun.Done():
364-
log.Debug("Queue %q is shutting down", q.GetName())
365-
return
366366
}
367367
}
368368
}

modules/testlogger/testlogger.go

Lines changed: 62 additions & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,10 @@ import (
1919
)
2020

2121
var (
22-
prefix string
23-
SlowTest = 10 * time.Second
24-
SlowFlush = 5 * time.Second
22+
prefix string
23+
TestTimeout = 10 * time.Minute
24+
TestSlowRun = 10 * time.Second
25+
TestSlowFlush = 5 * time.Second
2526
)
2627

2728
var WriterCloser = &testLoggerWriterCloser{}
@@ -89,79 +90,97 @@ func (w *testLoggerWriterCloser) Reset() {
8990
w.Unlock()
9091
}
9192

93+
// Printf takes a format and args and prints the string to os.Stdout
94+
func Printf(format string, args ...any) {
95+
if !log.CanColorStdout {
96+
for i := 0; i < len(args); i++ {
97+
if c, ok := args[i].(*log.ColoredValue); ok {
98+
args[i] = c.Value()
99+
}
100+
}
101+
}
102+
_, _ = fmt.Fprintf(os.Stdout, format, args...)
103+
}
104+
92105
// PrintCurrentTest prints the current test to os.Stdout
93106
func PrintCurrentTest(t testing.TB, skip ...int) func() {
94107
t.Helper()
95-
start := time.Now()
108+
runStart := time.Now()
96109
actualSkip := util.OptionalArg(skip) + 1
97110
_, filename, line, _ := runtime.Caller(actualSkip)
98111

99-
if log.CanColorStdout {
100-
_, _ = fmt.Fprintf(os.Stdout, "=== %s (%s:%d)\n", fmt.Formatter(log.NewColoredValue(t.Name())), strings.TrimPrefix(filename, prefix), line)
101-
} else {
102-
_, _ = fmt.Fprintf(os.Stdout, "=== %s (%s:%d)\n", t.Name(), strings.TrimPrefix(filename, prefix), line)
103-
}
112+
Printf("=== %s (%s:%d)\n", log.NewColoredValue(t.Name()), strings.TrimPrefix(filename, prefix), line)
113+
104114
WriterCloser.pushT(t)
105-
return func() {
106-
took := time.Since(start)
107-
if took > SlowTest {
108-
if log.CanColorStdout {
109-
_, _ = 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)))
110-
} else {
111-
_, _ = fmt.Fprintf(os.Stdout, "+++ %s is a slow test (took %v)\n", t.Name(), took)
115+
timeoutChecker := time.AfterFunc(TestTimeout, func() {
116+
l := 128 * 1024
117+
var stack []byte
118+
for {
119+
stack = make([]byte, l)
120+
n := runtime.Stack(stack, true)
121+
if n <= l {
122+
stack = stack[:n]
123+
break
112124
}
125+
l = n
113126
}
114-
timer := time.AfterFunc(SlowFlush, func() {
115-
if log.CanColorStdout {
116-
_, _ = fmt.Fprintf(os.Stdout, "+++ %s ... still flushing after %v ...\n", fmt.Formatter(log.NewColoredValue(t.Name(), log.Bold, log.FgRed)), SlowFlush)
117-
} else {
118-
_, _ = fmt.Fprintf(os.Stdout, "+++ %s ... still flushing after %v ...\n", t.Name(), SlowFlush)
119-
}
127+
Printf("!!! %s ... timeout: %v ... stacktrace:\n%s\n\n", log.NewColoredValue(t.Name(), log.Bold, log.FgRed), TestTimeout, string(stack))
128+
})
129+
return func() {
130+
flushStart := time.Now()
131+
slowFlushChecker := time.AfterFunc(TestSlowFlush, func() {
132+
Printf("+++ %s ... still flushing after %v ...\n", log.NewColoredValue(t.Name(), log.Bold, log.FgRed), TestSlowFlush)
120133
})
121134
if err := queue.GetManager().FlushAll(context.Background(), -1); err != nil {
122135
t.Errorf("Flushing queues failed with error %v", err)
123136
}
124-
timer.Stop()
125-
flushTook := time.Since(start) - took
126-
if flushTook > SlowFlush {
127-
if log.CanColorStdout {
128-
_, _ = 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)))
129-
} else {
130-
_, _ = fmt.Fprintf(os.Stdout, "+++ %s had a slow clean-up flush (took %v)\n", t.Name(), flushTook)
131-
}
132-
}
133-
WriterCloser.popT()
134-
}
135-
}
137+
slowFlushChecker.Stop()
138+
timeoutChecker.Stop()
136139

137-
// Printf takes a format and args and prints the string to os.Stdout
138-
func Printf(format string, args ...any) {
139-
if log.CanColorStdout {
140-
for i := 0; i < len(args); i++ {
141-
args[i] = log.NewColoredValue(args[i])
140+
runDuration := time.Since(runStart)
141+
flushDuration := time.Since(flushStart)
142+
if runDuration > TestSlowRun {
143+
Printf("+++ %s is a slow test (run: %v, flush: %v)\n", log.NewColoredValue(t.Name(), log.Bold, log.FgYellow), runDuration, flushDuration)
142144
}
145+
WriterCloser.popT()
143146
}
144-
_, _ = fmt.Fprintf(os.Stdout, "\t"+format, args...)
145147
}
146148

147149
// TestLogEventWriter is a logger which will write to the testing log
148150
type TestLogEventWriter struct {
149151
*log.EventWriterBaseImpl
150152
}
151153

152-
// NewTestLoggerWriter creates a TestLogEventWriter as a log.LoggerProvider
153-
func NewTestLoggerWriter(name string, mode log.WriterMode) log.EventWriter {
154+
// newTestLoggerWriter creates a TestLogEventWriter as a log.LoggerProvider
155+
func newTestLoggerWriter(name string, mode log.WriterMode) log.EventWriter {
154156
w := &TestLogEventWriter{}
155157
w.EventWriterBaseImpl = log.NewEventWriterBase(name, "test-log-writer", mode)
156158
w.OutputWriteCloser = WriterCloser
157159
return w
158160
}
159161

160-
func init() {
162+
func Init() {
161163
const relFilePath = "modules/testlogger/testlogger.go"
162164
_, filename, _, _ := runtime.Caller(0)
163165
if !strings.HasSuffix(filename, relFilePath) {
164166
panic("source code file path doesn't match expected: " + relFilePath)
165167
}
166168
prefix = strings.TrimSuffix(filename, relFilePath)
169+
170+
log.RegisterEventWriter("test", newTestLoggerWriter)
171+
172+
duration, err := time.ParseDuration(os.Getenv("GITEA_TEST_SLOW_RUN"))
173+
if err == nil && duration > 0 {
174+
TestSlowRun = duration
175+
}
176+
177+
duration, err = time.ParseDuration(os.Getenv("GITEA_TEST_SLOW_FLUSH"))
178+
if err == nil && duration > 0 {
179+
TestSlowFlush = duration
180+
}
181+
}
182+
183+
func Fatalf(format string, args ...any) {
184+
Printf(format+"\n", args...)
185+
os.Exit(1)
167186
}

tests/integration/README.md

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -99,18 +99,8 @@ We appreciate that some testing machines may not be very powerful and
9999
the default timeouts for declaring a slow test or a slow clean-up flush
100100
may not be appropriate.
101101

102-
You can either:
103-
104-
* Within the test ini file set the following section:
105-
106-
```ini
107-
[integration-tests]
108-
SLOW_TEST = 10s ; 10s is the default value
109-
SLOW_FLUSH = 5S ; 5s is the default value
110-
```
111-
112-
* Set the following environment variables:
102+
You can set the following environment variables:
113103

114104
```bash
115-
GITEA_SLOW_TEST_TIME="10s" GITEA_SLOW_FLUSH_TIME="5s" make test-sqlite
105+
GITEA_TEST_SLOW_RUN="10s" GITEA_TEST_SLOW_FLUSH="5s" make test-sqlite
116106
```

tests/integration/integration_test.go

Lines changed: 0 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -20,15 +20,13 @@ import (
2020
"strings"
2121
"sync/atomic"
2222
"testing"
23-
"time"
2423

2524
"code.gitea.io/gitea/models/auth"
2625
"code.gitea.io/gitea/models/unittest"
2726
"code.gitea.io/gitea/modules/graceful"
2827
"code.gitea.io/gitea/modules/json"
2928
"code.gitea.io/gitea/modules/log"
3029
"code.gitea.io/gitea/modules/setting"
31-
"code.gitea.io/gitea/modules/testlogger"
3230
"code.gitea.io/gitea/modules/util"
3331
"code.gitea.io/gitea/modules/web"
3432
"code.gitea.io/gitea/routers"
@@ -90,27 +88,6 @@ func TestMain(m *testing.M) {
9088
tests.InitTest(true)
9189
testWebRoutes = routers.NormalRoutes()
9290

93-
// integration test settings...
94-
if setting.CfgProvider != nil {
95-
testingCfg := setting.CfgProvider.Section("integration-tests")
96-
testlogger.SlowTest = testingCfg.Key("SLOW_TEST").MustDuration(testlogger.SlowTest)
97-
testlogger.SlowFlush = testingCfg.Key("SLOW_FLUSH").MustDuration(testlogger.SlowFlush)
98-
}
99-
100-
if os.Getenv("GITEA_SLOW_TEST_TIME") != "" {
101-
duration, err := time.ParseDuration(os.Getenv("GITEA_SLOW_TEST_TIME"))
102-
if err == nil {
103-
testlogger.SlowTest = duration
104-
}
105-
}
106-
107-
if os.Getenv("GITEA_SLOW_FLUSH_TIME") != "" {
108-
duration, err := time.ParseDuration(os.Getenv("GITEA_SLOW_FLUSH_TIME"))
109-
if err == nil {
110-
testlogger.SlowFlush = duration
111-
}
112-
}
113-
11491
os.Unsetenv("GIT_AUTHOR_NAME")
11592
os.Unsetenv("GIT_AUTHOR_EMAIL")
11693
os.Unsetenv("GIT_AUTHOR_DATE")
@@ -132,8 +109,6 @@ func TestMain(m *testing.M) {
132109
// Instead, "No tests were found", last nonsense log is "According to the configuration, subsequent logs will not be printed to the console"
133110
exitCode := m.Run()
134111

135-
testlogger.WriterCloser.Reset()
136-
137112
if err = util.RemoveAll(setting.Indexer.IssuePath); err != nil {
138113
fmt.Printf("util.RemoveAll: %v\n", err)
139114
os.Exit(1)

0 commit comments

Comments
 (0)