Files
cloud-services/services/ota_update_go/handlers/sub_feature_delete_test.go

59 lines
1.8 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 TestSubscriptionFeatureDelete(t *testing.T) {
results := mocks.MockORMResults{
ReturnedRows: 1,
AffectedRows: 1,
}
mock := mocks.MockSubscriptionFeatures{
DBMockHelper: mocks.DBMockHelper{
ORMResponse: &results,
},
}
services.GetDB().SetSubFeatures(&mock)
tests := []mocks.DBHttpTest{
{
Name: "No id",
Request: th.MakeTestRequest(http.MethodDelete, "http://example.com/subscriptionfeature", nil),
ExpectedStatus: http.StatusBadRequest,
ExpectedResponse: `{"message":"primary key required","error":"Bad Request"}`,
},
{
Name: "Zero id",
Request: th.MakeTestRequest(http.MethodDelete, "http://example.com/subscriptionfeature?id=0", nil),
ExpectedStatus: http.StatusBadRequest,
ExpectedResponse: `{"message":"primary key required","error":"Bad Request"}`,
},
{
Name: "Good id",
Request: th.MakeTestRequest(http.MethodDelete, "http://example.com/subscriptionfeature?id=ecfb89e0-ca03-4aa9-a43a-a9d703256edb", nil),
ExpectedStatus: http.StatusOK,
ExpectedResponse: `{"message":"Deleted"}`,
},
{
Name: "DB error",
Request: th.MakeTestRequest(http.MethodDelete, "http://example.com/subscriptionfeature?id=0557bd1d-76d3-41e5-a44e-13c479e55ab0", nil),
ExpectedStatus: http.StatusServiceUnavailable,
ExpectedResponse: `{"message":"something went wrong","error":"Service Unavailable"}`,
DBTestCase: mocks.DBTestCase{
MockError: fmt.Errorf("something went wrong"),
},
},
}
mocks.RunDBTests(t, tests, handlers.HandleSubscriptionFeatureDelete, &mock)
}