<?php
// $Id: SpamTest.php,v 1.18 2007/07/02 14:51:40 henoheno Exp $
// Copyright (C) 2007 heno
//
// Design test case for spam.php (called from runner.php)

if (! defined('SPAM_INI_FILE')) define('SPAM_INI_FILE', 'spam.ini.php');

require_once('spam.php');
require_once('PHPUnit/PHPUnit.php');

class SpamTest extends PHPUnit_TestCase
{
	function setup_string_null()
	{
		return array(
			'[NULL]'	=> NULL,
			'[TRUE]'	=> TRUE,
			'[FALSE]'	=> FALSE,
			'[array(foobar)]' => array('foobar'),
			'[]'		=> '',
			'[0]'		=> 0,
			'[1]'		=> 1
		);
	}

	function testFunc_strings()
	{
		// 1st argument: Null
		$this->assertEquals('',  strings(NULL,  0));
		$this->assertEquals('',  strings(TRUE,  0));
		$this->assertEquals('',  strings(FALSE, 0));
		$this->assertEquals('',  strings('',    0));
		$this->assertEquals('0', strings(0,     0));
		$this->assertEquals('1', strings(1,     0));

		// Setup
		$t1 = '1'    . "\n";
		$t2 = '12'   . "\n";
		$t3 = '123'  . "\n";
		$t4 = '1234' . "\n";
		$t5 = '12345';
		$test = $t1 . $t2 . $t3 . $t4 . $t5;

		// Minimum length
		$this->assertEquals($t1 . $t2 . $t3 . $t4 . $t5, strings($test, -1));
		$this->assertEquals($t1 . $t2 . $t3 . $t4 . $t5, strings($test,  0));
		$this->assertEquals($t1 . $t2 . $t3 . $t4 . $t5, strings($test,  1));
		$this->assertEquals(      $t2 . $t3 . $t4 . $t5, strings($test,  2));
		$this->assertEquals(            $t3 . $t4 . $t5, strings($test,  3));
		$this->assertEquals(                  $t4 . $t5, strings($test,  4));
		$this->assertEquals(                  $t4 . $t5, strings($test)); // Default
		$this->assertEquals(                        $t5, strings($test,  5));

		// Preserve the last newline
		$this->assertEquals($t4 . $t5,        strings($test       , 4));
		$this->assertEquals($t4 . $t5 . "\n", strings($test . "\n", 4));

		// Ignore sequential spaces, and spaces at the beginning/end of lines
		$test = '   A' . '	' . '   ' . 'B	';
		$this->assertEquals($test, strings($test, 0, FALSE));
		$this->assertEquals('A B', strings($test, 0, TRUE ));
	}

	function testFunc_array_count_leaves()
	{
		// Empty array = 0, if option is not set
		$array = array();
		$this->assertEquals(0, array_count_leaves($array, FALSE));
		$this->assertEquals(1, array_count_leaves($array, TRUE));
		$array = array(
			array(
				array()
			)
		);
		$this->assertEquals(0, array_count_leaves($array, FALSE));
		$this->assertEquals(1, array_count_leaves($array, TRUE));

		// One leaf = 1
		foreach(array(NULL, TRUE, FALSE, -1, 0, 1, '', 'foobar') as $value) {
			$this->assertEquals(1, array_count_leaves($value, FALSE));
			$this->assertEquals(1, array_count_leaves($value, TRUE));
		}

		// Compisite
		$array = array(
			1,
			'v1',
			array(),	// Empty array
			array(
				2,
				'v2',
				'k1' => TRUE,
				'k2' => FALSE,
				'k3' => array(),	// Empty array
				'k4' => array(
					3,
					'v3',
					'k5' => NULL,
					'k6' => array(),	// Empty array
				),
			),
			'k7'  => 4,
			'k8'  => 'v4',
			'k9'  => array(),	// Empty array
			'k10' => array(
				5,
				'v5',
				'k11' => NULL,
				'k12' => array(),	// Empty array
			),
		);
		$this->assertEquals(14, array_count_leaves($array, FALSE));
		$this->assertEquals(19, array_count_leaves($array, TRUE));
	}

