106 lines
3.3 KiB
Go
106 lines
3.3 KiB
Go
package handlers_test
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"otaupdate/handlers"
|
|
"otaupdate/services"
|
|
|
|
"github.com/fiskerinc/cloud-services/pkg/common"
|
|
"github.com/fiskerinc/cloud-services/pkg/db/queries/mocks"
|
|
"github.com/fiskerinc/cloud-services/pkg/httpclient/tester"
|
|
th "github.com/fiskerinc/cloud-services/pkg/testhelper"
|
|
"github.com/fiskerinc/cloud-services/pkg/testrunner"
|
|
)
|
|
|
|
const err404NotFound = "404 page not found\n"
|
|
|
|
func TestHandleSupplierUpdate(t *testing.T) {
|
|
route := "/supplier/:email"
|
|
mock := mocks.MockSupplierAccounts{}
|
|
reqGoodData := common.SupplierAccount{
|
|
SupplierOrganizationID: 14,
|
|
Company: "TEST",
|
|
Address: "100 Main",
|
|
Contact: "John Doe",
|
|
Telephone: "+1-555-555-5555",
|
|
Program: "Ocean",
|
|
ECUs: []string{"ADAS"},
|
|
}
|
|
|
|
tests := []testrunner.TestCase{
|
|
{
|
|
Name: "No data",
|
|
HttpTestCase: &tester.HttpTestCase{
|
|
Request: th.MakeTestRequest(http.MethodPut, "http://example.com/supplier", nil),
|
|
ExpectedStatus: http.StatusNotFound,
|
|
ExpectedResponse: err404NotFound,
|
|
},
|
|
},
|
|
{
|
|
Name: "Bad Data",
|
|
HttpTestCase: &tester.HttpTestCase{
|
|
Request: th.MakeTestRequest(http.MethodPut, "http://example.com/supplier/test", common.SupplierAccount{
|
|
Company: "TEST",
|
|
Address: "100 Main",
|
|
Contact: "John Doe",
|
|
Telephone: "+1-555-555-5555",
|
|
Program: "Ocean",
|
|
ECUs: []string{},
|
|
}),
|
|
ExpectedStatus: http.StatusBadRequest,
|
|
ExpectedResponse: `{"message":"primary key required","error":"Bad Request"}`,
|
|
},
|
|
},
|
|
{
|
|
Name: "Good data",
|
|
HttpTestCase: &tester.HttpTestCase{
|
|
Request: th.MakeTestRequest(http.MethodPut, "http://example.com/supplier/test@supplier.com", reqGoodData),
|
|
ExpectedStatus: http.StatusOK,
|
|
ExpectedResponse: `{"supplier_organization_id":"14","email":"test@supplier.com","company":"TEST","address":"100 Main","contact":"John Doe","telephone":"+1-555-555-5555","program":"Ocean","ecus":["ADAS"]}`,
|
|
},
|
|
},
|
|
{
|
|
Name: "Error unregistered email",
|
|
HttpTestCase: &tester.HttpTestCase{
|
|
Request: th.MakeTestRequest(http.MethodPut, "http://example.com/supplier/test@supplier.com", reqGoodData),
|
|
ExpectedStatus: http.StatusServiceUnavailable,
|
|
ExpectedResponse: `{"message":"account does not exist","error":"Service Unavailable"}`,
|
|
Setup: func() {
|
|
mock.ORMResponse = &mocks.MockORMResults{AffectedRows: 0}
|
|
services.GetDB().SetSupplierAccount(&mock)
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "Error",
|
|
HttpTestCase: &tester.HttpTestCase{Request: th.MakeTestRequest(http.MethodPut, "http://example.com/supplier/test@supplier.com", reqGoodData),
|
|
ExpectedStatus: http.StatusServiceUnavailable,
|
|
ExpectedResponse: `{"message":"something went wrong","error":"Service Unavailable"}`,
|
|
},
|
|
DBTestCase: &mocks.DBTestCase{
|
|
MockError: fmt.Errorf("something went wrong"),
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
if test.DBTestCase != nil {
|
|
test.DBTestCase.SetupDB(&mock)
|
|
}
|
|
|
|
mock.ORMResponse = &mocks.MockORMResults{AffectedRows: 1}
|
|
services.GetDB().SetSupplierAccount(&mock)
|
|
|
|
if test.HttpTestCase != nil && test.HttpTestCase.Setup != nil {
|
|
test.HttpTestCase.Setup()
|
|
}
|
|
|
|
w := test.HttpTestCase.TestWithParamPath(handlers.HandleSupplierUpdate, route)
|
|
|
|
test.HttpTestCase.ValidateHttp(t, test.Name, w)
|
|
}
|
|
}
|