Fixed errors from Behat. Modified the default unit on test.

Author: marksabbath

features/db-size.feature

@@ -13,7 +13,7 @@ Feature: Display database size
 
     And STDOUT should contain:
       """
-      KB
+      B
       """
 
   Scenario: Display only table sizes for a WordPress install
@@ -22,7 +22,7 @@ Feature: Display database size
     When I run `wp db size --tables`
     Then STDOUT should contain:
       """
-      wp_posts	80 KB
+      wp_posts	81920 B
       """
 
     But STDOUT should not contain:

src/DB_Command.php

@@ -596,6 +596,8 @@ public function size( $args, $assoc_args ) {
 		$rows = array();
 		$fields = array( 'Name', 'Size' );
 
+		$default_unit = ( empty( $size_format ) ) ? ' B' : '';
+
 		if ( $tables ) {
 
 			// Add all of the table sizes
@@ -612,7 +614,7 @@ public function size( $args, $assoc_args ) {
 				// Add the table size to the list.
 				$rows[] = array(
 					'Name'  => $table_name,
-					'Size'  => strtoupper( $table_bytes ) . " " . 'B',
+					'Size'  => strtoupper( $table_bytes ) . $default_unit,
 				);
 			}
 		} else {
@@ -627,7 +629,7 @@ public function size( $args, $assoc_args ) {
 			// Add the database size to the list.
 			$rows[] = array(
 				'Name'  => DB_NAME,
-				'Size'  => strtoupper( $db_bytes ) . " " . 'B',
+				'Size'  => strtoupper( $db_bytes ) . $default_unit,
 				);
 		}
 
@@ -657,17 +659,21 @@ public function size( $args, $assoc_args ) {
 							break;
 					}
 
-					$rows[ $index ]['Size'] = $row['Size'] / $divisor . " " . ucfirst( $size_format );
+					$rows[ $index ]['Size'] = ceil( $row['Size'] / $divisor ) . " " . strtoupper( $size_format );
 			}
 		}
 
-		// Display the rows.
-		$args = array(
-			'format' => $format,
-		);
+		if ( ! empty( $size_format) && ! $tables ) {
+			WP_CLI::Line( filter_var( $rows[0]['Size'], FILTER_SANITIZE_NUMBER_INT ) );
+		} else {
+			// Display the rows.
+			$args = array(
+				'format' => $format,
+			);
 
-		$formatter = new \WP_CLI\Formatter( $args, $fields );
-		$formatter->display_items( $rows );
+			$formatter = new \WP_CLI\Formatter( $args, $fields );
+			$formatter->display_items( $rows );
+		}
 	}
 
 	/**