X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fpq-gateway;a=blobdiff_plain;f=tests%2Flib%2Fpq%2FGateway%2FRowTest.php;h=00dbbbb6c9c2eb806a877e3177adb1036d05ed54;hp=fc0809468ea0a7aa2f804de834263ab8c410e5ab;hb=580991717f5e8bb237403757e2111a8d04aca616;hpb=0e2eb1f13ef60ce9a8709354136c42f7d87b2345

diff --git a/tests/lib/pq/Gateway/RowTest.php b/tests/lib/pq/Gateway/RowTest.php
index fc08094..00dbbbb 100644
--- a/tests/lib/pq/Gateway/RowTest.php
+++ b/tests/lib/pq/Gateway/RowTest.php
@@ -23,6 +23,7 @@ class RowTest extends \PHPUnit_Framework_TestCase {
 		$this->conn->exec(PQ_TEST_DATA);
 		Table::$defaultConnection = $this->conn;
 		$this->table = new Table("test");
+		$this->table->getQueryExecutor()->attach(new \QueryLogger());
 	}
 
 	protected function tearDown() {
@@ -51,4 +52,59 @@ class RowTest extends \PHPUnit_Framework_TestCase {
 		$row = new Row($this->table);
 		$this->assertSame($this->table, $row->getTable());
 	}
+	
+	function testPessimisticLock() {
+		$this->table->setLock(new Table\PessimisticLock);
+		$txn = $this->table->getConnection()->startTransaction();
+		$row = $this->table->find(null, null, 1)->current();
+		$row->data = "foo";
+		$row->update();
+		$txn->commit();
+		$this->assertSame("foo", $row->data->get());
+	}
+	
+	function testPessimisticLockFail() {
+		$this->table->setLock(new Table\PessimisticLock);
+		$txn = $this->table->getConnection()->startTransaction();
+		$row = $this->table->find(null, null, 1)->current();
+		$row->data = "foo";
+		executeInConcurrentTransaction(
+			$this->table->getQueryExecutor(),
+			"UPDATE {$this->table->getName()} SET data='bar' WHERE id=\$1", 
+			array($row->id->get()));
+		$this->setExpectedException("\\UnexpectedValueException", "Row has already been modified");
+		$row->update();
+		$txn->commit();
+	}
+	
+	function testOptimisticLock() {
+		$this->table->setLock(new Table\OptimisticLock("counter"));
+		$row = $this->table->find(null, null, 1)->current();
+		$cnt = $row->counter->get();
+		$row->data = "foo";
+		$row->update();
+		$this->assertEquals("foo", $row->data->get());
+		$this->assertEquals($cnt +1, $row->counter->get());
+	}
+	
+	function testOptimisticLockFail() {
+		$this->table->setLock(new Table\OptimisticLock("counter"));
+		$row = $this->table->find(null, null, 1)->current();
+		$cnt = $row->counter->get();
+		$row->data = "foo";
+		executeInConcurrentTransaction(
+			$this->table->getQueryExecutor(), 
+			"UPDATE {$this->table->getName()} SET counter = 10 WHERE id=\$1", 
+			array($row->id->get()));
+		$this->setExpectedException("\\UnexpectedValueException", "No row updated");
+		$row->update();
+	}
+	
+	function testRef() {
+		foreach ($this->table->find() as $row) {
+			foreach ($row->reftest() as $ref) {
+				$this->assertEquals($row->id->get(), $ref->test->id->get());
+			}
+		}
+	}
 }