<?php

class DBTest extends \Codeception\Test\Unit
{
	public function _before()
	{
		$oDB = Rhymix\Framework\DB::getInstance();
		$oDB->setDebugComment(false);
	}

	public function testGetInstance()
	{
		$oDB = Rhymix\Framework\DB::getInstance();
		$this->assertTrue($oDB instanceof Rhymix\Framework\DB);
		$this->assertEquals($oDB, \DB::getInstance());
		$this->assertTrue(\DB::getInstance() instanceof Rhymix\Framework\DB);
		$this->assertTrue($oDB->getHandle() instanceof Rhymix\Framework\Helpers\DBHelper);
	}

	public function testConnectDisconnect()
	{
		$oDB = Rhymix\Framework\DB::getInstance('master');
		$this->assertTrue(is_object($oDB->getHandle()));

		$oDB->disconnect();
		$this->assertTrue(is_null($oDB->getHandle()));
		$this->assertFalse($oDB->isConnected());

		$oDB->connect(config('db.master'));
		$this->assertTrue(is_object($oDB->getHandle()));
		$this->assertTrue($oDB->isConnected());

		$oDB = Rhymix\Framework\DB::getInstance('master');
		$this->assertTrue(is_object($oDB->getHandle()));
	}

	public function testCompatProperties()
	{
		$oDB = Rhymix\Framework\DB::getInstance();
		$this->assertEquals('mysql', $oDB->db_type);
		$this->assertEquals($oDB->getHandle()->getAttribute(\PDO::ATTR_SERVER_VERSION), $oDB->db_version);
		$this->assertEquals(Rhymix\Framework\Config::get('db.master.prefix'), $oDB->prefix);
		$this->assertTrue($oDB->use_prepared_statements);
		$this->assertNull($oDB->some_nonexistent_property);
	}

	public function testPrepare()
	{
		$oDB = Rhymix\Framework\DB::getInstance();
		$prefix = Rhymix\Framework\Config::get('db.master.prefix');

		$stmt = $oDB->prepare('SELECT * FROM documents WHERE document_srl = ?');
		$this->assertTrue($stmt instanceof Rhymix\Framework\Helpers\DBStmtHelper);
		if ($prefix)
		{
			$this->assertEquals('SELECT * FROM `' . $prefix . 'documents` AS `documents` WHERE document_srl = ?', $stmt->queryString);
		}
		else
		{
			$this->assertEquals('SELECT * FROM documents WHERE document_srl = ?', $stmt->queryString);
		}

		$this->assertTrue($stmt->execute([123]));
		$this->assertTrue($stmt->execute([456]));
		$this->assertTrue($stmt->execute([789]));
		$this->assertTrue(is_array($stmt->fetchAll()));
		$this->assertTrue($stmt->closeCursor());
	}

	public function testQuery()
	{
		$oDB = Rhymix\Framework\DB::getInstance();
		$prefix = Rhymix\Framework\Config::get('db.master.prefix');

		$stmt = $oDB->query('SELECT * FROM documents WHERE document_srl = 123');
		$this->assertTrue($stmt instanceof Rhymix\Framework\Helpers\DBStmtHelper);
		if ($prefix)
		{
			$this->assertEquals('SELECT * FROM `' . $prefix . 'documents` AS `documents` WHERE document_srl = 123', $stmt->queryString);
		}
		else
		{
			$this->assertEquals('SELECT * FROM documents WHERE document_srl = 123', $stmt->queryString);
		}

		$this->assertTrue(is_array($stmt->fetchAll()));
		$this->assertTrue($stmt->closeCursor());

		$stmt = $oDB->query('SELECT * FROM documents WHERE document_srl = ?', [123]);
		$this->assertTrue($stmt instanceof Rhymix\Framework\Helpers\DBStmtHelper);
		if ($prefix)
		{
			$this->assertEquals('SELECT * FROM `' . $prefix . 'documents` AS `documents` WHERE document_srl = ?', $stmt->queryString);
		}
		else
		{
			$this->assertEquals('SELECT * FROM documents WHERE document_srl = ?', $stmt->queryString);
		}

		$this->assertTrue(is_array($stmt->fetchAll()));
		$this->assertTrue($stmt->closeCursor());

		$stmt = $oDB->query('SELECT * FROM documents WHERE document_srl = ? AND status = ?', 123, 'PUBLIC');
		$this->assertTrue($stmt instanceof Rhymix\Framework\Helpers\DBStmtHelper);
		if ($prefix)
		{
			$this->assertEquals('SELECT * FROM `' . $prefix . 'documents` AS `documents` WHERE document_srl = ? AND status = ?', $stmt->queryString);
		}
		else
		{
			$this->assertEquals('SELECT * FROM documents WHERE document_srl = ? AND status = ?', $stmt->queryString);
		}

		$this->assertTrue(is_array($stmt->fetchAll()));
		$this->assertTrue($stmt->closeCursor());
	}

