Golang中的测试驱动开发实践案例分享

package main

import (
    "testing"
)

// 要测试的函数
func Add(a, b int) int {
    return a + b
}

func TestAdd(t *testing.T) {
    result := Add(3, 4)
    if result != 7 {
        t.Errorf("Add(3, 4) = %d; want 7", result)
    }
}

package main

import (
    "net/http"
    "net/http/httptest"
    "testing"
)

func TestHandler(t *testing.T) {
    // 创建一个测试用的HTTP请求
    req, err := http.NewRequest("GET", "/hello", nil)
    if err != nil {
        t.Fatal(err)
    }

    // 创建一个HTTP测试服务器
    rr := httptest.NewRecorder()
    handler := http.HandlerFunc(HelloHandler)

    // 使用HTTP测试服务器处理请求
    handler.ServeHTTP(rr, req)

    // 检查HTTP响应的状态码
    if status := rr.Code; status != http.StatusOK {
        t.Errorf("handler returned wrong status code: got %v want %v", status, http.StatusOK)
    }

    // 检查HTTP响应的正文内容
    expected := "Hello, World!"
    if rr.Body.String() != expected {
        t.Errorf("handler returned unexpected body: got %v want %v", rr.Body.String(), expected)
    }
}

func HelloHandler(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Hello, World!"))
}

package main

import (
    "testing"

    "github.com/stretchr/testify/assert"
)

func TestFizzBuzz(t *testing.T) {
    assert := assert.New(t)

    input := 15
    expect := "FizzBuzz"
    result := fizzBuzz(input)
    assert.Equalf(expect, result, "fizzBuzz(%d) = %s; want %s", input, result, expect)
}

func fizzBuzz(n int) string {
    if n%3 == 0 && n%5 == 0 {
        return "FizzBuzz"
    } else if n%3 == 0 {
        return "Fizz"
    } else if n%5 == 0 {
        return "Buzz"
    } else {
        return strconv.Itoa(n)
    }
}