@@ -19,6 +19,7 @@ import (
1919 "code.gitea.io/actions-proto-go/runner/v1/runnerv1connect"
2020 "connectrpc.com/connect"
2121 "github.com/stretchr/testify/assert"
22+ "github.com/stretchr/testify/require"
2223 "google.golang.org/protobuf/types/known/timestamppb"
2324)
2425
@@ -63,8 +64,8 @@ func (r *mockRunner) doPing(t *testing.T) {
6364 resp , err := r .client .pingServiceClient .Ping (context .Background (), connect .NewRequest (& pingv1.PingRequest {
6465 Data : "mock-runner" ,
6566 }))
66- assert .NoError (t , err )
67- assert .Equal (t , "Hello, mock-runner!" , resp .Msg .Data )
67+ require .NoError (t , err )
68+ require .Equal (t , "Hello, mock-runner!" , resp .Msg .Data )
6869}
6970
7071func (r * mockRunner ) doRegister (t * testing.T , name , token string , labels []string ) {
@@ -75,11 +76,15 @@ func (r *mockRunner) doRegister(t *testing.T, name, token string, labels []strin
7576 Version : "mock-runner-version" ,
7677 Labels : labels ,
7778 }))
78- assert .NoError (t , err )
79+ require .NoError (t , err )
7980 r .client = newMockRunnerClient (resp .Msg .Runner .Uuid , resp .Msg .Runner .Token )
8081}
8182
8283func (r * mockRunner ) registerAsRepoRunner (t * testing.T , ownerName , repoName , runnerName string , labels []string ) {
84+ if ! setting .Database .Type .IsSQLite3 () {
85+ // registering a mock runner when using a database other than SQLite leaves leftovers
86+ t .FailNow ()
87+ }
8388 session := loginUser (t , ownerName )
8489 token := getTokenForLoggedInUser (t , session , auth_model .AccessTokenScopeWriteRepository )
8590 req := NewRequest (t , "GET" , fmt .Sprintf ("/api/v1/repos/%s/%s/actions/runners/registration-token" , ownerName , repoName )).AddTokenAuth (token )
@@ -102,7 +107,7 @@ func (r *mockRunner) fetchTask(t *testing.T, timeout ...time.Duration) *runnerv1
102107 resp , err := r .client .runnerServiceClient .FetchTask (context .Background (), connect .NewRequest (& runnerv1.FetchTaskRequest {
103108 TasksVersion : 0 ,
104109 }))
105- assert .NoError (t , err )
110+ require .NoError (t , err )
106111 if resp .Msg .Task != nil {
107112 task = resp .Msg .Task
108113 break
@@ -128,7 +133,7 @@ func (r *mockRunner) execTask(t *testing.T, task *runnerv1.Task, outcome *mockTa
128133 Rows : []* runnerv1.LogRow {lr },
129134 NoMore : idx == len (outcome .logRows )- 1 ,
130135 }))
131- assert .NoError (t , err )
136+ require .NoError (t , err )
132137 assert .EqualValues (t , idx + 1 , resp .Msg .AckIndex )
133138 }
134139 sentOutputKeys := make ([]string , 0 , len (outcome .outputs ))
@@ -140,7 +145,7 @@ func (r *mockRunner) execTask(t *testing.T, task *runnerv1.Task, outcome *mockTa
140145 },
141146 Outputs : map [string ]string {outputKey : outputValue },
142147 }))
143- assert .NoError (t , err )
148+ require .NoError (t , err )
144149 sentOutputKeys = append (sentOutputKeys , outputKey )
145150 assert .ElementsMatch (t , sentOutputKeys , resp .Msg .SentOutputs )
146151 }
@@ -152,6 +157,6 @@ func (r *mockRunner) execTask(t *testing.T, task *runnerv1.Task, outcome *mockTa
152157 StoppedAt : timestamppb .Now (),
153158 },
154159 }))
155- assert .NoError (t , err )
160+ require .NoError (t , err )
156161 assert .Equal (t , outcome .result , resp .Msg .State .Result )
157162}
0 commit comments