Adds regression tests for missing magic method calls to ORM and IdiormResultSet

1 files changed, 15 insertions, 0 deletions

diff --git a/test/ORMTest.php b/test/ORMTest.php
index ebd3d17..e0e4900 100644
--- a/test/ORMTest.php
+++ b/test/ORMTest.php
@@ -84,4 +84,19 @@ class ORMTest extends PHPUnit_Framework_TestCase {
         $this->assertInstanceOf('MockPDOStatement', $statement);
     }
 
+    /**
+     * @expectedException IdiormMethodMissingException
+     */
+    public function testInvalidORMFunctionCallShouldCreateException() {
+        $orm = ORM::for_table('test');
+        $orm->invalidFunctionCall();
+    }
+
+    /**
+     * @expectedException IdiormMethodMissingException
+     */
+    public function testInvalidResultsSetFunctionCallShouldCreateException() {
+        $resultSet = ORM::for_table('test')->find_result_set();
+        $resultSet->invalidFunctionCall();
+    }
 }
\ No newline at end of file