<?php

class DnsSrvDiscovererTest extends PHPUnit_Framework_TestCase {
	/**
	 * @covers DnsSrvDiscoverer
	 * @dataProvider provideRecords
	 */
	public function testPickServer( $params, $expected ) {
		$discoverer = new DnsSrvDiscoverer( '_etcd._tcp.eqiad.wmnet' );
		$record = $discoverer->pickServer( $params );

		$this->assertEquals( $expected, $record );

	}

	public static function provideRecords() {
		return [
			[
				[ // record list
					[
						'target' => 'conf1003.eqiad.wmnet',
						'port' => 'SRV',
						'pri' => 0,
						'weight' => 1,
					],
					[
						'target' => 'conf1002.eqiad.wmnet',
						'port' => 'SRV',
						'pri' => 1,
						'weight' => 1,
					],
					[
						'target' => 'conf1001.eqiad.wmnet',
						'port' => 'SRV',
						'pri' => 2,
						'weight' => 1,
					],
				], // selected record
				[
					'target' => 'conf1003.eqiad.wmnet',
					'port' => 'SRV',
					'pri' => 0,
					'weight' => 1,
				]
			],
			[
				[ // record list
					[
						'target' => 'conf1003or2.eqiad.wmnet',
						'port' => 'SRV',
						'pri' => 0,
						'weight' => 1,
					],
					[
						'target' => 'conf1003or2.eqiad.wmnet',
						'port' => 'SRV',
						'pri' => 0,
						'weight' => 1,
					],
					[
						'target' => 'conf1001.eqiad.wmnet',
						'port' => 'SRV',
						'pri' => 2,
						'weight' => 1,
					],
					[
						'target' => 'conf1004.eqiad.wmnet',
						'port' => 'SRV',
						'pri' => 2,
						'weight' => 1,
					],
					[
						'target' => 'conf1005.eqiad.wmnet',
						'port' => 'SRV',
						'pri' => 3,
						'weight' => 1,
					],
				], // selected record
				[
					'target' => 'conf1003or2.eqiad.wmnet',
					'port' => 'SRV',
					'pri' => 0,
					'weight' => 1,
				]
			],
		];
	}

	public function testRemoveServer() {
		$dsd = new DnsSrvDiscoverer( 'localhost' );

		$servers = [
			[
				'target' => 'conf1001.eqiad.wmnet',
				'port' => 35,
				'pri' => 2,
				'weight' => 1,
			],
			[
				'target' => 'conf1004.eqiad.wmnet',
				'port' => 74,
				'pri' => 2,
				'weight' => 1,
			],
			[
				'target' => 'conf1005.eqiad.wmnet',
				'port' => 77,
				'pri' => 3,
				'weight' => 1,
			],
		];
		$server = $servers[1];

		$expected = [
			[
				'target' => 'conf1001.eqiad.wmnet',
				'port' => 35,
				'pri' => 2,
				'weight' => 1,
			],
			[
				'target' => 'conf1005.eqiad.wmnet',
				'port' => 77,
				'pri' => 3,
				'weight' => 1,
			],
		];

		$this->assertEquals(
			$expected,
			$dsd->removeServer( $server, $servers ),
			"Correct server removed"
		);
		$this->assertEquals(
			$expected,
			$dsd->removeServer( $server, $servers ),
			"Nothing to remove"
		);
	}
}