Gecko [ fernandoquevedodop.com ]

Name	Size	Permission	Action
CSVTest.php	2.13 KB	-rw-r--r--

Code Editor : CSVTest.php

<?php

namespace WP_CLI\Tests\CSV;

use WP_CLI\Tests\TestCase;
use WP_CLI\Iterators\CSV;

class CSVTest extends TestCase {

public function test_it_can_iterate_over_a_csv_file() {
		$filename = $this->create_csv_file(
			array(
				array( 'foo', 'bar' ),
				array( 'baz', 'qux' ),
			)
		);

$expected = array(
			0 => array(
				'foo' => 'baz',
				'bar' => 'qux',
			),
		);

foreach ( new CSV( $filename ) as $index => $row ) {
			$this->assertEquals( $expected[ $index ], $row );
		}
	}

public function test_it_can_iterate_over_a_csv_file_with_custom_delimiter() {
		$filename = $this->create_csv_file(
			array(
				array( 'foo|bar' ),
				array( 'baz|qux' ),
			),
			'|'
		);

$expected = array(
			0 => array(
				'foo|bar' => 'baz|qux',
			),
		);

foreach ( new CSV( $filename, '|' ) as $index => $row ) {
			$this->assertEquals( $expected[ $index ], $row );
		}
	}

public function test_it_can_iterate_over_a_csv_file_with_multiple_lines_in_a_value() {
		$filename = $this->create_csv_file(
			array(
				array( 'foo', "bar\nbaz" ),
				array( 'qux', "quux\nquuz" ),
			)
		);

$expected = array(
			0 => array(
				'foo'      => 'qux',
				"bar\nbaz" => "quux\nquuz",
			),
		);

foreach ( new CSV( $filename ) as $index => $row ) {
			$this->assertEquals( $expected[ $index ], $row );
		}
	}

public function test_it_can_iterate_over_a_csv_file_with_multiple_lines_and_comma_in_a_value() {
		$filename = $this->create_csv_file(
			array(
				array( 'foo', "bar\nbaz,qux" ),
				array( 'quux', "quuz\ncorge,grault" ),
			)
		);

$expected = array(
			0 => array(
				'foo'          => 'quux',
				"bar\nbaz,qux" => "quuz\ncorge,grault",
			),
		);

foreach ( new CSV( $filename ) as $index => $row ) {
			$this->assertEquals( $expected[ $index ], $row );
		}
	}

private function create_csv_file( $data, $delimiter = ',' ) {
		$filename = tempnam( sys_get_temp_dir(), 'wp-cli-tests-' );

$fp = fopen( $filename, 'wb' );

foreach ( $data as $row ) {
			fputcsv( $fp, $row, $delimiter, '"', '\\' );
		}

fclose( $fp );

register_shutdown_function(
			function () use ( $filename ) {
				unlink( $filename );
			}
		);

return $filename;
	}
}

${this.title}

Code Editor : CSVTest.php