Add getBalance() method (#2)

* Add getBalance method Minor fixes
IntegReady · Feb 17, 2020 · e4ec8e4 · e4ec8e4
1 parent e0f62ec
commit e4ec8e4
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/src/api/Envelope.php b/src/api/Envelope.php
@@ -420,15 +420,21 @@ public function getShowPrice()
      * @return bool
      */
     public function checkBalance()
+    {
+        return $this->getBalance() >= 3;
+    }
+
+    /**
+     * @return float
+     */
+    public function getBalance()
     {
         $accountInfoResponse = $this->soapClient->GetAccountInfo([
             'Authenticator' => $this->getAuthToken(),
         ]);
         $this->setAuthToken($accountInfoResponse->Authenticator);
 
-        $availableBalance = (double)$accountInfoResponse->AccountInfo->PostageBalance->AvailablePostage;
-
-        return $availableBalance >= 3;
+        return (double)$accountInfoResponse->AccountInfo->PostageBalance->AvailablePostage;
     }
 
     /**

diff --git a/src/api/EnvelopeInterface.php b/src/api/EnvelopeInterface.php
@@ -150,4 +150,14 @@ public function setShowPrice($flag);
      * @return bool
      */
     public function getShowPrice();
+
+    /**
+     * @return bool
+     */
+    public function checkBalance();
+
+    /**
+     * @return float
+     */
+    public function getBalance();
 }