mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 05:56:38 +08:00
ryan
89 lines
2.7 KiB
Go
89 lines
2.7 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package model
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/db"
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/stretchr/testify/require"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func setupZoneTestDB(t *testing.T) *gorm.DB {
|
|
t.Helper()
|
|
|
|
sqliteDB, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{
|
|
DisableForeignKeyConstraintWhenMigrating: true,
|
|
})
|
|
require.NoError(t, err)
|
|
require.NoError(t, sqliteDB.AutoMigrate(&Zone{}, &ZoneDomain{}))
|
|
db.SetDB(sqliteDB)
|
|
t.Cleanup(func() { db.SetDB(nil) })
|
|
return sqliteDB
|
|
}
|
|
|
|
func TestReplaceZoneDomainRouteBindingsRejectsForeignDomain(t *testing.T) {
|
|
conn := setupZoneTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
zone := Zone{Domain: "example.com"}
|
|
require.NoError(t, conn.Create(&zone).Error)
|
|
foreignRouteID := uint(11)
|
|
domain := ZoneDomain{
|
|
ZoneID: zone.ID,
|
|
ProxyRouteID: &foreignRouteID,
|
|
Domain: "api.example.com",
|
|
}
|
|
require.NoError(t, conn.Create(&domain).Error)
|
|
|
|
err := ReplaceZoneDomainRouteBindings(ctx, 12, []uint{domain.ID})
|
|
require.Error(t, err)
|
|
|
|
var got ZoneDomain
|
|
require.NoError(t, conn.First(&got, domain.ID).Error)
|
|
require.Equal(t, &foreignRouteID, got.ProxyRouteID)
|
|
}
|
|
|
|
func TestReplaceZoneDomainRouteBindingsReplacesCurrentRouteBindings(t *testing.T) {
|
|
conn := setupZoneTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
zone := Zone{Domain: "example.com"}
|
|
require.NoError(t, conn.Create(&zone).Error)
|
|
routeID := uint(21)
|
|
boundDomain := ZoneDomain{ZoneID: zone.ID, ProxyRouteID: &routeID, Domain: "old.example.com"}
|
|
requestedDomain := ZoneDomain{ZoneID: zone.ID, Domain: "new.example.com"}
|
|
require.NoError(t, conn.Create(&boundDomain).Error)
|
|
require.NoError(t, conn.Create(&requestedDomain).Error)
|
|
|
|
require.NoError(t, ReplaceZoneDomainRouteBindings(ctx, routeID, []uint{requestedDomain.ID}))
|
|
|
|
var domains []ZoneDomain
|
|
require.NoError(t, conn.Order("id asc").Find(&domains).Error)
|
|
require.Len(t, domains, 2)
|
|
require.Nil(t, domains[0].ProxyRouteID)
|
|
require.Equal(t, &routeID, domains[1].ProxyRouteID)
|
|
}
|
|
|
|
func TestListZoneDomainsByRouteID(t *testing.T) {
|
|
conn := setupZoneTestDB(t)
|
|
ctx := context.Background()
|
|
|
|
zone := Zone{Domain: "example.com"}
|
|
require.NoError(t, conn.Create(&zone).Error)
|
|
routeID := uint(31)
|
|
boundDomain := ZoneDomain{ZoneID: zone.ID, ProxyRouteID: &routeID, Domain: "api.example.com"}
|
|
unboundDomain := ZoneDomain{ZoneID: zone.ID, Domain: "www.example.com"}
|
|
require.NoError(t, conn.Create(&boundDomain).Error)
|
|
require.NoError(t, conn.Create(&unboundDomain).Error)
|
|
|
|
domains, err := ListZoneDomainsByRouteID(ctx, routeID)
|
|
require.NoError(t, err)
|
|
require.Len(t, domains, 1)
|
|
require.Equal(t, boundDomain.ID, domains[0].ID)
|
|
}
|