	public function testTransaction()
	{
		$oDB = Rhymix\Framework\DB::getInstance();
		$this->assertEquals(0, $oDB->getTransactionLevel());
		$this->assertEquals(1, $oDB->beginTransaction());
		$this->assertEquals(1, $oDB->getTransactionLevel());
		$this->assertEquals(2, $oDB->begin());
		$this->assertEquals(2, $oDB->getTransactionLevel());
		$this->assertEquals(1, $oDB->rollback());
		$this->assertEquals(1, $oDB->getTransactionLevel());
		$this->assertEquals(0, $oDB->commit());
		$this->assertEquals(0, $oDB->getTransactionLevel());
	}

	public function testAddPrefixes()
	{
		$oDB = Rhymix\Framework\DB::getInstance();
		$prefix = Rhymix\Framework\Config::get('db.master.prefix');

		$source = 'SELECT a, b, c FROM documents JOIN comments ON documents.document_srl = comment.document_srl WHERE documents.member_srl = ?';
		$target = 'SELECT a, b, c FROM `' . $prefix . 'documents` AS `documents` JOIN `' . $prefix . 'comments` AS `comments` ' .
			'ON documents.document_srl = comment.document_srl WHERE documents.member_srl = ?';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'SELECT a AS aa FROM documents as foo JOIN bar ON documents.a = bar.a';
		$target = 'SELECT a AS aa FROM `' . $prefix . 'documents` AS `foo` JOIN `' . $prefix . 'bar` AS `bar` ON documents.a = bar.a';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'INSERT INTO documents (a, b, c) VALUES (?, ?, ?)';
		$target = 'INSERT INTO `' . $prefix . 'documents` (a, b, c) VALUES (?, ?, ?)';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'INSERT INTO documents (a, b, c) SELECT d, e, f FROM old_documents WHERE g = ?';
		$target = 'INSERT INTO `' . $prefix . 'documents` (a, b, c) SELECT d, e, f FROM `' . $prefix . 'old_documents` AS `old_documents` WHERE g = ?';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'UPDATE documents SET a = ?, b = ? WHERE c = ?';
		$target = 'UPDATE `' . $prefix . 'documents` SET a = ?, b = ? WHERE c = ?';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'DELETE FROM documents WHERE d = ?';
		$target = 'DELETE FROM `' . $prefix . 'documents` WHERE d = ?';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'WITH cte AS (SELECT * FROM documents) SELECT * FROM cte WHERE document_srl = ?';
		$target = 'WITH cte AS (SELECT * FROM `' . $prefix . 'documents` AS `documents`) SELECT * FROM `cte` WHERE document_srl = ?';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'WITH RECURSIVE cte AS (SELECT * FROM documents INNER JOIN `cte`) SELECT * FROM cte JOIN member on cte.member_srl = member.member_srl';
		$target = 'WITH RECURSIVE cte AS (SELECT * FROM `' . $prefix . 'documents` AS `documents` INNER JOIN `cte`) SELECT * FROM `cte` JOIN `rx_member` AS `member` on cte.member_srl = member.member_srl';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'WITH RECURSIVE `cte` AS (SELECT * FROM tbl INNER JOIN cte AS h) SELECT * FROM cte WHERE a = ?';
		$target = 'WITH RECURSIVE `cte` AS (SELECT * FROM `' . $prefix . 'tbl` AS `tbl` INNER JOIN `cte` AS `h`) SELECT * FROM `cte` WHERE a = ?';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'INSERT INTO documents (a, b, c) VALUES (?, ?, ?) ON DUPLICATE KEY UPDATE b = ?, c = ?';
		$target = 'INSERT INTO `' . $prefix . 'documents` (a, b, c) VALUES (?, ?, ?) ON DUPLICATE KEY UPDATE b = ?, c = ?';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = "LOAD DATA LOCAL INFILE '/tmp/foo.csv' INTO TABLE foo_table FIELDS TERMINATED BY ',' ENCLOSED BY '\"' LINES TERMINATED BY '\\n' (a, b, c)";
		$target = "LOAD DATA LOCAL INFILE '/tmp/foo.csv' INTO TABLE `" . $prefix . "foo_table` FIELDS TERMINATED BY ',' ENCLOSED BY '\"' LINES TERMINATED BY '\\n' (a, b, c)";
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'ALTER TABLE documents ADD INDEX idx_foo (a, b)';
		$target = 'ALTER TABLE `' . $prefix . 'documents` ADD INDEX idx_foo (a, b)';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'TRUNCATE TABLE documents';
		$target = 'TRUNCATE TABLE `' . $prefix . 'documents`';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'DROP TABLE documents';
		$target = 'DROP TABLE `' . $prefix . 'documents`';
		$this->assertEquals($target, $oDB->addPrefixes($source));

		$source = 'update documents set a = ?, b = ? where c = ?';
		$this->assertEquals($source, $oDB->addPrefixes($source));

		$source = 'delete from documents where d = ?';
		$this->assertEquals($source, $oDB->addPrefixes($source));
	}

