storage enhancements

[m6w6/pq-gateway] / tests / lib / pq / Mapper / StorageTest.php

diff --git a/tests/lib/pq/Mapper/StorageTest.php b/tests/lib/pq/Mapper/StorageTest.php
index c57716269528a0741a5c9593615ef0303f0f5191..0889b13eb16c951e7cf3b3f4ecdbbf7df97e119b 100644 (file)
--- a/tests/lib/pq/Mapper/StorageTest.php
+++ b/tests/lib/pq/Mapper/StorageTest.php
@@ -80,4 +80,18 @@ class StorageTest extends PHPUnit_Framework_TestCase
                $this->storage->discard();
                $this->assertEquals("the day before", $this->storage->get(1)->data);
        }
+
+       function testOf() {
+               $obj = $this->storage->get(1);
+               $ref = new Storage($this->mapper, RefTestModel::class);
+               $this->assertSame($obj->ref1, $ref->of($obj, "test"));
+               $this->assertSame($obj->ref2, $ref->of($obj, "another_test"));
+       }
+
+       function testBy() {
+               $ref = new Storage($this->mapper, RefTestModel::class);
+               $obj = $ref->get([2,2]);
+               $this->assertSame($obj->one, $this->storage->by($obj, "test"));
+               $this->assertSame($obj->two, $this->storage->by($obj, "another_test"));
+       }
 }