X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;ds=sidebyside;f=tests%2Fsetup.inc;h=bec6c718515da035756d829d47f00b548e065a2b;hb=580991717f5e8bb237403757e2111a8d04aca616;hp=ee9733c12bb975238d16c5044f0754498a8b471e;hpb=0e2eb1f13ef60ce9a8709354136c42f7d87b2345;p=m6w6%2Fpq-gateway

diff --git a/tests/setup.inc b/tests/setup.inc
index ee9733c..bec6c71 100644
--- a/tests/setup.inc
+++ b/tests/setup.inc
@@ -42,3 +42,49 @@ SQL;
 spl_autoload_register(function($c) {
 	if (substr($c,0,3) == "pq\\") return require_once sprintf("%s/../lib/%s.php", __DIR__, strtr($c, "\\", "/"));
 });
+
+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);
+}
+
+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([
+					"S" => $result->statusMessage, 
+					"N" => $result->numRows, 
+					"C" => $result->numCols,
+					"A" => $result->affectedRows
+				]));
+		} 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()));
+		}
+	}
+}