	function testPhpFunc_array_unique()
	{
		$this->assertEquals(array(1), array_unique(array(1, 1)));

		// Keys are preserved, array()s inside are preserved
		$this->assertEquals(
			array(0, 2 => array(1, 1), 3 => 2),
			array_unique(
				array(0, 0, array(1, 1), 2, 2)
			)
		);

		// Keys are preserved
		$this->assertEquals(
			array(0, 2 => array(1, 1), 3 => 2),
			array_unique(array(0, 0, array(1, 1), 2, 2))
		);

		// ONLY the first array() is preserved
		$this->assertEquals(
			array(0 => array(1, 1)),
			array_unique(array_unique(array(0 => array(1, 1), 'a' => array(2,2), 'b' => array(3, 3))))
		);
	}

	// And array_unique_recursive()
	function testPhPFunc_array_merge_recursive()
	{
		$array1 = array(1);
		$array2 = array(1);
		$result = array_merge_recursive($array1, $array2);
		$this->assertEquals(array(1, 1), $result);
		$result = array_unique_recursive($result);
		$this->assertEquals(array(1),    $result);

		$array1 = array(2);
		$array2 = array(1);
		$result = array(2, 1);
		$this->assertEquals($result, array_merge_recursive($array1, $array2));

		// All NUMERIC keys are always renumbered from 0
		$array1 = array('10' => 'f3');
		$array2 = array('10' => 'f4');
		$result = array('f3', 'f4');
		$this->assertEquals($result, array_merge_recursive($array1, $array2));

		// One more thing ...
		$array1 = array('20' => 'f5');
		$array2 = array();
		$result = array('f5');
		$this->assertEquals($result, array_merge_recursive($array1, $array2));

		// Non-numeric keys and values will be marged as you think
		$array1 = array('a' => 'f1');
		$array2 = array('a' => 'f2');
		$result = array('a' => array('f1', 'f2'));
		$this->assertEquals($result, array_merge_recursive($array1, $array2));

		// Non-numeric keys: An array and a value will be marged
		$array1 = array('b' => array('k1'));
		$array2 = array('b' => 'k2');
		$result = array('b' => array('k1', 'k2'));
		$this->assertEquals($result, array_merge_recursive($array1, $array2));

		// Combination
		$array1 = array(
			2,
			'a' => 'f1',
			'10' => 'f3',
			'20' => 'f5',
			'b' => array('k1'),
		);
		$array2 = array(
			1,
			'a' => 'f2',
			'10' => 'f4',
			'b' => 'k2',
		);
		$result = array (
			2,
			'a' => array (
				'f1',
				'f2',
			),
			'f3',
			'f5',
			'b' => array (
				'k1',
				'k2',
			),
			1,
			'f4',
		);
		$this->assertEquals($result, array_merge_recursive($array1, $array2));

		// Values will not be unique
		$array1 = array(5, 4);
		$array2 = array(4, 5);
		$result = array_merge_recursive($array1, $array2);
		$this->assertEquals(array(5, 4, 4, 5), $result);
		$result = array_unique_recursive($result);
		$this->assertEquals(array(5, 4),       $result);

		// One more thing ...
		$array1 = array('b' => array('k3'));
		$array2 = array('b' => 'k3');
		$result = array_merge_recursive($array1, $array2);
		$this->assertEquals(array('b' => array('k3', 'k3')), $result);
		$result = array_unique_recursive($result);
		$this->assertEquals(array('b' => array('k3')),       $result);
	}

	function testFunc_generate_glob_regex()
	{
		// 1st argument: Null
		foreach($this->setup_string_null() as $key => $value){
			$this->assertEquals('', generate_glob_regex($value), $key);
		}

		$this->assertEquals('.*\.txt', generate_glob_regex('*.txt'));
		$this->assertEquals('A.A',     generate_glob_regex('A?A'));
	}

