add test

src/server/handler_user_test.go

@@ -0,0 +1,283 @@
+package server
+
+import (
+	"encoding/json"
+	"math/rand"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+	"time"
+
+	restful "github.com/emicklei/go-restful"
+	"github.com/linkernetworks/mongo"
+	"github.com/linkernetworks/vortex/src/config"
+	"github.com/linkernetworks/vortex/src/entity"
+	"github.com/linkernetworks/vortex/src/serviceprovider"
+	"github.com/stretchr/testify/suite"
+	"gopkg.in/mgo.v2/bson"
+)
+
+func init() {
+	rand.Seed(time.Now().UnixNano())
+}
+
+type UserTestSuite struct {
+	suite.Suite
+	sp      *serviceprovider.Container
+	wc      *restful.Container
+	session *mongo.Session
+}
+
+func (suite *UserTestSuite) SetupSuite() {
+	cf := config.MustRead("../../config/testing.json")
+	sp := serviceprovider.NewForTesting(cf)
+
+	suite.sp = sp
+	//init session
+	suite.session = sp.Mongo.NewSession()
+	//init restful container
+	suite.wc = restful.NewContainer()
+	service := newServiceService(suite.sp)
+	suite.wc.Add(service)
+}
+
+func (suite *UserTestSuite) TearDownSuite() {}
+
+func TestUserSuite(t *testing.T) {
+	suite.Run(t, new(UserTestSuite))
+}
+
+func (suite *UserTestSuite) TestCreateUser() {
+	user := entity.User{
+		ID:          bson.NewObjectId(),
+		UserName:    "John Doe",
+		Email:       "[email protected]",
+		Password:    "p@ssw0rd",
+		FirstName:   "John",
+		LastName:    "Doe",
+		PhoneNumber: "0911123456",
+		CreatedAt:   &time.Time{},
+	}
+
+	bodyBytes, err := json.MarshalIndent(user, "", "  ")
+	suite.NoError(err)
+
+	bodyReader := strings.NewReader(string(bodyBytes))
+	httpRequest, err := http.NewRequest("POST", "http://localhost:7890/v1/users", bodyReader)
+	suite.NoError(err)
+
+	httpRequest.Header.Add("Content-Type", "application/json")
+	httpWriter := httptest.NewRecorder()
+	suite.wc.Dispatch(httpWriter, httpRequest)
+	assertResponseCode(suite.T(), http.StatusOK, httpWriter)
+	defer suite.session.Remove(entity.UserCollectionName, "_id", user.ID)
+
+	// load data to check
+	retUser := entity.User{}
+	err = suite.session.FindOne(entity.UserCollectionName, bson.M{"_id": user.id}, &retUser)
+	suite.NoError(err)
+	suite.NotEqual("", retUser.ID)
+	suite.Equal(user.Username, retUser.Username)
+	suite.Equal(user.Email, retUser.Email)
+
+	// We use the new write but empty input which will cause the readEntity Error
+	httpWriter = httptest.NewRecorder()
+	suite.wc.Dispatch(httpWriter, httpRequest)
+	assertResponseCode(suite.T(), http.StatusBadRequest, httpWriter)
+	// Create again and it should fail since the name exist
+	bodyReader = strings.NewReader(string(bodyBytes))
+	httpRequest, err = http.NewRequest("POST", "http://localhost:7890/v1/users", bodyReader)
+	suite.NoError(err)
+	httpRequest.Header.Add("Content-Type", "application/json")
+	httpWriter = httptest.NewRecorder()
+	suite.wc.Dispatch(httpWriter, httpRequest)
+	assertResponseCode(suite.T(), http.StatusInternalServerError, httpWriter)
+}
+
+func (suite *UserTestSuite) TestCreateUserFail() {
+	user := entity.User{
+		ID:          bson.NewObjectId(),
+		UserName:    "John Doe",
+		FirstName:   "John",
+		LastName:    "Doe",
+		PhoneNumber: "0911123456",
+		CreatedAt:   &time.Time{},
+	}
+
+	bodyBytes, err := json.MarshalIndent(user, "", "  ")
+	suite.NoError(err)
+
+	bodyReader := strings.NewReader(string(bodyBytes))
+	httpRequest, err := http.NewRequest("POST", "http://localhost:7890/v1/users", bodyReader)
+	suite.NoError(err)
+
+	httpRequest.Header.Add("Content-Type", "application/json")
+	httpWriter := httptest.NewRecorder()
+	suite.wc.Dispatch(httpWriter, httpRequest)
+	assertResponseCode(suite.T(), http.StatusBadRequest, httpWriter)
+}
+
+func (suite *UserTestSuite) TestDeleteService() {
+	user := entity.User{
+		ID:          bson.NewObjectId(),
+		UserName:    "John Doe",
+		Email:       "[email protected]",
+		Password:    "p@ssw0rd",
+		FirstName:   "John",
+		LastName:    "Doe",
+		PhoneNumber: "0911123456",
+		CreatedAt:   &time.Time{},
+	}
+
+	err = suite.session.Insert(entity.UserCollectionName, &user)
+	suite.NoError(err)
+
+	bodyBytes, err := json.MarshalIndent(user, "", "  ")
+	suite.NoError(err)
+
+	bodyReader := strings.NewReader(string(bodyBytes))
+	httpRequest, err := http.NewRequest("DELETE", "http://localhost:7890/v1/users/"+user.ID.Hex(), bodyReader)
+	suite.NoError(err)
+
+	httpRequest.Header.Add("Content-Type", "application/json")
+	httpWriter := httptest.NewRecorder()
+	suite.wc.Dispatch(httpWriter, httpRequest)
+	assertResponseCode(suite.T(), http.StatusOK, httpWriter)
+
+	n, err := suite.session.Count(entity.UserCollectionName, bson.M{"_id": user.ID})
+	suite.NoError(err)
+	suite.Equal(0, n)
+}
+
+func (suite *UserTestSuite) TestDeleteUserWithInvalidID() {
+	httpRequest, err := http.NewRequest("DELETE", "http://localhost:7890/v1/users/"+bson.NewObjectId().Hex(), nil)
+	suite.NoError(err)
+
+	httpRequest.Header.Add("Content-Type", "application/json")
+	httpWriter := httptest.NewRecorder()
+	suite.wc.Dispatch(httpWriter, httpRequest)
+	assertResponseCode(suite.T(), http.StatusBadRequest, httpWriter)
+}
+
+// For Get/List, we only return mongo document
+func (suite *UserTestSuite) TestGetUser() {
+	user := entity.User{
+		ID:          bson.NewObjectId(),
+		UserName:    "John Doe",
+		Email:       "[email protected]",
+		Password:    "p@ssw0rd",
+		FirstName:   "John",
+		LastName:    "Doe",
+		PhoneNumber: "0911123456",
+		CreatedAt:   &time.Time{},
+	}
+	// Create data into mongo manually
+	suite.session.C(entity.UserCollectionName).Insert(user)
+	defer suite.session.Remove(entity.UserCollectionName, "_id", user.ID)
+
+	httpRequest, err := http.NewRequest("GET", "http://localhost:7890/v1/users/"+user.ID.Hex(), nil)
+	suite.NoError(err)
+
+	httpWriter := httptest.NewRecorder()
+	suite.wc.Dispatch(httpWriter, httpRequest)
+	assertResponseCode(suite.T(), http.StatusOK, httpWriter)
+
+	retUser = entity.User{}
+	err = json.Unmarshal(httpWriter.Body.Bytes(), &retUser)
+	suite.NoError(err)
+	suite.Equal(user.Name, retUser.Name)
+	suite.Equal(user.Email, retUser.Email)
+}
+
+func (suite *UserTestSuite) TestGetUserWithInvalidID() {
+	// Get data with non-exits ID
+	httpRequest, err := http.NewRequest("GET", "http://localhost:7890/v1/users/"+bson.NewObjectId().Hex(), nil)
+	suite.NoError(err)
+
+	httpWriter := httptest.NewRecorder()
+	suite.wc.Dispatch(httpWriter, httpRequest)
+	assertResponseCode(suite.T(), http.StatusNotFound, httpWriter)
+}
+
+func (suite *UserTestSuite) TestListService() {
+	users := []entity.User{}
+	count := 3
+	for i := 0; i < count; i++ {
+		users = append(users, entity.User{
+			ID:          bson.NewObjectId(),
+			UserName:    "John Doe",
+			Email:       "[email protected]",
+			Password:    "p@ssw0rd",
+			FirstName:   "John",
+			LastName:    "Doe",
+			PhoneNumber: "0911123456",
+		})
+	}
+
+	for _, u := range users {
+		suite.session.C(entity.UserCollectionName).Insert(u)
+		defer suite.session.Remove(entity.UserCollectionName, "_id", u.ID)
+	}
+
+	testCases := []struct {
+		page       string
+		pageSize   string
+		expectSize int
+	}{
+		{"", "", count},
+		{"1", "1", count},
+		{"1", "3", count},
+	}
+
+	for _, tc := range testCases {
+		caseName := "page:pageSize" + tc.page + ":" + tc.pageSize
+		suite.T().Run(caseName, func(t *testing.T) {
+			// list data by default page and page_size
+			url := "http://localhost:7890/v1/users/"
+			if tc.page != "" || tc.pageSize != "" {
+				url = "http://localhost:7890/v1/users?"
+				url += "page=" + tc.page + "%" + "page_size" + tc.pageSize
+			}
+			httpRequest, err := http.NewRequest("GET", url, nil)
+			suite.NoError(err)
+
+			httpWriter := httptest.NewRecorder()
+			suite.wc.Dispatch(httpWriter, httpRequest)
+			assertResponseCode(suite.T(), http.StatusOK, httpWriter)
+
+			retUsers := []entity.User{}
+			err = json.Unmarshal(httpWriter.Body.Bytes(), &retUsers)
+			suite.NoError(err)
+			suite.Equal(tc.expectSize, len(retUsers))
+			for i, u := range retUsers {
+				suite.Equal(users[i].Username, u.Username)
+				suite.Equal(users[i].Email, u.Email)
+			}
+		})
+	}
+}
+
+func (suite *UserTestSuite) TestListServiceWithInvalidPage() {
+	// Get data with non-exits ID
+	httpRequest, err := http.NewRequest("GET", "http://localhost:7890/v1/users?page=asdd", nil)
+	suite.NoError(err)
+
+	httpWriter := httptest.NewRecorder()
+	suite.wc.Dispatch(httpWriter, httpRequest)
+	assertResponseCode(suite.T(), http.StatusBadRequest, httpWriter)
+
+	httpRequest, err = http.NewRequest("GET", "http://localhost:7890/v1/users?page_size=asdd", nil)
+	suite.NoError(err)
+
+	httpWriter = httptest.NewRecorder()
+	suite.wc.Dispatch(httpWriter, httpRequest)
+	assertResponseCode(suite.T(), http.StatusBadRequest, httpWriter)
+
+	httpRequest, err = http.NewRequest("GET", "http://localhost:7890/v1/users?page=-1", nil)
+	suite.NoError(err)
+
+	httpWriter = httptest.NewRecorder()
+	suite.wc.Dispatch(httpWriter, httpRequest)
+	assertResponseCode(suite.T(), http.StatusInternalServerError, httpWriter)
+}