110 lines
3.3 KiB
Go
110 lines
3.3 KiB
Go
package handlers_test
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"otaupdate/handlers"
|
|
"otaupdate/services"
|
|
|
|
"github.com/fiskerinc/cloud-services/pkg/db/queries/mocks"
|
|
th "github.com/fiskerinc/cloud-services/pkg/testhelper"
|
|
)
|
|
|
|
func TestSubscriptionDelete(t *testing.T) {
|
|
results := mocks.MockORMResults{
|
|
ReturnedRows: 0,
|
|
AffectedRows: 1,
|
|
}
|
|
mock := mocks.MockSubscriptions{
|
|
DBMockHelper: mocks.DBMockHelper{
|
|
ORMResponse: &results,
|
|
},
|
|
}
|
|
services.GetDB().SetSubscriptions(&mock)
|
|
|
|
tests := []mocks.DBHttpTest{
|
|
{
|
|
Name: "No id",
|
|
Request: th.MakeTestRequest(http.MethodDelete, "http://example.com/subscription", nil),
|
|
ExpectedStatus: http.StatusBadRequest,
|
|
ExpectedResponse: `{"message":"primary key required","error":"Bad Request"}`,
|
|
DBTestCase: mocks.DBTestCase{
|
|
SetupMockResponse: func() {
|
|
results.AffectedRows = 1
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "Zero id",
|
|
Request: th.MakeTestRequest(http.MethodDelete, "http://example.com/subscription?id=0", nil),
|
|
ExpectedStatus: http.StatusBadRequest,
|
|
ExpectedResponse: `{"message":"primary key required","error":"Bad Request"}`,
|
|
DBTestCase: mocks.DBTestCase{
|
|
SetupMockResponse: func() {
|
|
results.AffectedRows = 1
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "Good id",
|
|
Request: th.MakeTestRequest(http.MethodDelete, "http://example.com/subscription?id=100", nil),
|
|
ExpectedStatus: http.StatusOK,
|
|
ExpectedResponse: `{"message":"Deleted"}`,
|
|
DBTestCase: mocks.DBTestCase{
|
|
SetupMockResponse: func() {
|
|
results.AffectedRows = 1
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "Good vin and sub name",
|
|
Request: th.MakeTestRequest(http.MethodDelete, "http://example.com/subscription?vin=1G1FP87S3GN100062&name=TEST", nil),
|
|
ExpectedStatus: http.StatusOK,
|
|
ExpectedResponse: `{"message":"Deleted"}`,
|
|
DBTestCase: mocks.DBTestCase{
|
|
SetupMockResponse: func() {
|
|
results.AffectedRows = 1
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "Bad vin",
|
|
Request: th.MakeTestRequest(http.MethodDelete, "http://example.com/subscription?vin=1G1FP87S3GN100&name=TEST", nil),
|
|
ExpectedStatus: http.StatusBadRequest,
|
|
ExpectedResponse: `{"message":"primary key required","error":"Bad Request"}`,
|
|
DBTestCase: mocks.DBTestCase{
|
|
SetupMockResponse: func() {
|
|
results.AffectedRows = 1
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "Record not found",
|
|
Request: th.MakeTestRequest(http.MethodDelete, "http://example.com/subscription?vin=1G1FP87S3GN100062&name=TEST", nil),
|
|
ExpectedStatus: http.StatusNotFound,
|
|
ExpectedResponse: `{"message":"Nothing deleted","error":"Not Found"}`,
|
|
DBTestCase: mocks.DBTestCase{
|
|
SetupMockResponse: func() {
|
|
results.AffectedRows = 0
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "DB error",
|
|
Request: th.MakeTestRequest(http.MethodDelete, "http://example.com/subscriptionpackage?id=100", nil),
|
|
ExpectedStatus: http.StatusServiceUnavailable,
|
|
ExpectedResponse: `{"message":"something went wrong","error":"Service Unavailable"}`,
|
|
DBTestCase: mocks.DBTestCase{
|
|
MockError: fmt.Errorf("something went wrong"),
|
|
SetupMockResponse: func() {
|
|
results.AffectedRows = 1
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
mocks.RunDBTests(t, tests, handlers.HandleSubscriptionDelete, &mock)
|
|
}
|