Golang反射在自动化测试中的应用

reflect.ValueOf

FieldByName

MethodByName

User

ID

Name

Email

User{ID: 1, Name: "TestUser", Email: "test@example.com"}

package main

import (
    "fmt"
    "reflect"
    "time"
)

type User struct {
    ID        int
    Name      string
    Email     string
    IsActive  bool
    CreatedAt time.Time
    // internalSecret string // 未导出字段，下面会讨论如何处理
}

// 假设这是一个简单的动态数据填充函数
func fillStruct(s interface{}) {
    v := reflect.ValueOf(s).Elem() // 获取可设置的值
    t := v.Type()

    for i := 0; i < t.NumField(); i++ {
        field := v.Field(i)
        fieldType := t.Field(i)

        if !field.CanSet() { // 无法设置的字段（如未导出字段）跳过
            continue
        }

        switch fieldType.Type.Kind() {
        case reflect.Int:
            field.SetInt(int64(i + 1)) // 简单填充
        case reflect.String:
            field.SetString(fmt.Sprintf("%s_%d", fieldType.Name, i))
        case reflect.Bool:
            field.SetBool(i%2 == 0)
        case reflect.Struct:
            if fieldType.Type == reflect.TypeOf(time.Time{}) {
                field.Set(reflect.ValueOf(time.Now()))
            }
            // 可以在这里递归调用fillStruct处理嵌套结构体
        }
    }
}

func ExampleFillStruct() {
    user := &User{}
    fillStruct(user)
    fmt.Printf("%+v\n", user)
    // 实际输出的时间会动态变化，这里只是示例结构
    // Output: {ID:1 Name:Name_1 Email:Email_2 IsActive:true CreatedAt:2023-10-27 10:00:00 +0000 UTC}
}

package main

import (
    "fmt"
    "reflect"
    "testing" // 引入testing包，通常在测试文件中使用
    "unsafe"  // 用于访问未导出字段，需谨慎使用
)

type myService struct {
    secretKey string // 未导出字段
    counter   int
}

func (s *myService) doSomethingInternal() string { // 未导出方法
    s.counter++
    return "done with " + s.secretKey
}

// 模拟测试函数，通常在_test.go文件中
func TestMyServiceInternal(t *testing.T) {
    service := &myService{secretKey: "initial_secret", counter: 0}

    // 1. 访问并修改未导出字段 (需要 unsafe 包，非常规操作)
    v := reflect.ValueOf(service).Elem()
    secretField := v.FieldByName("secretKey")

    if secretField.IsValid() {
        // 对于未导出字段，secretField.CanSet() 通常是 false。
        // 要修改它，需要 unsafe 包来获取其内存地址。
        ptrToSecretKey := unsafe.Pointer(secretField.UnsafeAddr())
        realSecretKeyPtr := (*string)(ptrToSecretKey)
        *realSecretKeyPtr = "new_secret_value"
        fmt.Println("Modified secretKey via unsafe:", service.secretKey)
    } else {
        t.Log("Could not find or access 'secretKey' field.")
    }

    // 2. 调用未导出方法
    fmt.Println("Before doSomethingInternal:", service.counter)
    method := v.MethodByName("doSomethingInternal")
    if method.IsValid() {
        results := method.Call(nil) // 调用无参数方法
        fmt.Println("After doSomethingInternal:", service.counter, "Result:", results[0].String())
    } else {
        t.Errorf("Method doSomethingInternal not found")
    }

    // 验证 counter 是否增加
    if service.counter != 1 {
        t.Errorf("Expected counter to be 1, got %d", service.counter)
    }
}

unsafe

unsafe