More fixes

Author: swissspidy

features/db-import.feature

@@ -144,7 +144,7 @@ Feature: Import a WordPress database
     When I try `wp db import --no-defaults --debug`
     Then STDERR should match #Debug \(db\): Running shell command: /usr/bin/env (mysql|mariadb) --no-defaults --no-auto-rehash#
 
-  @require-wp-4.2
+  @require-wp-4.2 @require-mysql-or-mariadb
   Scenario: Import db that has emoji in post
     Given a WP install
 
@@ -180,3 +180,32 @@ Feature: Import a WordPress database
       """
       🍣
       """
+
+  @require-wp-4.2 @require-sqlite
+  Scenario: Import db that has emoji in post
+    Given a WP install
+
+    When I run `wp post create --post_title="🍣"`
+    And I run `wp post list`
+    Then the return code should be 0
+    And STDOUT should contain:
+      """
+      🍣
+      """
+
+    When I try `wp db export wp_cli_test.sql --debug`
+    Then the return code should be 0
+    And the wp_cli_test.sql file should exist
+
+    When I run `wp db import --dbuser=wp_cli_test --dbpass=password1`
+    Then STDOUT should be:
+      """
+      Success: Imported from 'wp_cli_test.sql'.
+      """
+
+    When I run `wp post list`
+    Then the return code should be 0
+    And STDOUT should contain:
+      """
+      🍣
+      """

features/db-size.feature

@@ -187,6 +187,7 @@ Feature: Display database size
       MB
       """
 
+  @require-mysql-or-mariadb
   Scenario: Display database size in bytes with specific format for a WordPress install
     Given a WP install
 
@@ -207,6 +208,27 @@ Feature: Display database size
 
     But STDOUT should not be a number
 
+  @require-sqlite
+  Scenario: Display database size in bytes with specific format for a WordPress install
+    Given a WP install
+
+    When I run `wp db size --size_format=b --format=csv`
+    Then STDOUT should contain:
+      """
+      Name,Size
+      .ht.sqlite,"
+      """
+
+    But STDOUT should not be a number
+
+    When I run `wp db size --size_format=b --format=json`
+    Then STDOUT should contain:
+      """
+      [{"Name":".ht.sqlite","Size":"
+      """
+
+    But STDOUT should not be a number
+
   Scenario: Display all table sizes for a WordPress install
     Given a WP install
 

src/DB_Command.php

@@ -1244,6 +1244,10 @@ public function size( $args, $assoc_args ) {
 					$size_key = floor( log( (float) $row['Size'] ) / log( 1000 ) );
 					$sizes    = [ 'B', 'KB', 'MB', 'GB', 'TB' ];
 
+					if ( is_infinite( $size_key ) ) {
+						$size_key = 0;
+					}
+
 					$size_format = isset( $sizes[ $size_key ] ) ? $sizes[ $size_key ] : $sizes[0];
 				}
 

src/DB_Command_SQLite.php

@@ -323,7 +323,7 @@ protected function sqlite_export( $file, $assoc_args ) {
 		}
 
 		$exclude_tables = Utils\get_flag_value( $assoc_args, 'exclude_tables', '' );
-		$exclude_tables = explode( ',', trim( $assoc_args['exclude_tables'], ',' ) );
+		$exclude_tables = explode( ',', trim( $exclude_tables, ',' ) );
 		$exclude_tables = array_map( 'strtolower', $exclude_tables );
 
 		try {