go 单元测试实战

谁践踏了优雅 2021-07-24 13:28 434阅读 0赞

# 一 需求 #

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2NoZW5ncWl1bWluZw_size_16_color_FFFFFF_t_70][]

# 二 代码 #

**1 monster.go**

package monster
    
    import (
       "encoding/json"
       "fmt"
       "io/ioutil"
    )
    
    type Monster struct {
       Name  string
       Age   int
       Skill string
    }
    
    // 给 Monster 绑定方法 Store, 可以将一个 Monster 变量(对象),序列化后保存到文件中
    func (this *Monster) Store() bool {
       // 先序列化
       data, err := json.Marshal(this)
       if err != nil {
          fmt.Println("marshal err =", err)
          return false
       }
       // 保存到文件
       filePath := "d:/monster.ser"
       err = ioutil.WriteFile(filePath, data, 0666)
       if err != nil {
          fmt.Println("write file err =", err)
          return false
       }
       return true
    }
    
    // 给 Monster 绑定方法 ReStore , 可以将一个序列化的 Monster ,从文件中读取，并反序列化为 Monster 对象。
    func (this *Monster) ReStore() bool {
       // 1 先从文件中，读取序列化的字符串
       filePath := "d:/monster.ser"
       data, err := ioutil.ReadFile(filePath)
       if err != nil {
          fmt.Println("ReadFile err =", err)
          return false
       }
       // 2 读取到data []byte ,反序列化
       err = json.Unmarshal(data, this)
       if err != nil {
          fmt.Println("UnMarshal err =", err)
          return false
       }
       return true
    }

**2 monster\_test.go**

package monster
    
    import (
       "testing"
    )
    
    // 测试用例,测试 Store 方法
    func TestStore(t *testing.T) {
       //先创建一个Monster 实例
       monster := &Monster{
          Name:  "红孩儿",
          Age:   10,
          Skill: "吐火.",
       }
       res := monster.Store()
       if !res {
          t.Fatalf("monster.Store() 错误，希望为=%v 实际为=%v", true, res)
       }
       t.Logf("monster.Store() 测试成功!")
    }
    
    func TestReStore(t *testing.T) {
       // 先创建一个 Monster 实例，不需要指定字段的值
       var monster = &Monster{}
       res := monster.ReStore()
       if !res {
          t.Fatalf("monster.ReStore() 错误，希望为=%v 实际为=%v", true, res)
       }
       // 进一步判断
       if monster.Name != "红孩儿" {
          t.Fatalf("monster.ReStore() 错误，希望为=%v 实际为=%v", "红孩儿", monster.Name)
       }
       t.Logf("monster.ReStore() 测试成功!")
    }

**三 测试**

> E:\\gocode\\src\\chapter15\\testcase02>go test -v
> 
> === RUN   TestStore
> 
> \--- PASS: TestStore (0.00s)
> 
>         monster\_test.go:20: monster.Store() 测试成功!
> 
> === RUN   TestReStore
> 
> \--- PASS: TestReStore (0.00s)
> 
>         monster\_test.go:34: monster.ReStore() 测试成功!
> 
> PASS
> 
> ok      chapter15/testcase02    1.132s

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2NoZW5ncWl1bWluZw_size_16_color_FFFFFF_t_70]: /images/20210724/edae8e1bc74c4cb1bebfb6449d306f1f.png