	function testFunc_generate_host_regex()
	{
		// 1st argument: Null
		foreach($this->setup_string_null() as $key => $value){
			$this->assertEquals('', generate_host_regex($value), $key);
		}

		$this->assertEquals('localhost',             generate_host_regex('localhost'));
		$this->assertEquals('example\.org',          generate_host_regex('example.org'));
		$this->assertEquals('(?:.*\.)?example\.org', generate_host_regex('.example.org'));
		$this->assertEquals('.*\.example\.org',      generate_host_regex('*.example.org'));
		$this->assertEquals('.*\..*\.example\.org',  generate_host_regex('*.*.example.org'));
		$this->assertEquals('10\.20\.30\.40',        generate_host_regex('10.20.30.40'));

		// Should match with 192.168.0.0/16
		//$this->assertEquals('192\.168\.',       generate_host_regex('192.168.'));
	}

	function testFunc_get_blocklist()
	{
		if (! defined('SPAM_INI_FILE') || ! file_exists(SPAM_INI_FILE)) {
			$this->fail('SPAM_INI_FILE not defined or not found');
			return;
		}

		// get_blocklist_add()
		$array = array();

		get_blocklist_add($array,   'foo', 'bar');
		$this->assertEquals(1,      count($array));
		$this->assertEquals('bar',  $array['foo']);

		get_blocklist_add($array,   'hoge', 'fuga');
		$this->assertEquals(2,      count($array));
		$this->assertEquals('bar',  $array['foo']);
		$this->assertEquals('fuga', $array['hoge']);

		get_blocklist_add($array,   -1, '*.txt');
		$this->assertEquals(3,      count($array));
		$this->assertEquals('bar',  $array['foo']);
		$this->assertEquals('fuga', $array['hoge']);
		$this->assertEquals('/^.*\.txt$/i', $array['*.txt']);

		// get_blocklist()
		// ALL
		$array = get_blocklist();
		$this->assertTrue(isset($array['C']));
		$this->assertTrue(isset($array['goodhost']));
		// badhost
		$array = get_blocklist('B-1');
		$this->assertTrue(isset($array['*.blogspot.com']));
		// goodhost
		$array = get_blocklist('goodhost');
		$this->assertTrue(isset($array['IANA-examples']));
	}

	function testFunc_whois_responsibility()
	{
		// 1st argument: Null
		foreach($this->setup_string_null() as $key => $value){
			$this->assertEquals('',        whois_responsibility($value), $key);
		}

		// 'act.edu.au' is known as 3rd level domain
		$this->AssertEquals('bar.act.edu.au', whois_responsibility('foo.bar.act.edu.au'));
		$this->AssertEquals('bar.act.edu.au', whois_responsibility('bar.act.edu.au'));
		$this->AssertEquals('act.edu.au',  whois_responsibility('act.edu.au'));
		$this->AssertEquals('edu.au',      whois_responsibility('edu.au'));
		$this->AssertEquals('au',          whois_responsibility('au'));

		// 'co.uk' is known as 2nd level domain
		$this->AssertEquals('bar.co.uk',   whois_responsibility('foo.bar.co.uk'));
		$this->AssertEquals('bar.co.uk',   whois_responsibility('bar.co.uk'));
		$this->AssertEquals('co.uk',       whois_responsibility('co.uk'));
		$this->AssertEquals('uk',          whois_responsibility('uk'));

		// 'bar.uk' is not 2nd level (implicit responsibility)
		$this->AssertEquals('bar.uk',      whois_responsibility('foo.bar.uk'));
		$this->AssertEquals('bar.uk',      whois_responsibility('bar.uk'));

		// IPv4
		$this->AssertEquals('192.168.0.1', whois_responsibility('192.168.0.1'));

		// Invalid Top-Level Domain (With implicit)
		$this->AssertEquals('bar.local',  whois_responsibility('foo.bar.local'));	// Implicit responsibility
		$this->AssertEquals('bar.local',  whois_responsibility('bar.local'));
		$this->AssertEquals('local',      whois_responsibility('local'));
		$this->AssertEquals('localhost',  whois_responsibility('localhost'));
		$this->AssertEquals('s',          whois_responsibility('s'));
	}
}

?>