356 lines
10 KiB
Go
356 lines
10 KiB
Go
package repository
|
||
|
||
import (
|
||
"code.30cm.net/digimon/app-cloudep-wallet-service/internal/config"
|
||
"code.30cm.net/digimon/app-cloudep-wallet-service/pkg/domain/repository"
|
||
"code.30cm.net/digimon/app-cloudep-wallet-service/pkg/domain/wallet"
|
||
"code.30cm.net/digimon/app-cloudep-wallet-service/pkg/lib/sql_client"
|
||
"context"
|
||
"errors"
|
||
"fmt"
|
||
"github.com/shopspring/decimal"
|
||
"github.com/stretchr/testify/assert"
|
||
"gorm.io/gorm"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func SetupTestWalletRepository() (repository.WalletRepository, *gorm.DB, func(), error) {
|
||
host, port, _, tearDown, err := startMySQLContainer()
|
||
if err != nil {
|
||
return nil, nil, nil, fmt.Errorf("failed to start MySQL container: %w", err)
|
||
}
|
||
|
||
conf := config.Config{
|
||
MySQL: struct {
|
||
UserName string
|
||
Password string
|
||
Host string
|
||
Port string
|
||
Database string
|
||
MaxIdleConns int
|
||
MaxOpenConns int
|
||
ConnMaxLifetime time.Duration
|
||
LogLevel string
|
||
}{
|
||
UserName: MySQLUser,
|
||
Password: MySQLPassword,
|
||
Host: host,
|
||
Port: port,
|
||
Database: MySQLDatabase,
|
||
MaxIdleConns: 10,
|
||
MaxOpenConns: 100,
|
||
ConnMaxLifetime: 300,
|
||
LogLevel: "info",
|
||
},
|
||
}
|
||
|
||
db, err := sql_client.NewMySQLClient(conf)
|
||
if err != nil {
|
||
tearDown()
|
||
return nil, nil, nil, fmt.Errorf("failed to create db client: %w", err)
|
||
}
|
||
|
||
repo := MustCategoryRepository(WalletRepositoryParam{DB: db})
|
||
return repo, db, tearDown, nil
|
||
}
|
||
|
||
func TestWalletRepository_InitWallets(t *testing.T) {
|
||
repo, db, tearDown, err := SetupTestWalletRepository()
|
||
assert.NoError(t, err)
|
||
defer tearDown()
|
||
|
||
// 🔽 這裡加上建表 SQL
|
||
createTableSQL := `
|
||
CREATE TABLE IF NOT EXISTS wallet (
|
||
id BIGINT UNSIGNED NOT NULL AUTO_INCREMENT COMMENT '主鍵 ID（錢包唯一識別）',
|
||
brand VARCHAR(50) NOT NULL COMMENT '品牌/平台（多租戶識別）',
|
||
uid VARCHAR(64) NOT NULL COMMENT '使用者 UID',
|
||
asset VARCHAR(32) NOT NULL COMMENT '資產代碼（如 BTC、ETH、GEM_RED 等）',
|
||
balance DECIMAL(30, 18) UNSIGNED NOT NULL DEFAULT 0 COMMENT '資產餘額',
|
||
type TINYINT NOT NULL COMMENT '錢包類型',
|
||
create_at INTEGER NOT NULL DEFAULT 0 COMMENT '建立時間（Unix）',
|
||
update_at INTEGER NOT NULL DEFAULT 0 COMMENT '更新時間（Unix）',
|
||
PRIMARY KEY (id),
|
||
UNIQUE KEY uq_brand_uid_asset_type (brand, uid, asset, type),
|
||
KEY idx_uid (uid),
|
||
KEY idx_brand (brand)
|
||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci COMMENT='錢包';
|
||
`
|
||
|
||
err = db.Exec(createTableSQL).Error
|
||
assert.NoError(t, err)
|
||
|
||
tests := []struct {
|
||
name string
|
||
param []repository.Wallet
|
||
wantErr bool
|
||
}{
|
||
{
|
||
name: "insert single wallet",
|
||
param: []repository.Wallet{
|
||
{
|
||
Brand: "test-brand",
|
||
UID: "user001",
|
||
Asset: "BTC",
|
||
Balance: decimal.NewFromFloat(10.5),
|
||
Type: wallet.TypeAvailable,
|
||
},
|
||
},
|
||
wantErr: false,
|
||
},
|
||
{
|
||
name: "insert multiple wallets",
|
||
param: []repository.Wallet{
|
||
{
|
||
Brand: "test-brand",
|
||
UID: "user002",
|
||
Asset: "ETH",
|
||
Balance: decimal.NewFromFloat(5),
|
||
Type: wallet.TypeAvailable,
|
||
},
|
||
{
|
||
Brand: "test-brand",
|
||
UID: "user002",
|
||
Asset: "ETH",
|
||
Balance: decimal.NewFromFloat(1.2),
|
||
Type: wallet.TypeFreeze,
|
||
},
|
||
},
|
||
wantErr: false,
|
||
},
|
||
{
|
||
name: "insert duplicate primary key (should fail if unique constraint)",
|
||
param: []repository.Wallet{
|
||
{
|
||
Brand: "test-brand",
|
||
UID: "user001",
|
||
Asset: "BTC",
|
||
Balance: decimal.NewFromFloat(1),
|
||
Type: wallet.TypeAvailable, // 與第一筆測試資料相同
|
||
},
|
||
},
|
||
wantErr: true, // 預期會違反 UNIQUE constraint
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
err := repo.InitWallets(context.Background(), tt.param)
|
||
if tt.wantErr {
|
||
assert.Error(t, err)
|
||
} else {
|
||
assert.NoError(t, err)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestWalletRepository_QueryBalances(t *testing.T) {
|
||
repo, db, tearDown, err := SetupTestWalletRepository()
|
||
assert.NoError(t, err)
|
||
defer tearDown()
|
||
// 🔽 這裡加上建表 SQL
|
||
createTableSQL := `
|
||
CREATE TABLE IF NOT EXISTS wallet (
|
||
id BIGINT UNSIGNED NOT NULL AUTO_INCREMENT COMMENT '主鍵 ID（錢包唯一識別）',
|
||
brand VARCHAR(50) NOT NULL COMMENT '品牌/平台（多租戶識別）',
|
||
uid VARCHAR(64) NOT NULL COMMENT '使用者 UID',
|
||
asset VARCHAR(32) NOT NULL COMMENT '資產代碼（如 BTC、ETH、GEM_RED 等）',
|
||
balance DECIMAL(30, 18) UNSIGNED NOT NULL DEFAULT 0 COMMENT '資產餘額',
|
||
type TINYINT NOT NULL COMMENT '錢包類型',
|
||
create_at INTEGER NOT NULL DEFAULT 0 COMMENT '建立時間（Unix）',
|
||
update_at INTEGER NOT NULL DEFAULT 0 COMMENT '更新時間（Unix）',
|
||
PRIMARY KEY (id),
|
||
UNIQUE KEY uq_brand_uid_asset_type (brand, uid, asset, type),
|
||
KEY idx_uid (uid),
|
||
KEY idx_brand (brand)
|
||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci COMMENT='錢包';
|
||
`
|
||
|
||
err = db.Exec(createTableSQL).Error
|
||
assert.NoError(t, err)
|
||
|
||
ctx := context.Background()
|
||
|
||
// 建立初始化錢包資料
|
||
wallets := []repository.Wallet{
|
||
{Brand: "brand1", UID: "user1", Asset: "BTC", Balance: decimal.NewFromFloat(1.5), Type: wallet.TypeAvailable},
|
||
{Brand: "brand1", UID: "user1", Asset: "ETH", Balance: decimal.NewFromFloat(2.5), Type: wallet.TypeFreeze},
|
||
{Brand: "brand1", UID: "user2", Asset: "BTC", Balance: decimal.NewFromFloat(3.0), Type: wallet.TypeAvailable},
|
||
}
|
||
|
||
err = repo.InitWallets(ctx, wallets)
|
||
assert.NoError(t, err)
|
||
|
||
tests := []struct {
|
||
name string
|
||
query repository.BalanceQuery
|
||
expected int
|
||
}{
|
||
{
|
||
name: "Query all by UID",
|
||
query: repository.BalanceQuery{
|
||
UID: "user1",
|
||
},
|
||
expected: 2,
|
||
},
|
||
{
|
||
name: "Query by UID and Asset",
|
||
query: repository.BalanceQuery{
|
||
UID: "user1",
|
||
Asset: "BTC",
|
||
},
|
||
expected: 1,
|
||
},
|
||
{
|
||
name: "Query by UID with non-existing asset",
|
||
query: repository.BalanceQuery{
|
||
UID: "user1",
|
||
Asset: "GEM_RED",
|
||
},
|
||
expected: 0,
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
got, err := repo.QueryBalances(ctx, tt.query)
|
||
assert.NoError(t, err)
|
||
assert.Len(t, got, tt.expected)
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestWalletRepository_QueryBalancesByUIDs(t *testing.T) {
|
||
repo, db, tearDown, err := SetupTestWalletRepository()
|
||
assert.NoError(t, err)
|
||
defer tearDown()
|
||
|
||
// 🔽 這裡加上建表 SQL
|
||
createTableSQL := `
|
||
CREATE TABLE IF NOT EXISTS wallet (
|
||
id BIGINT UNSIGNED NOT NULL AUTO_INCREMENT COMMENT '主鍵 ID（錢包唯一識別）',
|
||
brand VARCHAR(50) NOT NULL COMMENT '品牌/平台（多租戶識別）',
|
||
uid VARCHAR(64) NOT NULL COMMENT '使用者 UID',
|
||
asset VARCHAR(32) NOT NULL COMMENT '資產代碼（如 BTC、ETH、GEM_RED 等）',
|
||
balance DECIMAL(30, 18) UNSIGNED NOT NULL DEFAULT 0 COMMENT '資產餘額',
|
||
type TINYINT NOT NULL COMMENT '錢包類型',
|
||
create_at INTEGER NOT NULL DEFAULT 0 COMMENT '建立時間（Unix）',
|
||
update_at INTEGER NOT NULL DEFAULT 0 COMMENT '更新時間（Unix）',
|
||
PRIMARY KEY (id),
|
||
UNIQUE KEY uq_brand_uid_asset_type (brand, uid, asset, type),
|
||
KEY idx_uid (uid),
|
||
KEY idx_brand (brand)
|
||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci COMMENT='錢包';
|
||
`
|
||
|
||
err = db.Exec(createTableSQL).Error
|
||
assert.NoError(t, err)
|
||
|
||
ctx := context.Background()
|
||
|
||
// 初始化錢包資料
|
||
initData := []repository.Wallet{
|
||
{Brand: "brand1", UID: "user1", Asset: "BTC", Balance: decimal.NewFromFloat(1.5), Type: wallet.TypeAvailable},
|
||
{Brand: "brand1", UID: "user2", Asset: "BTC", Balance: decimal.NewFromFloat(2.0), Type: wallet.TypeAvailable},
|
||
{Brand: "brand1", UID: "user2", Asset: "ETH", Balance: decimal.NewFromFloat(3.0), Type: wallet.TypeFreeze},
|
||
{Brand: "brand1", UID: "user3", Asset: "BTC", Balance: decimal.NewFromFloat(4.5), Type: wallet.TypeAvailable},
|
||
}
|
||
|
||
err = repo.InitWallets(ctx, initData)
|
||
assert.NoError(t, err)
|
||
|
||
tests := []struct {
|
||
name string
|
||
uids []string
|
||
query repository.BalanceQuery
|
||
expected int
|
||
}{
|
||
{
|
||
name: "Query all users with BTC",
|
||
uids: []string{"user1", "user2", "user3"},
|
||
query: repository.BalanceQuery{Asset: "BTC"},
|
||
expected: 3,
|
||
},
|
||
{
|
||
name: "Query specific users with filter by type",
|
||
uids: []string{"user2"},
|
||
query: repository.BalanceQuery{Kinds: []wallet.Types{wallet.TypeAvailable}},
|
||
expected: 1,
|
||
},
|
||
{
|
||
name: "Query with no matches",
|
||
uids: []string{"user2"},
|
||
query: repository.BalanceQuery{Asset: "DOGE"},
|
||
expected: 0,
|
||
},
|
||
{
|
||
name: "Query all for user2",
|
||
uids: []string{"user2"},
|
||
query: repository.BalanceQuery{},
|
||
expected: 2,
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
result, err := repo.QueryBalancesByUIDs(ctx, tt.uids, tt.query)
|
||
assert.NoError(t, err)
|
||
assert.Len(t, result, tt.expected)
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestWalletRepository_NewDB(t *testing.T) {
|
||
repo, _, tearDown, err := SetupTestWalletRepository()
|
||
assert.NoError(t, err)
|
||
defer tearDown()
|
||
|
||
tx := repo.NewDB()
|
||
assert.NotNil(t, tx)
|
||
assert.NoError(t, tx.Commit().Error)
|
||
}
|
||
|
||
func TestWalletRepository_Transaction(t *testing.T) {
|
||
repo, _, tearDown, err := SetupTestWalletRepository()
|
||
assert.NoError(t, err)
|
||
defer tearDown()
|
||
|
||
t.Run("commit success", func(t *testing.T) {
|
||
err := repo.Transaction(func(tx *gorm.DB) error {
|
||
return nil // 模擬成功流程
|
||
})
|
||
assert.NoError(t, err)
|
||
})
|
||
|
||
t.Run("rollback due to fn error", func(t *testing.T) {
|
||
customErr := errors.New("rollback me")
|
||
err := repo.Transaction(func(tx *gorm.DB) error {
|
||
return customErr
|
||
})
|
||
assert.ErrorIs(t, err, customErr)
|
||
})
|
||
}
|
||
|
||
func TestWalletRepository_Session(t *testing.T) {
|
||
repo, _, tearDown, err := SetupTestWalletRepository()
|
||
assert.NoError(t, err)
|
||
defer tearDown()
|
||
|
||
sess := repo.Session("userX", "BTC")
|
||
assert.NotNil(t, sess)
|
||
}
|
||
|
||
func TestWalletRepository_SessionWithTx(t *testing.T) {
|
||
repo, _, tearDown, err := SetupTestWalletRepository()
|
||
assert.NoError(t, err)
|
||
defer tearDown()
|
||
|
||
tx := repo.NewDB()
|
||
defer tx.Rollback()
|
||
|
||
sess := repo.SessionWithTx(tx, "userY", "ETH")
|
||
assert.NotNil(t, sess)
|
||
}
|