Dragonfly2

1
/*
2
 *     Copyright 2023 The Dragonfly Authors
3
 *
4
 * Licensed under the Apache License, Version 2.0 (the "License");
5
 * you may not use this file except in compliance with the License.
6
 * You may obtain a copy of the License at
7
 *
8
 *      http://www.apache.org/licenses/LICENSE-2.0
9
 *
10
 * Unless required by applicable law or agreed to in writing, software
11
 * distributed under the License is distributed on an "AS IS" BASIS,
12
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
 * See the License for the specific language governing permissions and
14
 * limitations under the License.
15
 */
16

17
package rpcserver
18

19
import (
20
	"reflect"
21
	"testing"
22

23
	"github.com/stretchr/testify/assert"
24
	"go.uber.org/mock/gomock"
25

26
	"d7y.io/dragonfly/v2/trainer/config"
27
	storagemocks "d7y.io/dragonfly/v2/trainer/storage/mocks"
28
	trainingmocks "d7y.io/dragonfly/v2/trainer/training/mocks"
29
)
30

31
func TestRPCServer_New(t *testing.T) {
32
	tests := []struct {
33
		name   string
34
		expect func(t *testing.T, s any)
35
	}{
36
		{
37
			name: "new server",
38
			expect: func(t *testing.T, s any) {
39
				assert := assert.New(t)
40
				assert.Equal(reflect.TypeOf(s).Elem().Name(), "Server")
41
			},
42
		},
43
	}
44

45
	for _, tc := range tests {
46
		t.Run(tc.name, func(t *testing.T) {
47
			ctl := gomock.NewController(t)
48
			defer ctl.Finish()
49
			storage := storagemocks.NewMockStorage(ctl)
50
			training := trainingmocks.NewMockTraining(ctl)
51

52
			svr := New(&config.Config{}, storage, training)
53
			tc.expect(t, svr)
54
		})
55
	}
56
}
57