eigenclass stores object reference

shuber · Mar 1, 2011 · c596621 · c596621
1 parent c5bfd14
commit c596621
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/lib/eigenclass.php b/lib/eigenclass.php
@@ -2,9 +2,11 @@
 
 class Eigenclass extends Klass {
 
-    function __construct($class) {
-        if (is_object($class)) $class = get_class($class);
-        parent::__construct($class, null, false);
+    protected $_object;
+
+    function __construct(&$object) {
+        $this->_object = $object;
+        parent::__construct(get_class($object), null, false);
     }
 
     function __include($modules, $instance = false) {
@@ -23,6 +25,10 @@ function callee($method, &$caller = null) {
         }
     }
 
+    function object() {
+        return $this->_object;
+    }
+
     function reference() {
         return parent::instance($this->_name);
     }

diff --git a/test/unit/eigenclass_test.php b/test/unit/eigenclass_test.php
@@ -47,8 +47,8 @@ function test_should_return_class_reference() {
             assert_identical($this->user_class, $this->user->__class()->reference());
         }
 
-        function test_should_not_create_class() {
-            assert_throws('InvalidArgumentException', function() { new Eigenclass('InvalidClass'); });
+        function test_should_return_object() {
+            assert_identical($this->user_class, $this->user_class->__class()->object());
         }
 
     }