1 files changed, 30 insertions, 2 deletions
diff --git a/src/components/protocol_handler/test/service_status_update_handler_test.cc b/src/components/protocol_handler/test/service_status_update_handler_test.cc
index b587194d98..8da89bdcd1 100644
--- a/src/components/protocol_handler/test/service_status_update_handler_test.cc
+++ b/src/components/protocol_handler/test/service_status_update_handler_test.cc
@@ -73,7 +73,8 @@ class ServiceStatusUpdateHandlerTest
 
   Common_ServiceEvent::eType GetServiceEvent(ServiceStatus status) {
     switch (status) {
-      case ServiceStatus::SERVICE_ACCEPTED: {
+      case ServiceStatus::SERVICE_ACCEPTED:
+      case ServiceStatus::PROTECTION_DISABLED: {
         return Common_ServiceEvent::REQUEST_ACCEPTED;
       }
       case ServiceStatus::SERVICE_RECEIVED: {
@@ -82,7 +83,8 @@ class ServiceStatusUpdateHandlerTest
       case ServiceStatus::SERVICE_START_FAILED:
       case ServiceStatus::PTU_FAILED:
       case ServiceStatus::CERT_INVALID:
-      case ServiceStatus::INVALID_TIME: {
+      case ServiceStatus::INVALID_TIME:
+      case ServiceStatus::PROTECTION_ENFORCED: {
         return Common_ServiceEvent::REQUEST_REJECTED;
       }
       default: { return Common_ServiceEvent::INVALID_ENUM; }
@@ -108,6 +110,14 @@ class ServiceStatusUpdateHandlerTest
         auto reason = Common_ServiceStatusUpdateReason::INVALID_TIME;
         return reason;
       }
+      case ServiceStatus::PROTECTION_ENFORCED: {
+        auto reason = Common_ServiceStatusUpdateReason::PROTECTION_ENFORCED;
+        return reason;
+      }
+      case ServiceStatus::PROTECTION_DISABLED: {
+        auto reason = Common_ServiceStatusUpdateReason::PROTECTION_DISABLED;
+        return reason;
+      }
       default: {
         auto reason = Common_ServiceStatusUpdateReason::INVALID_ENUM;
         return reason;
@@ -170,6 +180,24 @@ INSTANTIATE_TEST_CASE_P(
         ServiceUpdate(ServiceType::kMobileNav, ServiceStatus::PTU_FAILED),
         ServiceUpdate(ServiceType::kRpc, ServiceStatus::PTU_FAILED)));
 
+INSTANTIATE_TEST_CASE_P(
+    OnServiceUpdate_PROTECTION_ENFRORCED,
+    ServiceStatusUpdateHandlerTest,
+    ::testing::Values(
+        ServiceUpdate(ServiceType::kAudio, ServiceStatus::PROTECTION_ENFORCED),
+        ServiceUpdate(ServiceType::kMobileNav,
+                      ServiceStatus::PROTECTION_ENFORCED),
+        ServiceUpdate(ServiceType::kRpc, ServiceStatus::PROTECTION_ENFORCED)));
+
+INSTANTIATE_TEST_CASE_P(
+    OnServiceUpdate_PROTECTION_DISABLED,
+    ServiceStatusUpdateHandlerTest,
+    ::testing::Values(
+        ServiceUpdate(ServiceType::kAudio, ServiceStatus::PROTECTION_DISABLED),
+        ServiceUpdate(ServiceType::kMobileNav,
+                      ServiceStatus::PROTECTION_DISABLED),
+        ServiceUpdate(ServiceType::kRpc, ServiceStatus::PROTECTION_DISABLED)));
+
 TEST_P(ServiceStatusUpdateHandlerTest, OnServiceUpdate) {
   auto service_event_ = GetServiceEvent(GetParam().service_status_);
   auto reason_ = GetUpdateReason(GetParam().service_status_);