Add optional incremental to recordUsage

dukenst2006 · Aug 23, 2016 · fe08cd8 · fe08cd8
1 parent 5c26e8f
commit fe08cd8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/LaraPlans/Models/PlanSubscription.php b/src/LaraPlans/Models/PlanSubscription.php
@@ -286,7 +286,7 @@ public function featureEnabled($feature_code)
      * @param int $uses
      * @return \Gerardojbaez\LaraPlans\Models\PlanSubscriptionUsage
      */
-    public function recordUsage($feature_code, $uses = 1)
+    public function recordUsage($feature_code, $uses = 1, $incremental = true)
     {
         $feature = new Feature($feature_code);
 
@@ -310,7 +310,7 @@ public function recordUsage($feature_code, $uses = 1)
             }
         }
 
-        $usage->used += $uses;
+        $usage->used = ($incremental ? $usage->used + $uses : $uses);
 
         $usage->save();
 

diff --git a/tests/integration/Models/PlanSubscriptionTest.php b/tests/integration/Models/PlanSubscriptionTest.php
@@ -153,6 +153,12 @@ public function it_can_record_feature_usage()
         $usage = $this->subscription->recordUsage('listings_per_month');
 
         $this->assertInstanceOf(PlanSubscriptionUsage::class, $usage);
+        $this->assertEquals(1, $usage->used);
+
+        // Record fixed amount (not incremental)
+        $usage = $this->subscription->recordUsage('listings_per_month', 2, false);
+        $this->assertInstanceOf(PlanSubscriptionUsage::class, $usage);
+        $this->assertEquals(2, $usage->used);
     }
 
     /**