Fix single Invoice (not invoice) and other finalisable items still sh…

…owing badge count Addresses #507
msupply-foundation · Sep 9, 2019 · 269b556 · 269b556
1 parent fec67a5
commit 269b556
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/widgets/NavigationBar.js b/src/widgets/NavigationBar.js
@@ -19,7 +19,7 @@ export class NavigationBar extends React.Component {
 
   componentWillReceiveProps(props) {
     const dataType = this.getDataTypeFromRouteName(props);
-    if (dataType) this.refreshData(dataType);
+    this.refreshData(dataType);
   }
 
   getDataTypeFromRouteName(props) {
@@ -40,9 +40,12 @@ export class NavigationBar extends React.Component {
   }
 
   refreshData = (dataType) => {
+    const unfinalisedCount = dataType != '' ? 
+      this.props.database.objects(dataType).filtered('status != "finalised"').length :
+      0;
+
     this.setState({
-      unfinalisedCount: this.props.database
-        .objects(dataType).filtered('status != "finalised"').length,
+      unfinalisedCount: unfinalisedCount,
     });
   }