func GetUserById(rw http.ResponseWriter, r *http.Request, enc encoding.Encoder, params martini.Params) string {
var err error
var user customer.CustomerUser
qs := r.URL.Query()
key := qs.Get("key")
if params["id"] != "" {
user.Id = params["id"]
} else if r.FormValue("id") != "" {
user.Id = r.FormValue("id")
} else {
err = errors.New("Trouble getting customer user ID")
apierror.GenerateError("Trouble getting customer user ID", err, rw, r)
return ""
}
if err = user.Get(key); err != nil {
apierror.GenerateError("Trouble getting customer user", err, rw, r)
return ""
}
return encoding.Must(enc.Encode(user))
}
func UpdateCustomerUser(rw http.ResponseWriter, r *http.Request, enc encoding.Encoder, params martini.Params) string {
var err error
var cu customer.CustomerUser
qs := r.URL.Query()
key := qs.Get("key")
if params["id"] != "" {
cu.Id = params["id"]
} else if r.FormValue("id") != "" {
cu.Id = r.FormValue("id")
} else {
err = errors.New("Trouble getting customer user ID")
apierror.GenerateError("Trouble getting customer user ID", err, rw, r)
return ""
}
if err = cu.Get(key); err != nil {
apierror.GenerateError("Trouble getting customer user", err, rw, r)
return ""
}
if strings.ToLower(r.Header.Get("Content-Type")) == "application/json" {
var data []byte
if data, err = ioutil.ReadAll(r.Body); err != nil {
apierror.GenerateError("Trouble reading request body while updating customer user", err, rw, r)
return ""
}
if err = json.Unmarshal(data, &cu); err != nil {
apierror.GenerateError("Trouble unmarshalling json request body while updating customer user", err, rw, r)
return ""
}
} else {
name := r.FormValue("name")
email := r.FormValue("email")
isActive := r.FormValue("isActive")
locationID := r.FormValue("locationID")
isSudo := r.FormValue("isSudo")
notCustomer := r.FormValue("notCustomer")
if name != "" {
cu.Name = name
}
if email != "" {
cu.Email = email
}
if isActive != "" {
if cu.Active, err = strconv.ParseBool(isActive); err != nil {
cu.Active = false
}
}
if locationID != "" {
if cu.Location.Id, err = strconv.Atoi(locationID); err != nil {
apierror.GenerateError("Trouble getting location ID", err, rw, r)
return ""
}
}
if isSudo != "" {
if cu.Sudo, err = strconv.ParseBool(isSudo); err != nil {
cu.Sudo = false
}
}
if notCustomer != "" {
if cu.NotCustomer, err = strconv.ParseBool(notCustomer); err != nil {
cu.NotCustomer = false
}
}
}
if err = cu.UpdateCustomerUser(); err != nil {
apierror.GenerateError("Trouble updating customer user", err, rw, r)
return ""
}
return encoding.Must(enc.Encode(cu))
}
func TestCustomer(t *testing.T) {
var err error
var c customer.Customer
var cu customer.CustomerUser
once.Do(initDtx)
defer apicontextmock.DeMock(dtx)
cu.Id = dtx.UserID
err = cu.Get(dtx.APIKey)
if err != nil {
t.Log(err)
}
c.Users = append(c.Users, cu)
Convey("Testing customer/Customer", t, func() {
//test create customer
c.Name = "Jason Voorhees"
c.Email = "*****@*****.**"
bodyBytes, _ := json.Marshal(c)
bodyJson := bytes.NewReader(bodyBytes)
thyme := time.Now()
testThatHttp.Request("put", "/customer", "", "?key=", SaveCustomer, bodyJson, "application/json")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &c)
So(err, ShouldBeNil)
So(c, ShouldHaveSameTypeAs, customer.Customer{})
So(c.Id, ShouldBeGreaterThan, 0)
//test update customer
c.Fax = "666-1313"
c.State.Id = 1
bodyBytes, _ = json.Marshal(c)
bodyJson = bytes.NewReader(bodyBytes)
thyme = time.Now()
testThatHttp.Request("put", "/customer/", ":id", strconv.Itoa(c.Id)+"?key=", SaveCustomer, bodyJson, "application/json")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &c)
So(err, ShouldBeNil)
So(c, ShouldHaveSameTypeAs, customer.Customer{})
So(c.Id, ShouldBeGreaterThan, 0)
thyme = time.Now()
testThatHttp.RequestWithDtx("post", "/customer", "", "?key=", GetCustomer, nil, "", dtx)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &c)
So(err, ShouldBeNil)
So(c, ShouldHaveSameTypeAs, customer.Customer{})
So(c.Id, ShouldBeGreaterThan, 0)
// get customer locations
thyme = time.Now()
testThatHttp.RequestWithDtx("get", "/customer/locations", "", "?key=", GetLocations, nil, "", dtx)
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()*6)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &c.Locations)
So(err, ShouldBeNil)
So(c.Locations, ShouldHaveSameTypeAs, []customer.CustomerLocation{})
// //get user
thyme = time.Now()
testThatHttp.RequestWithDtx("post", "/customer/user", "", "?key="+dtx.APIKey, GetUser, nil, "", nil)
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &cu)
So(err, ShouldBeNil)
So(cu, ShouldHaveSameTypeAs, customer.CustomerUser{})
//get users
thyme = time.Now()
testThatHttp.RequestWithDtx("get", "/customer/users", "", "?key=", GetUsers, nil, "", dtx)
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()*5)
So(testThatHttp.Response.Code, ShouldEqual, 200)
var cus []customer.CustomerUser
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &cus)
So(err, ShouldBeNil)
So(cus, ShouldHaveSameTypeAs, []customer.CustomerUser{})
// get customer price
// price.CustID = c.Id
// price.Create()
// thyme = time.Now()
// testThatHttp.Request("get", "/new/customer/price/", ":id", strconv.Itoa(p.ID)+"?key="+apiKey, GetCustomerPrice, nil, "")
// So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
// So(testThatHttp.Response.Code, ShouldEqual, 200)
// var price float64
// err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &price)
// So(err, ShouldBeNil)
// So(price, ShouldHaveSameTypeAs, 7.1)
// //get customer cart reference
// ci.CustID = c.Id
// ci.Create()
// thyme = time.Now()
// testThatHttp.Request("get", "/new/customer/cartRef/", ":id", strconv.Itoa(p.ID)+"?key="+apiKey, GetCustomerCartReference, nil, "")
// So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
// So(testThatHttp.Response.Code, ShouldEqual, 200)
// var reference int
// err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &reference)
// So(err, ShouldBeNil)
// So(reference, ShouldHaveSameTypeAs, 7)
//test delete customer
thyme = time.Now()
testThatHttp.Request("delete", "/customer/", ":id", strconv.Itoa(c.Id)+"?key=", DeleteCustomer, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &c)
So(err, ShouldBeNil)
So(c, ShouldHaveSameTypeAs, customer.Customer{})
So(c.Id, ShouldBeGreaterThan, 0)
})
//cleanup
err = cu.Delete()
err = c.Delete()
}
