hotime/example/app/mysql_test.go

package app

import (
	"fmt"

	. "code.hoteas.com/golang/hotime"
	. "code.hoteas.com/golang/hotime/common"
)

var MysqlTest = CtrTest{
	"tables": {Desc: "查询 MySQL 所有表", Func: func(a *Api) {
		if a.DB().Type != "mysql" {
			return
		}
		a.Verify(func(a *Api) error {
			result := a.Resp().GetBody().GetMap("result")
			if result == nil {
				return fmt.Errorf("result 为 nil")
			}
			tables := result.GetSlice("tables")
			if tables == nil {
				return fmt.Errorf("tables 字段缺失")
			}
			if len(tables) == 0 {
				return fmt.Errorf("MySQL 表列表为空，至少应有测试表")
			}
			return nil
		}).Get("SHOW TABLES", 0, Map{
			"tables": Slice{},
		})
	}},

	"describe": {Desc: "查询 MySQL 表结构", Func: func(a *Api) {
		if a.DB().Type != "mysql" {
			return
		}

		// ======== 错误用例 ========
		a.Get("不传table参数", 1, "请提供 table 参数")

		// ======== 正确请求 + 结构校验 + Verify ========
		a.Query(Map{"table": "admin"}).
			Verify(func(a *Api) error {
				result := a.Resp().GetBody().GetMap("result")
				if result == nil {
					return fmt.Errorf("result 为 nil")
				}
				if result.GetString("table") != "admin" {
					return fmt.Errorf("table 期望 'admin', 实际 '%s'", result.GetString("table"))
				}
				columns := result.GetSlice("columns")
				if len(columns) == 0 {
					return fmt.Errorf("columns 为空，admin 表应有字段定义")
				}
				return nil
			}).
			Get("DESCRIBE admin", 0, Map{
				"table": "sample", "columns": Slice{}, "sample_data": Slice{},
			})
	}},

	"rawsql": {Desc: "MySQL 原生 SQL 测试", Func: func(a *Api) {
		if a.DB().Type != "mysql" {
			return
		}
		a.Verify(func(a *Api) error {
			result := a.Resp().GetBody().GetMap("result")
			if result == nil {
				return fmt.Errorf("result 为 nil")
			}
			if !result.GetBool("success") {
				return fmt.Errorf("MySQL 原生 SQL 测试未通过")
			}
			tests := result.GetSlice("tests")
			if len(tests) == 0 {
				return fmt.Errorf("tests 数组为空")
			}
			for i := 0; i < len(tests); i++ {
				item := tests.GetMap(i)
				if item != nil && !item.GetBool("result") {
					return fmt.Errorf("子项 '%s' 失败", item.GetString("name"))
				}
			}
			return nil
		}).Get("MySQL原生SQL", 0, Map{
			"name": "sample", "success": true,
			"tests": Slice{Map{"name": "sample", "result": true}},
		})
	}},
}