woocommerce/tests/unit-tests/session/class-wc-tests-session-hand...

132 lines
4.3 KiB
PHP

<?php
/**
* Class WC_Tests_Session_Handler file.
*
* @package WooCommerce\Tests\Session
*/
/**
* Tests for the WC_Session_Handler class.
*/
class WC_Tests_Session_Handler extends WC_Unit_Test_Case {
public function setUp() {
parent::setUp();
$this->handler = new WC_Session_Handler();
$this->create_session();
}
public function test_save_data_should_insert_new_row() {
$current_session_data = $this->get_session_from_db( $this->session_key );
// delete session to make sure a new row is created in the DB.
$this->handler->delete_session( $this->session_key );
$this->assertFalse( wp_cache_get( $this->cache_prefix . $this->session_key, WC_SESSION_CACHE_GROUP ) );
$this->handler->set( 'cart', 'new cart' );
$this->handler->save_data();
$updated_session_data = $this->get_session_from_db( $this->session_key );
$this->assertEquals( $current_session_data->session_id + 1, $updated_session_data->session_id );
$this->assertEquals( $this->session_key, $updated_session_data->session_key );
$this->assertEquals( maybe_serialize( array( 'cart' => 'new cart' ) ), $updated_session_data->session_value );
$this->assertTrue( is_numeric( $updated_session_data->session_expiry ) );
$this->assertEquals( array( 'cart' => 'new cart' ), wp_cache_get( $this->cache_prefix . $this->session_key, WC_SESSION_CACHE_GROUP ) );
}
public function test_save_data_should_replace_existing_row() {
$current_session_data = $this->get_session_from_db( $this->session_key );
$this->handler->set( 'cart', 'new cart' );
$this->handler->save_data();
$updated_session_data = $this->get_session_from_db( $this->session_key );
$this->assertEquals( $current_session_data->session_id, $updated_session_data->session_id );
$this->assertEquals( $this->session_key, $updated_session_data->session_key );
$this->assertEquals( maybe_serialize( array( 'cart' => 'new cart' ) ), $updated_session_data->session_value );
$this->assertTrue( is_numeric( $updated_session_data->session_expiry ) );
}
public function test_get_session_should_use_cache() {
$session = $this->handler->get_session( $this->session_key );
$this->assertEquals( array( 'cart' => 'fake cart' ), $session );
}
public function test_get_session_should_not_use_cache() {
wp_cache_delete( $this->cache_prefix . $this->session_key, WC_SESSION_CACHE_GROUP );
$session = $this->handler->get_session( $this->session_key );
$this->assertEquals( array( 'cart' => 'fake cart' ), $session );
}
public function test_get_session_should_return_default_value() {
$default_session = array( 'session' => 'default' );
$session = $this->handler->get_session( 'non-existent key', $default_session );
$this->assertEquals( $default_session, $session );
}
public function test_delete_session() {
global $wpdb;
$this->handler->delete_session( $this->session_key );
$session_id = $wpdb->get_var(
$wpdb->prepare(
"SELECT `session_id` FROM {$wpdb->prefix}woocommerce_sessions WHERE session_key = %s",
$this->session_key
)
);
$this->assertFalse( wp_cache_get( $this->cache_prefix . $this->session_key, WC_SESSION_CACHE_GROUP ) );
$this->assertNull( $session_id );
}
public function test_update_session_timestamp() {
global $wpdb;
$timestamp = 1537970882;
$this->handler->update_session_timestamp( $this->session_key, $timestamp );
$session_expiry = $wpdb->get_var(
$wpdb->prepare(
"SELECT session_expiry FROM {$wpdb->prefix}woocommerce_sessions WHERE session_key = %s",
$this->session_key
)
);
$this->assertEquals( $timestamp, $session_expiry );
}
/**
* Helper function to create a WC session and save it to the DB.
*/
protected function create_session() {
$this->handler->init();
wp_set_current_user( 1 );
$this->handler->set( 'cart', 'fake cart' );
$this->handler->save_data();
$this->session_key = $this->handler->get_customer_id();
$this->cache_prefix = WC_Cache_Helper::get_cache_prefix( WC_SESSION_CACHE_GROUP );
}
/**
* Helper function to get session data from DB.
*
* @param string $session_key
* @return stdClass
*/
protected function get_session_from_db( $session_key ) {
global $wpdb;
$session_data = $wpdb->get_row(
$wpdb->prepare(
"SELECT * FROM {$wpdb->prefix}woocommerce_sessions WHERE `session_key` = %s",
$session_key
)
);
return $session_data;
}
}