(译)如何测试 HTTPS 服务

原文:How to test Go HTTPS Services

译文为第一人称视角:
在DNSimple的微服务中,在Go的文档中很难找到描述有关如何测试HTTP(S)端服务的文档。大部分的文章,我发现要么是不完整,要么就是缺少一些测试样例,特别是有关集成测试。

以下是我写的一些服务样例。我希望这个分享,在你的下一个Go项目中会节省一些测试的时间。如果你需要一些灵感来扩这个测试,请查阅我们的Go客户端接口,它会影响域、域名服务器和证书接口的交互。

开始编写测试代码:

# microservice.go
package microservice

import (
    "fmt"
    "net/http"
)

func NewServer(port string) *http.Server {
    addr := fmt.Sprintf(":%s", port)

    mux := http.NewServeMux()
    mux.HandleFunc("/", handler)

    return &http.Server{
        Addr:    addr,
        Handler: mux,
    }
}

func handler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Hello World")
}

这是一个简单版本的项目。我们将会设置一些暴露出去的函数来举例这个服务并使用net/http包 和写一些私有函数,来处理这些接口的方法。
现在让我们来看写这个单元测。他关键的代码部分是使用了httptest包:

# microservice_test.go
package microservice

import (
    "bytes"
    "crypto/tls"
    "io/ioutil"
    "net/http"
    "net/http/httptest"
    "testing"
    "time"
)

var (
    port = "8080"
)

//
// Unit Tests
//

func TestHandler(t *testing.T) {
    expected := []byte("Hello World")

    req, err := http.NewRequest("GET", buildUrl("/"), nil)
    if err != nil {
        t.Fatal(err)
    }

    res := httptest.NewRecorder()

    handler(res, req)

    if res.Code != http.StatusOK {
        t.Errorf("Response code was %v; want 200", res.Code)
    }

    if bytes.Compare(expected, res.Body.Bytes()) != 0 {
        t.Errorf("Response body was '%v'; want '%v'", expected, res.Body)
    }
}

func buildUrl(path string) string {
    return urlFor("http", port, path)
}

func urlFor(scheme string, serverPort string, path string) string {
    return scheme + "://localhost:" + serverPort + path
}

前面几行设置了预设的主体和一个Get请求。它是Go的标准HTTP测试库的一部分,实例化了一个新的*httptest.ResponseRecorder方法:

ResponseRecorder方法实现了http.ResponseWriter,这个记录是为接下来的变化而进行的测试。
有了这个对象,我们就可以直接调用程序处理,然后我们可以维护它,以至于这些返回的HTTP代码和主体都是如预期中的一样。

集成测试:
单元测试是非常好的,但是如果我们想包含SSL/TLS握手来测试微服务实际的HTTP(S)请求呢?
在开始之前,我们需要一个子签名的测试证书:

go run $GOROOT/src/crypto/tls/generate_cert.go --rsa-bits 1024 --host 127.0.0.1,::1,localhost --ca --start-date "Jan 1 00:00:00 1970" --duration=1000000h

这个命令将会创建两个文件:

➜ ls *.pem
cert.pem key.pem

现在开始写我们的集成测试:

func TestHTTPSServer(t *testing.T) {
    srv := NewServer(httpsPort)
    go srv.ListenAndServeTLS("cert.pem", "key.pem")
    defer srv.Close()
    time.Sleep(100 * time.Millisecond)

    tr := &http.Transport{
        TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
    }

    client := &http.Client{Transport: tr}
    res, err := client.Get(buildSecureUrl("/"))

    if err != nil {
        t.Fatal(err)
    }

    defer res.Body.Close()

    if res.StatusCode != http.StatusOK {
        t.Errorf("Response code was %v; want 200", res.StatusCode)
    }

    body, err := ioutil.ReadAll(res.Body)
    if err != nil {
        t.Fatal(err)
    }

    expected := []byte("Hello World")

    if bytes.Compare(expected, body) != 0 {
        t.Errorf("Response body was '%v'; want '%v'", expected, body)
    }
}

首先我们构建服务器(srv),然后我们启动它,分离在一个线程中,否则这个进程将会挂起在这一行。我们对这个进程进行了一些睡眠,好让这个服务能够启动。
现在,我们可以构造这个客户端是InsecureSkipVerify: true,就好像我们说的一样客户端不需要验证签名就可以进行SSl握手。请记住,我们的签名不是由一个已知的CA机构颁发的,而是来自于我们自己的计算机。
最后一个测试部分,我么可以正真的执行一个HTTP请求来客户端,和我们要维护的代码以及响应主体。

总结:

善于阅读Go的官方文档,Go的标准库拥有所有的测试工具。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 211,743评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,296评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,285评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,485评论 1 283
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,581评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,821评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,960评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,719评论 0 266
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,186评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,516评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,650评论 1 340
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,329评论 4 330
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,936评论 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,757评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,991评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,370评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,527评论 2 349

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,633评论 18 139
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 171,777评论 25 707
  • 文章来自:http://blog.csdn.net/mj813/article/details/52451355 ...
    好大一只鹏阅读 9,189评论 2 126
  • 七律·巴渝风云 巴渝自古多骁汉, 暴葬元朝“上帝鞭”。 万弹狂轰围壮士, 千声怒吼御倭蛮。 碧园谈判唇枪热, 洞...
    任尔风云我自逍阅读 515评论 0 1
  • 把有序数组打碎成无序数组,相同的就自然组合了 https://segmentfault.com/q/1010000...
    angelababa阅读 2,826评论 0 0