From 74039a8011838a1d14680dc14150d56c7a277693 Mon Sep 17 00:00:00 2001
From: Korina Cordero <cordero.korina@gmail.com>
Date: Wed, 5 Aug 2020 06:48:17 +0000
Subject: [PATCH] Set service types to battery sales, warranty replacement,
 jumpstart, and warranty claim. #458

---
 src/Ramcar/CMBServiceType.php                        | 4 +++-
 src/Service/InvoiceGenerator/CMBInvoiceGenerator.php | 4 ++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/Ramcar/CMBServiceType.php b/src/Ramcar/CMBServiceType.php
index 63d18f96..607b9c43 100644
--- a/src/Ramcar/CMBServiceType.php
+++ b/src/Ramcar/CMBServiceType.php
@@ -6,11 +6,13 @@ class CMBServiceType extends NameValue
 {
     const BATTERY_REPLACEMENT_NEW       = 'battery_new';
     const BATTERY_REPLACEMENT_WARRANTY  = 'battery_warranty';
+    const WARRANTY_CLAIM                = 'warranty_claim';
     const JUMPSTART                     = 'jumpstart';
 
     const COLLECTION = [
         'battery_new' => 'Battery Sales',
-        'battery_warranty' => 'Under Warranty',
+        'battery_warranty' => 'Warranty Replacement',
+        'warranty_claim' => 'Warranty Claim',
         'jumpstart' => 'Jumpstart',
     ];
 }
diff --git a/src/Service/InvoiceGenerator/CMBInvoiceGenerator.php b/src/Service/InvoiceGenerator/CMBInvoiceGenerator.php
index 69b4e55a..f2e75a78 100644
--- a/src/Service/InvoiceGenerator/CMBInvoiceGenerator.php
+++ b/src/Service/InvoiceGenerator/CMBInvoiceGenerator.php
@@ -86,6 +86,10 @@ class CMBInvoiceGenerator implements InvoiceGeneratorInterface
             case CMBServiceType::BATTERY_REPLACEMENT_WARRANTY:
                 $this->processWarranty($total, $criteria, $invoice);
                 break;
+            case CMBServiceType::WARRANTY_CLAIM:
+                // TODO: this will change once we confirm what needs to be computed
+                $this->processOtherServices($total, $invoice, $stype);
+                break;
             //case ServiceType::POST_RECHARGED:
             //    $this->processRecharge($total, $invoice);
             //    break;