X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fpq-gateway;a=blobdiff_plain;f=tests%2Fsetup.inc;h=fa7af13428ec71522b02d220dd6d81eb3ff0eb8e;hp=798d4a4824ad8eba9213e97e64ab5b593cb5d645;hb=80c4d645e1650304b1dbcae85dbf17a6593896e4;hpb=20d2b6bcce8f1c7a1aaa375b86ffb5be30674956

diff --git a/tests/setup.inc b/tests/setup.inc
index 798d4a4..fa7af13 100644
--- a/tests/setup.inc
+++ b/tests/setup.inc
@@ -1,28 +1,83 @@
 <?php
 
-define("PQ_DSN", "");
-define("PQ_TEST_TABLE_NAME", "test");
-define("PQ_TEST_CREATE_TABLE", sprintf(
-<<<SQL
-	create table %s (
+ini_set("date.timezone", "UTC");
+ini_set("error_reporting", E_ALL);
+
+const PQ_TEST_DSN = "";
+
+const PQ_TEST_SETUP_SQL = <<<SQL
+	create extension if not exists hstore;
+	
+	drop table if exists test cascade;
+	create table test (
 		id serial primary key, 
 		created timestamp, 
 		counter int, 
 		number decimal, 
-		data text
-	)
-SQL
-	, PQ_TEST_TABLE_NAME));
-define("PQ_TEST_DROP_TABLE", sprintf("drop table if exists %s", PQ_TEST_TABLE_NAME));
-define("PQ_TEST_CREATE_DATA", sprintf(
-<<<SQL
-	insert into %1\$s values (default, 'yesterday', -1, -1.1, 'yesterday');
-	insert into %1\$s values (default, 'today', 0, 0, 'today');
-	insert into %1\$s values (default, 'tomorrow', 1, 1.1, 'tomorrow');
-SQL
-	, PQ_TEST_TABLE_NAME
-));
+		data text,
+		list int[],
+		prop hstore
+	);
+	
+	drop table if exists reftest cascade;
+	create table reftest (
+		test_id integer not null references test on delete cascade,
+		another_test_id integer not null references test on delete cascade
+	);
+	
+	insert into test values (default, 'yesterday', -1, -1.1, 'yesterday', '{-1,0,1}');
+	insert into test values (default, 'today', 0, 0, 'today', '{0,1,2}');
+	insert into test values (default, 'tomorrow', 1, 1.1, 'tomorrow', '{1,2,3}');
+	
+	insert into reftest values (1,3);
+	insert into reftest values (2,2);
+	insert into reftest values (3,1);
+SQL;
+
+const PQ_TEST_TEARDOWN_SQL = <<<SQL
+	drop table if exists test cascade;
+	drop table if exists reftest cascade;
+SQL;
+
+include_once __DIR__ . "/../lib/autoload.php";
+
+function executeInConcurrentTransaction(\pq\Query\ExecutorInterface $exec, $sql, array $params = null) {
+	$conn = $exec->getConnection();
+	$exec->setConnection(new pq\Connection(PQ_TEST_DSN));
+	$exec->execute(new \pq\Query\Writer($sql, $params), function(){});
+	$exec->setConnection($conn);
+}
 
-spl_autoload_register(function($c) {
-	if (substr($c,0,3) == "pq\\") return require_once sprintf("%s/../lib/%s.php", __DIR__, strtr($c, "\\", "/"));
-});
+class QueryLogger implements \SplObserver
+{
+	protected $fp;
+	
+	function __construct($logfile = null) {
+		if (!isset($logfile)) {
+			$logfile = __DIR__."/query.log";
+		}
+		if (!$this->fp = @fopen($logfile, "a")) {
+			throw new \RuntimeException(error_get_last()["message"]);
+		}
+	}
+	
+	function __destruct() {
+		if (is_resource($this->fp)) {
+			fclose($this->fp);
+		}
+	}
+	
+	function update(\SplSubject $executor) {
+		$result = $executor->getResult();
+		if (isset($result)) {
+			fprintf($this->fp, "[%s] R %s\n", 
+				date_create()->format("Y-m-d H:i:s"),
+				json_encode($result));
+		} elseif (($query = $executor->getQuery())) {
+			fprintf($this->fp, "[%s] Q %s %% %s\n", 
+				date_create()->format("Y-m-d H:i:s"),
+				preg_replace("/\s+/", " ", $query), 
+				json_encode($query->getParams()));
+		}
+	}
+}