0
@@ -12,17 +12,17 @@ ActiveRecord::Base.establish_connection(:adapter => 'mysql',
0
                                         :database => MYSQL_DB
0
                                         )
0
 
0
-describe 'Mysql Adapter' do
0
+class MysqlAdapterTest < Test::Unit::TestCase
0
 
0
-  before do
0
+  def setup
0
     TestDB.up
0
   end
0
   
0
-  after do
0
+  def teardown
0
     TestDB.down
0
   end
0
 
0
-  it 'should add a foreign key' do
0
+  def test_should_add_a_foreign_key
0
     db = ActiveRecord::Base.connection
0
     db.add_foreign_key(:on => {:table => :widgets, :column => :factory_id}, 
0
                        :references => {:table => :factories, :column => :id})
0
@@ -32,10 +32,9 @@ describe 'Mysql Adapter' do
0
                   "WHERE constraint_schema = 'test' AND " +
0
                   "constraint_name = 'fk_widgets_factory_id_factories_id'")[0]
0
 
0
-    key['TABLE_NAME'].should.equal 'widgets'
0
-    key['COLUMN_NAME'].should.equal 'factory_id'
0
-    key['REFERENCED_TABLE_NAME'].should.equal 'factories'
0
-    key['REFERENCED_COLUMN_NAME'].should.equal 'id'
0
+    assert_equal key['TABLE_NAME'], 'widgets'
0
+    assert_equal key['COLUMN_NAME'], 'factory_id'
0
+    assert_equal key['REFERENCED_TABLE_NAME'], 'factories'
0
+    assert_equal key['REFERENCED_COLUMN_NAME'], 'id'
0
   end
0
-
0
 end

0
@@ -1,5 +1,5 @@
0
 require File.join(File.dirname(__FILE__), '../lib/foreign_key_migrations')
0
-require 'bacon'
0
+require 'test/unit'
0
 require 'mocha'
0
 
0
 ActiveRecord::Migration.verbose = false

0
@@ -49,7 +49,17 @@ class AddDropForeignKeyTest < Test::Unit::TestCase
0
     assert(@adapter.add_foreign_key(
0
       :on => 'employees', :references => 'departments'))
0
   end
0
+
0
+  def test_add_foreign_key_data_with_table_names_only_should_work_with_symbols
0
+    @adapter.expects(:execute).once \
0
+      .with("ALTER TABLE employees ADD CONSTRAINT " +
0
+            "fk_employees_department_id_departments_id FOREIGN KEY (department_id) " +
0
+            "REFERENCES departments(id)").returns(true)
0
     
0
+    assert(@adapter.add_foreign_key(:on => :employees, :references => :departments))
0
+  end
0
+
0
+  
0
   def test_delete_foreign_key_should_work_with_full_data
0
     @adapter.expects(:execute).once.with(
0
       "ALTER TABLE employees DROP FOREIGN KEY " +