Fix issue #7

src/Models/ElementModel.php

@@ -46,6 +46,13 @@ class ElementModel extends BitrixModel
      */
     const IBLOCK_ID = null;
 
+    /**
+     * Bitrix entity object.
+     *
+     * @var object
+     */
+    public static $bxObject;
+
     /**
      * Corresponding object class name.
      *

src/Models/SectionModel.php

@@ -43,6 +43,13 @@ class SectionModel extends BitrixModel
      */
     const IBLOCK_ID = null;
 
+    /**
+     * Bitrix entity object.
+     *
+     * @var object
+     */
+    public static $bxObject;
+
     /**
      * Corresponding object class name.
      *

src/Models/UserModel.php

@@ -35,6 +35,13 @@
  */
 class UserModel extends BitrixModel
 {
+    /**
+     * Bitrix entity object.
+     *
+     * @var object
+     */
+    public static $bxObject;
+
     /**
      * Corresponding object class name.
      *

tests/ElementModelTest.php

@@ -4,6 +4,8 @@
 
 use Arrilot\BitrixModels\Queries\ElementQuery;
 use Arrilot\Tests\BitrixModels\Stubs\TestElement;
+use Arrilot\Tests\BitrixModels\Stubs\TestSection;
+use Arrilot\Tests\BitrixModels\Stubs\TestUser;
 use Mockery as m;
 
 class ElementModelTest extends TestCase
@@ -393,4 +395,20 @@ public function testAccessors()
         $this->assertTrue(isset($element['ACCESSOR_THREE']));
         $this->assertFalse(!empty($element['ACCESSOR_THREE']));
     }
+
+    public function testItPlaysNiceWithOtherBitrixModels()
+    {
+        $elementObject = m::mock('element_object');
+        TestElement::$bxObject = $elementObject;
+
+        $sectionObject = m::mock('section_object');
+        TestSection::$bxObject = $sectionObject;
+
+        $userObject = m::mock('user_object');
+        TestUser::$bxObject = $userObject;
+
+        $this->assertSame(TestElement::$bxObject, $elementObject);
+        $this->assertSame(TestSection::$bxObject, $sectionObject);
+        $this->assertSame(TestUser::$bxObject, $userObject);
+    }
 }