Fix wc_notice_count logic and add test

alexminza · Jan 10, 2018 · 97673e2 · 97673e2
1 parent d7718df
commit 97673e2
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/includes/wc-notice-functions.php b/includes/wc-notice-functions.php
@@ -38,7 +38,7 @@ function wc_notice_count( $notice_type = '' ) {
 	} elseif ( empty( $notice_type ) ) {
 
 		foreach ( $all_notices as $notices ) {
-			$notice_count += absint( sizeof( $all_notices ) );
+			$notice_count += count( $notices );
 		}
 	}
 

diff --git a/tests/unit-tests/util/notice-functions.php b/tests/unit-tests/util/notice-functions.php
@@ -35,10 +35,24 @@ function test_wc_notice_count() {
 		wc_add_notice( 'Bogus Error Notice', 'error' );
 		$this->assertEquals( 1, wc_notice_count( 'error' ) );
 
-		// multiple notices of different types
-		wc_add_notice( 'Bogus Notice 2', 'success' );
+		// multiple notices of different types.
+		wc_clear_notices();
+		wc_add_notice( 'Bogus 1', 'success' );
+		wc_add_notice( 'Bogus 2', 'success' );
+		wc_add_notice( 'Bogus Notice 1', 'notice' );
+		wc_add_notice( 'Bogus Notice 2', 'notice' );
+		wc_add_notice( 'Bogus Error Notice 1', 'error' );
+		wc_add_notice( 'Bogus Error Notice 2', 'error' );
+		$this->assertEquals( 6, wc_notice_count() );
+
+		// repeat with duplicates.
+		wc_add_notice( 'Bogus 1', 'success' );
+		wc_add_notice( 'Bogus 2', 'success' );
+		wc_add_notice( 'Bogus Notice 1', 'notice' );
+		wc_add_notice( 'Bogus Notice 2', 'notice' );
+		wc_add_notice( 'Bogus Error Notice 1', 'error' );
 		wc_add_notice( 'Bogus Error Notice 2', 'error' );
-		$this->assertEquals( 4, wc_notice_count() );
+		$this->assertEquals( 12, wc_notice_count() );
 	}
 
 	/**