tests/codeigniter/database/query_builder/count_test.php

<?php

class Count_test extends CI_TestCase {

	/**
	 * @var object Database/Query Builder holder
	 */
	protected $db;

	public function set_up()
	{
		$this->db = Mock_Database_Schema_Skeleton::init(DB_DRIVER);

		Mock_Database_Schema_Skeleton::create_tables();
		Mock_Database_Schema_Skeleton::create_data();
	}

	// ------------------------------------------------------------------------

	/**
	 * @see ./mocks/schema/skeleton.php
	 */
	public function test_count_all()
	{
		$job_count = $this->db->count_all('job');
		
		// Check the result
		$this->assertEquals(4, $job_count);
	}

	// ------------------------------------------------------------------------

	/**
	 * @see ./mocks/schema/skeleton.php
	 */
	public function test_count_all_results()
	{
		$job_count = $this->db->like('name', 'ian')
		                      ->count_all_results('job');
		
		// Check the result
		$this->assertEquals(2, $job_count);
	}
}