Add bootstrap controller unit test

internal/config/config.go

@@ -30,23 +30,32 @@ var envs = map[string][]string{
 	"refresh_interval":     {"REFRESH_INTERVAL"},
 }
 
+type Interface struct {
+}
+type Interfaces map[string]Interface
+
+type Logger struct {
+	Level string `mapstructure:"level"`
+}
+
+type Stun struct {
+	Address string `mapstructure:"address"`
+}
+
+type Cloudflare struct {
+	ApiKey   string `mapstructure:"api_key"`
+	ApiEmail string `mapstructure:"api_email"`
+	ApiToken string `mapstructure:"api_token"`
+	ZoneName string `mapstructure:"zone_name"`
+}
+
 type Config struct {
-	WireGuard  string `mapstructure:"wg"`
-	Interfaces map[string]struct {
-	} `mapstructure:"interfaces"`
+	WireGuard       string        `mapstructure:"wg"`
+	Interfaces      Interfaces    `mapstructure:"interfaces"`
 	RefreshInterval time.Duration `mapstructure:"refresh_interval"`
-	Log             struct {
-		Level string `mapstructure:"level"`
-	} `mapstructure:"log"`
-	Stun struct {
-		Address string `mapstructure:"address"`
-	} `mapstructure:"stun"`
-	Cloudflare struct {
-		ApiKey   string `mapstructure:"api_key"`
-		ApiEmail string `mapstructure:"api_email"`
-		ApiToken string `mapstructure:"api_token"`
-		ZoneName string `mapstructure:"zone_name"`
-	} `mapstructure:"cloudflare"`
+	Log             Logger        `mapstructure:"log"`
+	Stun            Stun          `mapstructure:"stun"`
+	Cloudflare      Cloudflare    `mapstructure:"cloudflare"`
 }
 
 func Load() (*Config, error) {

internal/ctrl/api.go

@@ -0,0 +1,11 @@
+//go:generate mockgen -destination=./mock/mock_api.go -package=mock_ctrl . WireGuardClient
+
+package ctrl
+
+import (
+	"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
+)
+
+type WireGuardClient interface {
+	Device(deviceName string) (*wgtypes.Device, error)
+}

internal/ctrl/bootstrap.go

@@ -6,18 +6,17 @@ import (
 	"github.com/rs/zerolog"
 	"github.com/tjjh89017/stunmesh-go/internal/config"
 	"github.com/tjjh89017/stunmesh-go/internal/entity"
-	"golang.zx2c4.com/wireguard/wgctrl"
 )
 
 type BootstrapController struct {
-	wg      *wgctrl.Client
+	wg      WireGuardClient
 	config  *config.Config
 	devices DeviceRepository
 	peers   PeerRepository
 	logger  zerolog.Logger
 }
 
-func NewBootstrapController(wg *wgctrl.Client, config *config.Config, devices DeviceRepository, peers PeerRepository, logger *zerolog.Logger) *BootstrapController {
+func NewBootstrapController(wg WireGuardClient, config *config.Config, devices DeviceRepository, peers PeerRepository, logger *zerolog.Logger) *BootstrapController {
 	return &BootstrapController{
 		wg:      wg,
 		config:  config,

internal/ctrl/bootstrap_test.go

@@ -0,0 +1,133 @@
+package ctrl_test
+
+import (
+	"context"
+	"errors"
+	"testing"
+
+	"github.com/rs/zerolog"
+	"github.com/tjjh89017/stunmesh-go/internal/config"
+	"github.com/tjjh89017/stunmesh-go/internal/ctrl"
+	mock "github.com/tjjh89017/stunmesh-go/internal/ctrl/mock"
+	gomock "go.uber.org/mock/gomock"
+	"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
+)
+
+func TestBootstrap_WithError(t *testing.T) {
+	mockCtrl := gomock.NewController(t)
+	defer mockCtrl.Finish()
+
+	mockWgClient := mock.NewMockWireGuardClient(mockCtrl)
+	mockDevices := mock.NewMockDeviceRepository(mockCtrl)
+	mockPeers := mock.NewMockPeerRepository(mockCtrl)
+	logger := zerolog.Nop()
+	config := &config.Config{
+		WireGuard: "wg0",
+	}
+
+	mockWgClient.EXPECT().Device("wg0").Return(nil, errors.New("device not found"))
+
+	bootstrap := ctrl.NewBootstrapController(
+		mockWgClient,
+		config,
+		mockDevices,
+		mockPeers,
+		&logger,
+	)
+
+	bootstrap.Execute(context.TODO())
+}
+
+func TestBootstrap_WithSingleInterface(t *testing.T) {
+	mockCtrl := gomock.NewController(t)
+	defer mockCtrl.Finish()
+
+	mockWgClient := mock.NewMockWireGuardClient(mockCtrl)
+	mockDevices := mock.NewMockDeviceRepository(mockCtrl)
+	mockPeers := mock.NewMockPeerRepository(mockCtrl)
+	logger := zerolog.Nop()
+	config := &config.Config{
+		WireGuard: "wg0",
+	}
+
+	mockDevice := &wgtypes.Device{
+		Name:       "wg0",
+		ListenPort: 51820,
+		PrivateKey: [32]byte{},
+		Peers: []wgtypes.Peer{
+			{
+				PublicKey: [32]byte{},
+			},
+		},
+	}
+
+	mockWgClient.EXPECT().Device("wg0").Return(mockDevice, nil)
+	mockDevices.EXPECT().Save(gomock.Any(), gomock.Any())
+	mockPeers.EXPECT().Save(gomock.Any(), gomock.Any())
+
+	bootstrap := ctrl.NewBootstrapController(
+		mockWgClient,
+		config,
+		mockDevices,
+		mockPeers,
+		&logger,
+	)
+
+	bootstrap.Execute(context.TODO())
+}
+
+func TestBootstrap_WithMultipleInterfaces(t *testing.T) {
+	mockCtrl := gomock.NewController(t)
+	defer mockCtrl.Finish()
+
+	mockWgClient := mock.NewMockWireGuardClient(mockCtrl)
+	mockDevices := mock.NewMockDeviceRepository(mockCtrl)
+	mockPeers := mock.NewMockPeerRepository(mockCtrl)
+	logger := zerolog.Nop()
+	config := &config.Config{
+		Interfaces: map[string]config.Interface{
+			"wg0": {},
+			"wg1": {},
+		},
+	}
+
+	mockDevice0 := &wgtypes.Device{
+		Name:       "wg0",
+		ListenPort: 51820,
+		PrivateKey: [32]byte{},
+		Peers: []wgtypes.Peer{
+			{
+				PublicKey: [32]byte{},
+			},
+		},
+	}
+
+	mockDevice1 := &wgtypes.Device{
+		Name:       "wg1",
+		ListenPort: 51821,
+		PrivateKey: [32]byte{},
+		Peers: []wgtypes.Peer{
+			{
+				PublicKey: [32]byte{},
+			},
+			{
+				PublicKey: [32]byte{},
+			},
+		},
+	}
+
+	mockWgClient.EXPECT().Device("wg0").Return(mockDevice0, nil)
+	mockWgClient.EXPECT().Device("wg1").Return(mockDevice1, nil)
+	mockDevices.EXPECT().Save(gomock.Any(), gomock.Any()).Times(2)
+	mockPeers.EXPECT().Save(gomock.Any(), gomock.Any()).Times(3)
+
+	bootstrap := ctrl.NewBootstrapController(
+		mockWgClient,
+		config,
+		mockDevices,
+		mockPeers,
+		&logger,
+	)
+
+	bootstrap.Execute(context.TODO())
+}

wire.go

@@ -24,6 +24,7 @@ func setup() (*daemon.Daemon, error) {
 	wire.Build(
 		config.Load,
 		wgctrl.New,
+		wire.Bind(new(ctrl.WireGuardClient), new(*wgctrl.Client)),
 		provideCloudflareApi,
 		provideStore,
 		wire.Bind(new(plugin.Store), new(*store.CloudflareStore)),