dqls67891
2017-06-26 20:49
浏览 37
已采纳

如何在go中测试端点?

I wrote a small test function in go. I'm having hard time in making a request to actual endpoint and test it. I tried importing the file which has the handler function (I think I'm trying to import whole directory : import (".")). Both my project.go and handler_test.go are in the same directory (I don't think this matters). Could someone give me heads up so that I can write more tests. Here is my project.go:

package main

import (
    "encoding/json"
    "net/http"

    "github.com/gorilla/mux"
    "github.com/rs/cors"

)

type Person struct {
    ID        string   `json:"id,omitempty"`
    Firstname string   `json:"firstname,omitempty"`
    Lastname  string   `json:"lastname,omitempty"`
    Address   *Address `json:"address,omitempty"`
}

type Address struct {
    City  string `json:"city,omitempty"`
    State string `json:"state,omitempty"`
}

var people []Person;

func GetPersonEndpoint(w http.ResponseWriter, req *http.Request) {
    params := mux.Vars(req)
    for _, item := range people {
        if item.ID == params["id"] {
            json.NewEncoder(w).Encode(item)
            return
        }
    }
    json.NewEncoder(w).Encode(&Person{})
}


func GetPeopleEndpoint(w http.ResponseWriter, req *http.Request) {
 json.NewEncoder(w).Encode(people)
}

func CreatePersonEndpoint(w http.ResponseWriter, req *http.Request) {
    params := mux.Vars(req)
    var person Person
    _ = json.NewDecoder(req.Body).Decode(&person)
    person.ID = params["id"]
    people = append(people, person)
    json.NewEncoder(w).Encode(people)
}

func DeletePersonEndpoint(w http.ResponseWriter, req *http.Request) {
    params := mux.Vars(req)
    for index, item := range people {
        if item.ID == params["id"] {
            people = append(people[:index], people[index+1:]...)
            break
        }
    }
    json.NewEncoder(w).Encode(people)
}

func main() {
    Router := mux.NewRouter()
    people = append(people, Person{ID: "1", Firstname: "sarath", Lastname: "v", Address: &Address{City: "sunnyvale", State: "CA"}})
    people = append(people, Person{ID: "2", Firstname: "dead", Lastname: "pool"})

    // router.PathPrefix("/tmpfiles/").Handler(http.StripPrefix("/tmpfiles/", http.FileServer(http.Dir("."))))

    Router.HandleFunc("/people", GetPeopleEndpoint).Methods("GET")
    Router.HandleFunc("/people/{id}", GetPersonEndpoint).Methods("GET")
    Router.HandleFunc("/people/{id}", CreatePersonEndpoint).Methods("POST")

   c := cors.New(cors.Options{
    AllowedOrigins: []string{"http://localhost:3000"},
    AllowCredentials: true,
})

// Insert the middleware
   handler := c.Handler(Router)
   http.ListenAndServe(":12345", handler)

}

Here is my handler_test.go. In this code I'm testing GetPersonEndPoint.

package main

import (
  "."
  "net/http"
  "net/http/httptest"
  "testing"
  "encoding/json"
)

func checkResponseCode(t *testing.T, expected, actual int) {
    if expected != actual {
        t.Errorf("Expected response code %d. Got %d
", expected, actual)
    }
}

func executeRequest(req *http.Request) *httptest.ResponseRecorder {
    rr := httptest.NewRecorder()
     handler := http.HandlerFunc(GetPersonEndpoint)
     handler.ServeHTTP(rr, req)
     if status := rr.Code; status != http.StatusOK {
       fmt.Printf("Handler returned wrong status code: got %v want %v" , status, http.statusOk);
      }
    return rr
}

func TestGetPersonEndPoint(t *testing.T){
  req, _ := http.NewRequest("GET", "/people/5", nil)
  response := executeRequest(req)
  checkResponseCode(t, http.StatusNotFound, response.Code)
   var m map[string]string
   json.Unmarshal(response.Body.Bytes(), &m)
  if m["error"] != "Product not found" {
        t.Errorf("Expected the 'error' key of the response to be set to 'Product not found'. Got '%s'", m["error"])
    }
}

And finally this is the error:

./new.go:14: main redeclared in this block
    previous declaration at ./myproject.go:62
./new.go:20: not enough arguments in call to server.ListenAndServeTLS
    have ()
    want (string, string)

1条回答 默认 最新

相关推荐 更多相似问题