mirror of
https://gitea.com/gitea/runner.git
synced 2026-11-04 14:35:32 +00:00
silverwind
6f0d4f1b48
Migrates the runner to `connectrpc.com/connect/v2` by pinning actionslib v1.5.0, which carries https://gitea.com/gitea/actionslib/pulls/28. The wire behaviour towards Gitea is unchanged and works against Gitea servers on connect v1 and v2. The client keeps v1's unlimited response size, and the error codes and messages the runner checks stay the same. Reviewed-on: https://gitea.com/gitea/runner/pulls/1325 Reviewed-by: bircni <[email protected]> Co-authored-by: silverwind <[email protected]>
65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
// Copyright 2026 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package mocks
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
runnerv1 "gitea.dev/actionslib/runner/v1"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type Client struct {
|
|
mock.Mock
|
|
AddressValue string
|
|
}
|
|
|
|
func (m *Client) Address() string {
|
|
return m.AddressValue
|
|
}
|
|
|
|
func call[Request, Response any](ctx context.Context, m *Client, method string, request *Request) (*Response, error) {
|
|
returns := m.MethodCalled(method, ctx, request)
|
|
if callback, ok := returns.Get(0).(func(context.Context, *Request) (*Response, error)); ok {
|
|
return callback(ctx, request)
|
|
}
|
|
var response *Response
|
|
if value := returns.Get(0); value != nil {
|
|
var ok bool
|
|
response, ok = value.(*Response)
|
|
if !ok {
|
|
panic(fmt.Sprintf("unexpected response type %T for %s", value, method))
|
|
}
|
|
}
|
|
return response, returns.Error(1)
|
|
}
|
|
|
|
func (m *Client) Declare(ctx context.Context, request *runnerv1.DeclareRequest) (*runnerv1.DeclareResponse, error) {
|
|
return call[runnerv1.DeclareRequest, runnerv1.DeclareResponse](ctx, m, "Declare", request)
|
|
}
|
|
|
|
func (m *Client) FetchTask(ctx context.Context, request *runnerv1.FetchTaskRequest) (*runnerv1.FetchTaskResponse, error) {
|
|
return call[runnerv1.FetchTaskRequest, runnerv1.FetchTaskResponse](ctx, m, "FetchTask", request)
|
|
}
|
|
|
|
func (m *Client) UpdateLog(ctx context.Context, request *runnerv1.UpdateLogRequest) (*runnerv1.UpdateLogResponse, error) {
|
|
return call[runnerv1.UpdateLogRequest, runnerv1.UpdateLogResponse](ctx, m, "UpdateLog", request)
|
|
}
|
|
|
|
func (m *Client) UpdateTask(ctx context.Context, request *runnerv1.UpdateTaskRequest) (*runnerv1.UpdateTaskResponse, error) {
|
|
return call[runnerv1.UpdateTaskRequest, runnerv1.UpdateTaskResponse](ctx, m, "UpdateTask", request)
|
|
}
|
|
|
|
func NewClient(t interface {
|
|
mock.TestingT
|
|
Cleanup(func())
|
|
},
|
|
) *Client {
|
|
client := &Client{}
|
|
client.Test(t)
|
|
t.Cleanup(func() { client.AssertExpectations(t) })
|
|
return client
|
|
}
|