Add use_id_column method to allow specifying the ID column to use on a per-instance basis

1 files changed, 12 insertions, 0 deletions

diff --git a/test/test_queries.php b/test/test_queries.php
index 013b897..a7fbafa 100644
--- a/test/test_queries.php
+++ b/test/test_queries.php
@@ -136,5 +136,17 @@
     $expected = 'SELECT * FROM `widget_nozzle` WHERE `primary_key` = "5"';
     Tester::check_equal("Setting: id_column_overrides, third test", $expected);
 
+    ORM::for_table('widget')->use_id_column('new_id')->find_one(5);
+    $expected = 'SELECT * FROM `widget` WHERE `new_id` = "5"';
+    Tester::check_equal("Instance ID column, first test", $expected);
+
+    ORM::for_table('widget_handle')->use_id_column('new_id')->find_one(5);
+    $expected = 'SELECT * FROM `widget_handle` WHERE `new_id` = "5"';
+    Tester::check_equal("Instance ID column, second test", $expected);
+
+    ORM::for_table('widget_nozzle')->use_id_column('new_id')->find_one(5);
+    $expected = 'SELECT * FROM `widget_nozzle` WHERE `new_id` = "5"';
+    Tester::check_equal("Instance ID column, third test", $expected);
+
     Tester::report();
 ?>