feat: Added account balance to discount (#587)

* Added account balance to discount scraper * Added account balance to discount scraper Co-authored-by: Shai Cohen <[email protected]>
eshaham · Jul 10, 2021 · df01263 · df01263
1 parent a8343f5
commit df01263
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/scrapers/discount.ts b/src/scrapers/discount.ts
@@ -24,6 +24,10 @@ interface ScrapedTransaction {
   OperationDescriptionToDisplay: string;
 }
 
+interface CurrentAccountInfo {
+  AccountBalance: number;
+}
+
 interface ScrapedAccountData {
   UserAccountsData: {
     DefaultAccountNumber: string;
@@ -34,6 +38,7 @@ interface ScrapedTransactionData {
   Error?: { MsgText: string };
   CurrentAccountLastTransactions?: {
     OperationEntry: ScrapedTransaction[];
+    CurrentAccountInfo: CurrentAccountInfo;
   };
 }
 
@@ -99,6 +104,7 @@ async function fetchAccountData(page: Page, options: ScaperOptions): Promise<Sca
     success: true,
     accounts: [{
       accountNumber,
+      balance: txnsResult.CurrentAccountLastTransactions.CurrentAccountInfo.AccountBalance,
       txns: [...completedTxns, ...pendingTxns],
     }],
   };

diff --git a/src/tests/tests-utils.ts b/src/tests/tests-utils.ts
@@ -72,6 +72,7 @@ export function exportTransactions(fileName: string, accounts: TransactionsAccou
       ...account.txns.map((txn) => {
         return {
           account: account.accountNumber,
+          balance: account.balance,
           ...txn,
           date: moment(txn.date).format('DD/MM/YYYY'),
           processedDate: moment(txn.processedDate).format('DD/MM/YYYY'),