woocommerce/tests/unit-tests/log/log-handler-db.php

<?php

/**
 * Class WC_Tests_Log_Handler_DB
 * @package WooCommerce\Tests\Log
 * @since 3.0.0
 */
class WC_Tests_Log_Handler_DB extends WC_Unit_Test_Case {
	public function setUp() {
		parent::setUp();
		WC_Log_Handler_DB::flush();
	}

	public function tearDown() {
		WC_Log_Handler_DB::flush();
		parent::tearDown();
	}

	/**
	 * Test handle writes to database correctly.
	 *
	 * @since 3.0.0
	 */
	public function test_handle() {
		global $wpdb;

		$handler = new WC_Log_Handler_DB( array( 'threshold' => 'debug' ) );
		$time = time();
		$context = array( 1, 2, 'a', 'b', 'key' => 'value' );

		$handler->handle( $time, 'debug', 'msg_debug', array( 'source' => 'source_debug' ) );
		$handler->handle( $time, 'info', 'msg_info', array( 'source' => 'source_info' ) );
		$handler->handle( $time, 'notice', 'msg_notice', array( 'source' => 'source_notice' ) );
		$handler->handle( $time, 'warning', 'msg_warning', array( 'source' => 'source_warning' ) );
		$handler->handle( $time, 'error', 'msg_error', array( 'source' => 'source_error' ) );
		$handler->handle( $time, 'critical', 'msg_critical', array( 'source' => 'source_critical' ) );
		$handler->handle( $time, 'alert', 'msg_alert', array( 'source' => 'source_alert' ) );
		$handler->handle( $time, 'emergency', 'msg_emergency', array( 'source' => 'source_emergency' ) );

		$handler->handle( $time, 'debug', 'context_test', $context );

		$log_entries = $wpdb->get_results( "SELECT timestamp, level, message, source, context FROM {$wpdb->prefix}woocommerce_log", ARRAY_A );

		$expected_ts = date( 'Y-m-d H:i:s', $time );
		$expected = array(
			array(
				'timestamp' => $expected_ts,
				'level' => WC_Log_Levels::get_level_severity( 'debug' ),
				'message' => 'msg_debug',
				'source' => 'source_debug',
				'context' => serialize( array( 'source' => 'source_debug' ) ),
			),
			array(
				'timestamp' => $expected_ts,
				'level' => WC_Log_Levels::get_level_severity( 'info' ),
				'message' => 'msg_info',
				'source' => 'source_info',
				'context' => serialize( array( 'source' => 'source_info' ) ),
			),
			array(
				'timestamp' => $expected_ts,
				'level' => WC_Log_Levels::get_level_severity( 'notice' ),
				'message' => 'msg_notice',
				'source' => 'source_notice',
				'context' => serialize( array( 'source' => 'source_notice' ) ),
			),
			array(
				'timestamp' => $expected_ts,
				'level' => WC_Log_Levels::get_level_severity( 'warning' ),
				'message' => 'msg_warning',
				'source' => 'source_warning',
				'context' => serialize( array( 'source' => 'source_warning' ) ),
			),
			array(
				'timestamp' => $expected_ts,
				'level' => WC_Log_Levels::get_level_severity( 'error' ),
				'message' => 'msg_error',
				'source' => 'source_error',
				'context' => serialize( array( 'source' => 'source_error' ) ),
			),
			array(
				'timestamp' => $expected_ts,
				'level' => WC_Log_Levels::get_level_severity( 'critical' ),
				'message' => 'msg_critical',
				'source' => 'source_critical',
				'context' => serialize( array( 'source' => 'source_critical' ) ),
			),
			array(
				'timestamp' => $expected_ts,
				'level' => WC_Log_Levels::get_level_severity( 'alert' ),
				'message' => 'msg_alert',
				'source' => 'source_alert',
				'context' => serialize( array( 'source' => 'source_alert' ) ),
			),
			array(
				'timestamp' => $expected_ts,
				'level' => WC_Log_Levels::get_level_severity( 'emergency' ),
				'message' => 'msg_emergency',
				'source' => 'source_emergency',
				'context' => serialize( array( 'source' => 'source_emergency' ) ),
			),
			array(
				'timestamp' => $expected_ts,
				'level' => WC_Log_Levels::get_level_severity( 'debug' ),
				'message' => 'context_test',
				'source' => pathinfo( __FILE__, PATHINFO_FILENAME ),
				'context' => serialize( $context ),
			),
		);

		$this->assertEquals( $log_entries, $expected );
	}


	/**
	 * Test flush.
	 *
	 * @since 3.0.0
	 */
	public function test_flush() {
		global $wpdb;

		$handler = new WC_Log_Handler_DB( array( 'threshold' => 'debug' ) );
		$time = time();

		$handler->handle( $time, 'debug', '', array() );

		$log_entries = $wpdb->get_results( "SELECT timestamp, level, message, source FROM {$wpdb->prefix}woocommerce_log" );
		$this->assertCount( 1, $log_entries );

		WC_Log_Handler_DB::flush();

		$log_entries = $wpdb->get_results( "SELECT timestamp, level, message, source FROM {$wpdb->prefix}woocommerce_log" );
		$this->assertCount( 0, $log_entries );
	}

}