	public function testIsTableColumnIndexExists()
	{
		$oDB = Rhymix\Framework\DB::getInstance();
		$this->assertTrue($oDB->isTableExists('documents'));
		$this->assertTrue($oDB->isColumnExists('documents', 'document_srl'));
		$this->assertTrue($oDB->isIndexExists('documents', 'idx_regdate'));
		$this->assertFalse($oDB->isTableExists('nxdocuments'));
		$this->assertFalse($oDB->isColumnExists('documents', 'document_nx'));
		$this->assertFalse($oDB->isIndexExists('documents', 'idx_regex'));
	}

	public function testGetColumnInfo()
	{
		$oDB = Rhymix\Framework\DB::getInstance();
		$info = $oDB->getColumnInfo('documents', 'document_srl');
		$this->assertTrue(is_object($info));
		$this->assertEquals('document_srl', $info->name);
		$this->assertEquals('bigint', $info->dbtype);
		$this->assertEquals('bignumber', $info->xetype);
		$this->assertNull($info->default_value);
		$this->assertTrue($info->notnull);
	}

	public function testGetIndexInfo()
	{
		$oDB = Rhymix\Framework\DB::getInstance();

		$info = $oDB->getIndexInfo('member', 'idx_nick_name');
		$this->assertTrue(is_object($info));
		$this->assertFalse($info->is_unique);
		$this->assertEquals(1, count($info->columns));
		$this->assertEquals('nick_name', $info->columns[0]->name);
		$this->assertNull($info->columns[0]->size);
		$this->assertNotNull($info->columns[0]->cardinality);

		$info = $oDB->getIndexInfo('module_extra_vars', 'unique_module_vars');
		$this->assertTrue(is_object($info));
		$this->assertTrue($info->is_unique);
		$this->assertEquals(2, count($info->columns));
		$this->assertEquals('module_srl', $info->columns[0]->name);
		$this->assertEquals('name', $info->columns[1]->name);
		$this->assertNull($info->columns[1]->size);

		$info = $oDB->getIndexInfo('documents', 'idx_nonexistent');
		$this->assertNull($info);
	}

	public function testIsValidOldPassword()
	{
		$oDB = Rhymix\Framework\DB::getInstance();
		$password = 'foobar^\'1233243';
		$saved1 = '*AA82FF6C7930626A138D0CF3E42D9581D60A85BB';
		$saved2 = '5567cb961d0e218b';
		$saved3 = str_replace('A', 'E', $saved1);
		$saved4 = str_replace('6', '9', $saved2);
		$this->assertTrue($oDB->isValidOldPassword($password, $saved1));
		$this->assertTrue($oDB->isValidOldPassword($password, $saved2));
		$this->assertFalse($oDB->isValidOldPassword($password, $saved3));
		$this->assertFalse($oDB->isValidOldPassword($password, $saved4));
	}

	public function testAddQuotes()
	{
		$oDB = Rhymix\Framework\DB::getInstance();
		$string = 'hello world \' or 1 = 1';
		$result = 'hello world \\\' or 1 = 1';
		$this->assertEquals($result, $oDB->addQuotes($string));
		$this->assertEquals('foobar', $oDB->addQuotes('foobar'));
		$this->assertEquals('123.45', $oDB->addQuotes(123.45));
		$this->assertEquals('-12345', $oDB->addQuotes(-12345));
	}
}