test/test_rhubarb.rb in spqr-0.1.0 vs test/test_rhubarb.rb in spqr-0.1.1

@@ -9,20 +9,34 @@
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 # 
 #     http://www.apache.org/licenses/LICENSE-2.0
 
-require 'rubygems'
 require 'rhubarb/rhubarb'
 require 'test/unit'
 
 class TestClass 
   include Rhubarb::Persisting  
   declare_column :foo, :integer
   declare_column :bar, :string
 end
 
+module RhubarbNamespace
+  class NMTC
+    include Rhubarb::Persisting  
+    declare_column :foo, :integer
+    declare_column :bar, :string
+  end
+
+  class NMTC2
+    include Rhubarb::Persisting  
+    declare_column :foo, :integer
+    declare_column :bar, :string
+    declare_table_name('namespacetestcase')
+  end
+end
+
 class TestClass2 
   include Rhubarb::Persisting  
   declare_column :fred, :integer
   declare_column :barney, :string
   declare_index_on :fred
@@ -85,10 +99,12 @@
     klasses << SelfRef
     klasses << CustomQueryTable
     klasses << ToRef
     klasses << FromRef
     klasses << FreshTestTable
+    klasses << RhubarbNamespace::NMTC
+    klasses << RhubarbNamespace::NMTC2
 
     klasses.each { |klass| klass.create_table }
 
     @flist = []
   end
@@ -106,9 +122,17 @@
     r = Rhubarb::Reference.new(TestClass)
     assert(r.referent == TestClass, "Referent of managed reference instance incorrect")
     assert(r.column == "row_id", "Column of managed reference instance incorrect")
     assert(r.to_s == "references TestClass(row_id)", "string representation of managed reference instance incorrect")
     assert(r.managed_ref?, "managed reference should return true for managed_ref?")
+  end
+
+  def test_namespace_table_name
+    assert_equal "nmtc", RhubarbNamespace::NMTC.table_name
+  end
+
+  def test_set_table_name
+    assert_equal "namespacetestcase", RhubarbNamespace::NMTC2.table_name
   end
 
   def test_reference_ctor_string
     r = Rhubarb::Reference.new("TestClass")
     assert(r.referent == "TestClass", "Referent of string-backed reference instance incorrect")