1 files changed, 304 insertions, 0 deletions
diff --git a/tests/codeigniter/libraries/Session_test.php b/tests/codeigniter/libraries/Session_test.php
new file mode 100644
index 000000000..db7efb6b7
--- /dev/null
+++ b/tests/codeigniter/libraries/Session_test.php
@@ -0,0 +1,304 @@
+<?php
+
+/**
+ * Session driver library unit test
+ */
+class Session_test extends CI_TestCase {
+	protected $settings = array('use_cookies' => 0, 'use_only_cookies' => 0, 'cache_limiter' => false);
+	protected $setting_vals = array();
+	protected $cookie_vals;
+	protected $session;
+
+	/**
+	 * Set up test framework
+	 */
+	public function set_up()
+	{
+		// Override settings
+		foreach ($this->settings as $name => $value) {
+			$this->setting_vals[$name] = ini_get('session.'.$name);
+			ini_set('session.'.$name, $value);
+		}
+
+		// Start with clean environment
+		$this->cookie_vals = $_COOKIE;
+		$_COOKIE = array();
+
+		// Establish necessary support classes
+		$obj = new stdClass;
+		foreach (array('config' => 'cfg', 'load' => 'load', 'input' => 'in') as $name => $abbr) {
+			$class = $this->ci_core_class($abbr);
+			$obj->$name = new $class;
+		}
+		$this->ci_instance($obj);
+
+		// Attach session instance locally
+		$config = array('sess_encrypt_cookie' => FALSE, 'sess_use_database' => FALSE, 'sess_table_name' => '',
+			'sess_expiration' => 7200, 'sess_expire_on_close' => FALSE, 'sess_match_ip' => FALSE,
+			'sess_match_useragent' => TRUE, 'sess_cookie_name' => 'ci_session', 'cookie_path' => '',
+			'cookie_domain' => '', 'cookie_secure' => FALSE, 'cookie_httponly' => FALSE, 'sess_time_to_update' => 300,
+			'time_reference' => 'local', 'cookie_prefix' => '', 'encryption_key' => 'foobar',
+			'sess_valid_drivers' => array('Mock_Libraries_Session_native', 'Mock_Libraries_Session_cookie'));
+		$this->session = new Mock_Libraries_Session($config);
+	}
+
+	/**
+	 * Tear down test framework
+	 */
+	public function tear_down()
+	{
+		// Restore environment
+		if (session_id()) session_destroy();
+		$_SESSION = array();
+		$_COOKIE = $this->cookie_vals;
+
+		// Restore settings
+		foreach ($this->settings as $name => $value) {
+			ini_set('session.'.$name, $this->setting_vals[$name]);
+		}
+	}
+
+	/**
+	 * Test set_userdata() function
+	 */
+	public function test_set_userdata()
+	{
+		// Set a userdata message for each driver
+		$cmsg = 'Some test data';
+		$this->session->cookie->set_userdata('test1', $cmsg);
+		$nmsg = 'Other test data';
+		$this->session->native->set_userdata('test1', $nmsg);
+
+		// Verify independent messages
+		$this->assertEquals($this->session->cookie->userdata('test1'), $cmsg);
+		$this->assertEquals($this->session->native->userdata('test1'), $nmsg);
+	}
+
+	/**
+	 * Test the has_userdata() function
+	 */
+	public function test_has_userdata()
+	{
+		// Set a userdata message for each driver
+		$cmsg = 'My test data';
+		$this->session->cookie->set_userdata('test2', $cmsg);
+		$nmsg = 'Your test data';
+		$this->session->native->set_userdata('test2', $nmsg);
+
+		// Verify independent messages
+		$this->assertTrue($this->session->cookie->has_userdata('test2'));
+		$this->assertTrue($this->session->native->has_userdata('test2'));
+	}
+
+	/**
+	 * Test the all_userdata() function
+	 */
+	public function test_all_userdata()
+	{
+		// Set a specific series of data for each driver
+		$cdata = array('one' => 'first', 'two' => 'second', 'three' => 'third', 'foo' => 'bar', 'bar' => 'baz');
+		$this->session->cookie->set_userdata($cdata);
+		$ndata = array('one' => 'gold', 'two' => 'silver', 'three' => 'bronze', 'foo' => 'baz', 'bar' => 'foo');
+		$this->session->native->set_userdata($ndata);
+
+		// Make sure all values are present
+		$call = $this->session->cookie->all_userdata();
+		foreach ($cdata as $key => $value) {
+			$this->assertEquals($value, $call[$key]);
+		}
+		$nall = $this->session->native->all_userdata();
+		foreach ($ndata as $key => $value) {
+			$this->assertEquals($value, $nall[$key]);
+		}
+	}
+
+	/**
+	 * Test the unset_userdata() function
+	 */
+	public function test_unset_userdata()
+	{
+		// Set a userdata message for each driver
+		$cmsg = 'Other test data';
+		$this->session->cookie->set_userdata('test3', $cmsg);
+		$nmsg = 'Sundry test data';
+		$this->session->native->set_userdata('test3', $nmsg);
+
+		// Verify independent messages
+		$this->assertEquals($this->session->cookie->userdata('test3'), $cmsg);
+		$this->assertEquals($this->session->native->userdata('test3'), $nmsg);
+
+		// Unset them and verify absence
+		$this->session->cookie->unset_userdata('test3');
+		$this->session->native->unset_userdata('test3');
+		$this->assertEquals($this->session->cookie->userdata('test3'), NULL);
+		$this->assertEquals($this->session->native->userdata('test3'), NULL);
+	}
+
+	/**
+	 * Test the set_flashdata() function
+	 */
+	public function test_set_flashdata()
+	{
+		// Set flashdata message for each driver
+		$cmsg = 'Some flash data';
+		$this->session->cookie->set_flashdata('test4', $cmsg);
+		$nmsg = 'Other flash data';
+		$this->session->native->set_flashdata('test4', $nmsg);
+
+		// Simulate page reload
+		$this->session->cookie->reload();
+		$this->session->native->reload();
+
+		// Verify independent messages
+		$this->assertEquals($this->session->cookie->flashdata('test4'), $cmsg);
+		$this->assertEquals($this->session->native->flashdata('test4'), $nmsg);
+
+		// Simulate next page reload
+		$this->session->cookie->reload();
+		$this->session->native->reload();
+
+		// Verify absence of messages
+		$this->assertEquals($this->session->cookie->flashdata('test4'), NULL);
+		$this->assertEquals($this->session->native->flashdata('test4'), NULL);
+	}
+
+	/**
+	 * Test the keep_flashdata() function
+	 */
+	public function test_keep_flashdata()
+	{
+		// Set flashdata message for each driver
+		$cmsg = 'My flash data';
+		$this->session->cookie->set_flashdata('test5', $cmsg);
+		$nmsg = 'Your flash data';
+		$this->session->native->set_flashdata('test5', $nmsg);
+
+		// Simulate page reload and verify independent messages
+		$this->session->cookie->reload();
+		$this->session->native->reload();
+		$this->assertEquals($this->session->cookie->flashdata('test5'), $cmsg);
+		$this->assertEquals($this->session->native->flashdata('test5'), $nmsg);
+
+		// Keep messages
+		$this->session->cookie->keep_flashdata('test5');
+		$this->session->native->keep_flashdata('test5');
+
+		// Simulate next page reload and verify message persistence
+		$this->session->cookie->reload();
+		$this->session->native->reload();
+		$this->assertEquals($this->session->cookie->flashdata('test5'), $cmsg);
+		$this->assertEquals($this->session->native->flashdata('test5'), $nmsg);
+
+		// Simulate next page reload and verify absence of messages
+		$this->session->cookie->reload();
+		$this->session->native->reload();
+		$this->assertEquals($this->session->cookie->flashdata('test5'), NULL);
+		$this->assertEquals($this->session->native->flashdata('test5'), NULL);
+	}
+
+	/**
+	 * Test the all_flashdata() function
+	 */
+	public function test_all_flashdata()
+	{
+		// Set a specific series of data for each driver
+		$cdata = array('one' => 'first', 'two' => 'second', 'three' => 'third', 'foo' => 'bar', 'bar' => 'baz');
+		$this->session->cookie->set_flashdata($cdata);
+		$ndata = array('one' => 'gold', 'two' => 'silver', 'three' => 'bronze', 'foo' => 'baz', 'bar' => 'foo');
+		$this->session->native->set_flashdata($ndata);
+
+		// Simulate page reload and make sure all values are present
+		$this->session->cookie->reload();
+		$this->session->native->reload();
+		$this->assertEquals($cdata, $this->session->cookie->all_flashdata());
+		$this->assertEquals($ndata, $this->session->native->all_flashdata());
+	}
+
+	/**
+	 * Test the set_tempdata() function
+	 */
+	public function test_set_tempdata()
+	{
+		// Set tempdata message for each driver - 1 second timeout
+		$cmsg = 'Some temp data';
+		$this->session->cookie->set_tempdata('test6', $cmsg, 1);
+		$nmsg = 'Other temp data';
+		$this->session->native->set_tempdata('test6', $nmsg, 1);
+
+		// Simulate page reload and verify independent messages
+		$this->session->cookie->reload();
+		$this->session->native->reload();
+		$this->assertEquals($this->session->cookie->tempdata('test6'), $cmsg);
+		$this->assertEquals($this->session->native->tempdata('test6'), $nmsg);
+
+		// Wait 2 seconds, simulate page reload and verify message absence
+		sleep(2);
+		$this->session->cookie->reload();
+		$this->session->native->reload();
+		$this->assertEquals($this->session->cookie->tempdata('test6'), NULL);
+		$this->assertEquals($this->session->native->tempdata('test6'), NULL);
+	}
+
+	/**
+	 * Test the unset_tempdata() function
+	 */
+	public function test_unset_tempdata()
+	{
+		// Set tempdata message for each driver - 1 second timeout
+		$cmsg = 'My temp data';
+		$this->session->cookie->set_tempdata('test7', $cmsg, 1);
+		$nmsg = 'Your temp data';
+		$this->session->native->set_tempdata('test7', $nmsg, 1);
+
+		// Verify independent messages
+		$this->assertEquals($this->session->cookie->tempdata('test7'), $cmsg);
+		$this->assertEquals($this->session->native->tempdata('test7'), $nmsg);
+
+		// Unset data and verify message absence
+		$this->session->cookie->unset_tempdata('test7');
+		$this->session->native->unset_tempdata('test7');
+		$this->assertEquals($this->session->cookie->tempdata('test7'), NULL);
+		$this->assertEquals($this->session->native->tempdata('test7'), NULL);
+	}
+
+	/**
+	 * Test the sess_regenerate() function
+	 */
+	public function test_sess_regenerate()
+	{
+		// Get current session id, regenerate, and compare
+		// Cookie driver
+		$oldid = $this->session->cookie->userdata('session_id');
+		$this->session->cookie->sess_regenerate();
+		$newid = $this->session->cookie->userdata('session_id');
+		$this->assertFalse($oldid === $newid);
+
+		// Native driver - bug #55267 (https://bugs.php.net/bug.php?id=55267) prevents testing this
+		/*$oldid = session_id();
+		$this->session->native->sess_regenerate();
+		$oldid = session_id();
+		$this->assertFalse($oldid === $newid);*/
+	}
+
+	/**
+	 * Test the sess_destroy() function
+	 */
+	public function test_sess_destroy()
+	{
+		// Set a userdata message, destroy session, and verify absence
+		$msg = 'More test data';
+
+		// Cookie driver
+		$this->session->cookie->set_userdata('test8', $msg);
+		$this->assertEquals($this->session->cookie->userdata('test8'), $msg);
+		$this->session->cookie->sess_destroy();
+		$this->assertEquals($this->session->cookie->userdata('test8'), NULL);
+
+		// Native driver
+		$this->session->native->set_userdata('test8', $msg);
+		$this->assertEquals($this->session->native->userdata('test8'), $msg);
+		$this->session->native->sess_destroy();
+		$this->assertEquals($this->session->native->userdata('test8'), NULL);
+	}
+}
+