//already have vehicleID (vcdb_vehicle.ID)? get parts
func (v *CurtVehicle) GetPartsFromVehicleConfig(dtx *apicontext.DataContext) (ps []products.Part, err error) {
//get parts
var p products.Part
//get part id
db, err := sql.Open("mysql", database.ConnectionString())
if err != nil {
return ps, err
}
defer db.Close()
stmt, err := db.Prepare(getPartID)
if err != nil {
return ps, err
}
defer stmt.Close()
res, err := stmt.Query(v.ID)
for res.Next() {
err = res.Scan(&p.ID)
if err != nil {
return ps, err
}
//get part -- adds some weight
err = p.FromDatabase(getBrandsFromDTX(dtx))
if err != nil {
return ps, err
}
ps = append(ps, p)
}
defer res.Close()
return ps, err
}
func GetRelated(w http.ResponseWriter, r *http.Request, params martini.Params, enc encoding.Encoder, dtx *apicontext.DataContext) string {
id, _ := strconv.Atoi(params["part"])
p := products.Part{
ID: id,
}
parts, err := p.GetRelated(dtx)
if err != nil {
apierror.GenerateError("Trouble getting related parts", err, w, r)
return ""
}
return encoding.Must(enc.Encode(parts))
}
//Redundant
func Packaging(w http.ResponseWriter, r *http.Request, params martini.Params, enc encoding.Encoder, dtx *apicontext.DataContext) string {
id, err := strconv.Atoi(params["part"])
if err != nil {
apierror.GenerateError("Trouble getting part ID", err, w, r)
return ""
}
p := products.Part{
ID: id,
}
if err = p.Get(dtx); err != nil {
apierror.GenerateError("Trouble getting part packages", err, w, r)
return ""
}
return encoding.Must(enc.Encode(p.Packages))
}
//Sort of Redundant
func InstallSheet(w http.ResponseWriter, r *http.Request, params martini.Params, dtx *apicontext.DataContext) {
id, err := strconv.Atoi(strings.Split(params["part"], ".")[0])
if err != nil {
apierror.GenerateError("Trouble getting part ID", err, w, r)
return
}
p := products.Part{
ID: id,
}
err = p.Get(dtx)
if err != nil {
apierror.GenerateError("Trouble getting part", err, w, r)
return
}
var text string
for _, content := range p.Content {
if content.ContentType.Type == "installationSheet" {
text = content.Text
}
}
if text == "" {
apierror.GenerateError("No Installation Sheet", err, w, r)
return
}
data, err := rest.GetPDF(text, r)
if err != nil {
apierror.GenerateError("Error getting PDF", err, w, r)
return
}
w.Header().Set("Content-Length", strconv.Itoa(len(data)))
w.Header().Set("Content-Type", "application/pdf")
w.Header().Set("Access-Control-Allow-Origin", "*")
w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
w.Header().Set("Access-Control-Allow-Headers", "Origin")
w.Write(data)
}
//Redundant
func ActiveApprovedReviews(w http.ResponseWriter, r *http.Request, params martini.Params, enc encoding.Encoder, dtx *apicontext.DataContext) string {
id, err := strconv.Atoi(params["part"])
if err != nil {
apierror.GenerateError("Trouble getting part ID", err, w, r)
return ""
}
p := products.Part{
ID: id,
}
if err = p.Get(dtx); err != nil {
apierror.GenerateError("Trouble getting part reviews", err, w, r)
return ""
}
var revs []products.Review
for _, rev := range p.Reviews {
if rev.Active == true && rev.Approved == true {
revs = append(revs, rev)
}
}
return encoding.Must(enc.Encode(revs))
}
func Prices(w http.ResponseWriter, r *http.Request, params martini.Params, enc encoding.Encoder, dtx *apicontext.DataContext) string {
id, err := strconv.Atoi(params["part"])
if err != nil {
apierror.GenerateError("Trouble getting part ID", err, w, r)
return ""
}
p := products.Part{
ID: id,
}
priceChan := make(chan int)
custChan := make(chan int)
go func() {
err = p.Get(dtx)
priceChan <- 1
}()
go func() {
price, custErr := customer.GetCustomerPrice(dtx, p.ID)
if custErr != nil {
err = custErr
}
p.Pricing = append(p.Pricing, products.Price{0, 0, "Customer", price, false, time.Now()})
custChan <- 1
}()
<-priceChan
<-custChan
if err != nil {
apierror.GenerateError("Trouble getting part prices", err, w, r)
return ""
}
return encoding.Must(enc.Encode(p.Pricing))
}
func PartNumber(rw http.ResponseWriter, r *http.Request, params martini.Params, enc encoding.Encoder, dtx *apicontext.DataContext) string {
var p products.Part
var err error
p.PartNumber = params["part"]
if p.PartNumber == "" {
apierror.GenerateError("Trouble getting old part number", err, rw, r)
return ""
}
if err = p.GetPartByPartNumber(); err != nil {
apierror.GenerateError("Trouble getting part by old part number", err, rw, r)
return ""
}
//TODO - remove when curt & aries vehicle application data are in sync
if p.Brand.ID == 3 {
mgoVehicles, err := vehicle.ReverseMongoLookup(p.ID)
if err != nil {
apierror.GenerateError("Trouble getting part by old part number", err, rw, r)
return ""
}
for _, v := range mgoVehicles {
vehicleApplication := products.VehicleApplication{
Year: v.Year,
Make: v.Make,
Model: v.Model,
Style: v.Style,
}
p.Vehicles = append(p.Vehicles, vehicleApplication)
}
} //END TODO
return encoding.Must(enc.Encode(p))
}
func TestParts(t *testing.T) {
var err error
var p products.Part
p.ID = 10999 //set part number here for use in creating related objects
var price products.Price
// var cu customer.CustomerUser
var cat products.Category
cat.Create()
//create install sheet content type
var contentType custcontent.ContentType
contentType.Type = "InstallationSheet"
err = contentType.Create()
//create install sheet content
var installSheetContent products.Content
installSheetContent.Text = "https://www.curtmfg.com/masterlibrary/13201/installsheet/CM_13201_INS.PDF"
installSheetContent.ContentType.Id = contentType.Id
err = installSheetContent.Create()
//create video type -- used in creating video during video test
var vt video.VideoType
vt.Name = "test type"
vt.Create()
//key types
var pub, pri, auth apiKeyType.ApiKeyType
if database.GetCleanDBFlag() != "" {
//setup apiKeyTypes
pub.Type = "public"
pri.Type = "private"
auth.Type = "authentication"
pub.Create()
pri.Create()
auth.Create()
}
dtx, err := apicontextmock.Mock()
if err != nil {
t.Log(err)
}
Convey("TestingParts", t, func() {
//test create part
p.Categories = append(p.Categories, cat)
p.OldPartNumber = "8675309"
p.ShortDesc = "test part"
p.Content = append(p.Content, installSheetContent)
bodyBytes, _ := json.Marshal(p)
bodyJson := bytes.NewReader(bodyBytes)
thyme := time.Now()
testThatHttp.RequestWithDtx("post", "/part", "", "?key="+dtx.APIKey, CreatePart, bodyJson, "application/json", dtx)
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &p)
So(err, ShouldBeNil)
So(p, ShouldHaveSameTypeAs, products.Part{})
So(p.ID, ShouldEqual, 10999)
p.BindCustomer(dtx)
var custPrice customer.Price
custPrice.CustID = dtx.CustomerID
custPrice.PartID = p.ID
err = custPrice.Create()
//test create price
price.Price = 987
price.PartId = p.ID
price.Type = "test"
bodyBytes, _ = json.Marshal(price)
bodyJson = bytes.NewReader(bodyBytes)
thyme = time.Now()
testThatHttp.RequestWithDtx("post", "/price", "", "?key="+dtx.APIKey, SavePrice, bodyJson, "application/json", dtx)
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &price)
So(err, ShouldBeNil)
So(price, ShouldHaveSameTypeAs, products.Price{})
So(price.Id, ShouldBeGreaterThan, 0)
//test update price
price.Type = "tester"
bodyBytes, _ = json.Marshal(price)
bodyJson = bytes.NewReader(bodyBytes)
thyme = time.Now()
testThatHttp.RequestWithDtx("post", "/price/", ":id", strconv.Itoa(price.Id)+"?key="+dtx.APIKey, SavePrice, bodyJson, "application/json", dtx)
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &price)
So(err, ShouldBeNil)
So(price, ShouldHaveSameTypeAs, products.Price{})
So(price.Type, ShouldNotEqual, "test")
// //test get part prices
thyme = time.Now()
testThatHttp.RequestWithDtx("get", "/part/", ":part/prices", strconv.Itoa(p.ID)+"/prices?key="+dtx.APIKey, Prices, nil, "", dtx)
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
var prices []products.Price
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &prices)
So(err, ShouldBeNil)
So(prices, ShouldHaveSameTypeAs, []products.Price{})
// //test get part categories
thyme = time.Now()
testThatHttp.Request("get", "/part/", ":part/categories", strconv.Itoa(p.ID)+"/categories?key="+dtx.APIKey, Categories, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
var cats []products.Category
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &cats)
So(err, ShouldBeNil)
So(cats, ShouldHaveSameTypeAs, []products.Category{})
//test get part install sheet
thyme = time.Now()
testThatHttp.Request("get", "/part/", ":part", strconv.Itoa(p.ID)+".pdf?key="+dtx.APIKey, InstallSheet, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()*6) //three seconds
So(testThatHttp.Response.Code, ShouldEqual, 200)
//test get videos
//create part video
var partVid video.Video
err = partVid.Create(dtx)
err = partVid.CreateJoinPart(p.ID)
thyme = time.Now()
testThatHttp.Request("get", "/part/videos/", ":part", strconv.Itoa(p.ID)+"?key="+dtx.APIKey, Videos, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
var ps []products.PartVideo
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &ps)
So(err, ShouldBeNil)
So(len(ps), ShouldBeGreaterThan, 0)
//get active approved reviews
//create active approved review
var review products.Review
review.PartID = p.ID
review.Rating = 1
review.Active = true
review.Approved = true
err = review.Create(dtx)
thyme = time.Now()
testThatHttp.Request("get", "/part/reviews/", ":part", strconv.Itoa(p.ID)+"?key="+dtx.APIKey, ActiveApprovedReviews, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
var reviews products.Reviews
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &reviews)
So(err, ShouldBeNil)
So(len(reviews), ShouldBeGreaterThan, 0)
review.Delete(dtx) //teardown - part has FK constraint on review.partID
//get packaging - no package created in test
thyme = time.Now()
testThatHttp.Request("get", "/part/packages/", ":part", strconv.Itoa(p.ID)+"?key="+dtx.APIKey, Packaging, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
var pack []products.Package
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &pack)
So(err, ShouldBeNil)
So(pack, ShouldHaveSameTypeAs, []products.Package{})
//get content
thyme = time.Now()
testThatHttp.Request("get", "/part/content/", ":part", strconv.Itoa(p.ID)+"?key="+dtx.APIKey, GetContent, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
var content products.Content
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &content)
So(err, ShouldBeNil)
So(content, ShouldHaveSameTypeAs, products.Content{})
//get attributes
thyme = time.Now()
testThatHttp.Request("get", "/part/attributes/", ":part", strconv.Itoa(p.ID)+"?key="+dtx.APIKey, Attributes, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
var attributes []products.Attribute
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &attributes)
So(err, ShouldBeNil)
So(attributes, ShouldHaveSameTypeAs, []products.Attribute{})
//test get images
thyme = time.Now()
testThatHttp.Request("get", "/part/images/", ":part", strconv.Itoa(p.ID)+"?key="+dtx.APIKey, Images, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
var images []products.Image
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &images)
So(err, ShouldBeNil)
So(images, ShouldHaveSameTypeAs, []products.Image{})
//test get vehicles
thyme = time.Now()
testThatHttp.Request("get", "/part/vehicles/", ":part", strconv.Itoa(p.ID)+"?key="+dtx.APIKey, Vehicles, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
var vs []products.Vehicle
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &vs)
So(err, ShouldBeNil)
So(vs, ShouldHaveSameTypeAs, []products.Vehicle{})
//test get related
thyme = time.Now()
testThatHttp.Request("get", "/part/related/", ":part", strconv.Itoa(p.ID)+"?key="+dtx.APIKey, GetRelated, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
var parts []products.Part
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &parts)
So(err, ShouldBeNil)
So(parts, ShouldHaveSameTypeAs, []products.Part{})
//test get
thyme = time.Now()
t.Log(strconv.Itoa(p.ID))
testThatHttp.Request("get", "/part/", ":part", strconv.Itoa(p.ID)+"?key="+dtx.APIKey, Get, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds())
So(testThatHttp.Response.Code, ShouldEqual, 200)
var part products.Part
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &part)
So(err, ShouldBeNil)
So(part, ShouldHaveSameTypeAs, products.Part{})
//test latest
thyme = time.Now()
testThatHttp.Request("get", "/part/latest", "", "?key="+dtx.APIKey, Latest, nil, "")
t.Log("Get latest parts benchmark: ", time.Since(thyme))
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &parts)
So(err, ShouldBeNil)
So(parts, ShouldHaveSameTypeAs, []products.Part{})
// test all
thyme = time.Now()
testThatHttp.Request("get", "/part/all", "", "?key="+dtx.APIKey, AllBasics, nil, "")
t.Log("Get all basic parts benchmark: ", time.Since(thyme))
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &parts)
So(err, ShouldBeNil)
So(parts, ShouldHaveSameTypeAs, []products.Part{})
//test featured
thyme = time.Now()
testThatHttp.Request("get", "/part/featured", "", "?key="+dtx.APIKey, Featured, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()*3) //3 seconds!
t.Log("Get featured parts benchmark: ", time.Since(thyme))
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &parts)
So(err, ShouldBeNil)
So(parts, ShouldHaveSameTypeAs, []products.Part{})
//test get all parts
thyme = time.Now()
testThatHttp.Request("get", "/part", "", "?key="+dtx.APIKey, All, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()*6) //6 seconds
t.Log("Get all parts benchmark: ", time.Since(thyme))
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &parts)
So(err, ShouldBeNil)
So(parts, ShouldHaveSameTypeAs, []products.Part{})
//test get price
thyme = time.Now()
testThatHttp.Request("get", "/price/", ":id", strconv.Itoa(price.Id)+"?key="+dtx.APIKey, GetPrice, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &price)
So(err, ShouldBeNil)
So(price, ShouldHaveSameTypeAs, products.Price{})
//test get old part Number
thyme = time.Now()
testThatHttp.Request("get", "/part/old/", ":part", p.OldPartNumber+"?key="+dtx.APIKey, OldPartNumber, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &p)
So(err, ShouldBeNil)
So(p, ShouldHaveSameTypeAs, products.Part{})
So(p.OldPartNumber, ShouldEqual, "8675309")
So(p.ID, ShouldEqual, 10999)
//test delete price
thyme = time.Now()
testThatHttp.Request("delete", "/price/", ":id", strconv.Itoa(price.Id)+"?key="+dtx.APIKey, DeletePrice, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &price)
So(err, ShouldBeNil)
So(price, ShouldHaveSameTypeAs, products.Price{})
//test update part
p.OldPartNumber = "8675309"
p.InstallSheet, err = url.Parse("www.sheetsrus.com")
bodyBytes, _ = json.Marshal(p)
bodyJson = bytes.NewReader(bodyBytes)
thyme = time.Now()
testThatHttp.Request("put", "/part/", ":id", strconv.Itoa(p.ID)+"?key="+dtx.APIKey, UpdatePart, bodyJson, "application/json")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds())
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &p)
So(err, ShouldBeNil)
So(p, ShouldHaveSameTypeAs, products.Part{})
So(p.OldPartNumber, ShouldEqual, "8675309")
So(p.ID, ShouldEqual, 10999)
//test delete part
thyme = time.Now()
testThatHttp.Request("delete", "/part/", ":id", strconv.Itoa(p.ID)+"?key="+dtx.APIKey, DeletePart, nil, "")
So(time.Since(thyme).Nanoseconds(), ShouldBeLessThan, time.Second.Nanoseconds()/2)
So(testThatHttp.Response.Code, ShouldEqual, 200)
err = json.Unmarshal(testThatHttp.Response.Body.Bytes(), &p)
So(err, ShouldBeNil)
So(p, ShouldHaveSameTypeAs, products.Part{})
// //teardown
custPrice.Delete()
partVid.Delete(dtx)
partVid.DeleteJoinPart(p.ID)
})
//teardown
p.Delete(dtx)
cat.Delete(dtx)
if database.GetCleanDBFlag() != "" {
pub.Delete()
pri.Delete()
auth.Delete()
}
contentType.Delete()
installSheetContent.Delete()
vt.Delete()
_ = apicontextmock.DeMock(dtx)
}