在Go中使用httptest.NewServer的自定义URL

时间:2017-02-13 22:21:24

标签: unit-testing http go

我通过在Go语言中创建一个http测试服务器,在一些休息调用上运行UT。我的代码如下。

type student struct{
FirstName string
LastName string
}

func testGetStudentName() {
    testServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    response = new(student)
    response.FirstName = "Some"
    response.LastName = "Name"
    b, err := json.Marshal(response)
    if err == nil {
            fmt.Fprintln(w, string(b[:]))
        }
    }))
    defer ts.Close()
    student1 := base.getStudent("123")
    log.Print("testServerUrl",testServer.URL) //prints out http://127.0.0.1:49931 ( port changes every time this is run)

   ts.URL = "http://127.0.0.1:8099" //this assignment does not quite change the URL of the created test server.
}

在正在测试的文件中,

var baseURL = "http://originalUrl.com"
var mockUrl = "http://127.0.0.1:49855"
func Init(mockServer bool){
    if mockServer {
        baseURL = mockUrl
    }
}

func getStudent(id String){
     url := baseUrl + "/student/" + id
     req, err := http.NewRequest("GET", url, nil)
}

从我的测试中调用此init。

这将创建一个新的测试服务器并在随机端口上运行调用。我可以在我指定的端口上运行此服务器吗?

1 个答案:

答案 0 :(得分:6)

大多数应用程序使用在NewServer或NewUnstartedServer中分配的端口,因为此端口不会与计算机上使用的端口冲突。他们不是分配端口,而是将服务的基本URL设置为test server's URL

如果您确实要设置侦听端口,请执行以下操作:

// create a listener with the desired port.
l, err := net.Listen("tcp", "127.0.0.1:8080")
if err != nil {
    log.Fatal(err)
}

ts := httptest.NewUnstartedServer(handler)

// NewUnstartedServer creates a listener. Close that listener and replace 
// with the one we created.
ts.Listener.Close()
ts.Listener = l

// Start the server.
ts.Start()

// Run your tests here.

// Cleanup.
ts.Close()