Consolidate tests for time and datetime columns options, limit and precision

This commit is contained in:
Yasuo Honda 2015-11-30 16:58:58 +00:00
parent 589cef086f
commit 87ee4f4149
2 changed files with 12 additions and 59 deletions

@ -10,6 +10,7 @@ class Foo < ActiveRecord::Base; end
setup do
@connection = ActiveRecord::Base.connection
Foo.reset_column_information
end
teardown do
@ -20,24 +21,24 @@ def test_datetime_data_type_with_precision
@connection.create_table(:foos, force: true)
@connection.add_column :foos, :created_at, :datetime, precision: 0
@connection.add_column :foos, :updated_at, :datetime, precision: 5
assert_equal 0, activerecord_column_option('foos', 'created_at', 'precision')
assert_equal 5, activerecord_column_option('foos', 'updated_at', 'precision')
assert_equal 0, Foo.columns_hash['created_at'].precision
assert_equal 5, Foo.columns_hash['updated_at'].precision
end
def test_timestamps_helper_with_custom_precision
@connection.create_table(:foos, force: true) do |t|
t.timestamps precision: 4
end
assert_equal 4, activerecord_column_option('foos', 'created_at', 'precision')
assert_equal 4, activerecord_column_option('foos', 'updated_at', 'precision')
assert_equal 4, Foo.columns_hash['created_at'].precision
assert_equal 4, Foo.columns_hash['updated_at'].precision
end
def test_passing_precision_to_datetime_does_not_set_limit
@connection.create_table(:foos, force: true) do |t|
t.timestamps precision: 4
end
assert_nil activerecord_column_option('foos', 'created_at', 'limit')
assert_nil activerecord_column_option('foos', 'updated_at', 'limit')
assert_nil Foo.columns_hash['created_at'].limit
assert_nil Foo.columns_hash['updated_at'].limit
end
def test_invalid_datetime_precision_raises_error
@ -48,14 +49,6 @@ def test_invalid_datetime_precision_raises_error
end
end
def test_database_agrees_with_activerecord_about_precision
@connection.create_table(:foos, force: true) do |t|
t.timestamps precision: 4
end
assert_equal 4, database_datetime_precision('foos', 'created_at')
assert_equal 4, database_datetime_precision('foos', 'updated_at')
end
def test_formatting_datetime_according_to_precision
@connection.create_table(:foos, force: true) do |t|
t.datetime :created_at, precision: 0
@ -91,21 +84,5 @@ def test_datetime_precision_with_zero_should_be_dumped
end
end
private
def database_datetime_precision(table_name, column_name)
results = @connection.exec_query("SELECT column_name, datetime_precision FROM information_schema.columns WHERE table_name = '#{table_name}'")
result = results.find do |result_hash|
result_hash["column_name"] == column_name
end
result && result["datetime_precision"].to_i
end
def activerecord_column_option(tablename, column_name, option)
result = @connection.columns(tablename).find do |column|
column.name == column_name
end
result && result.send(option)
end
end
end

@ -10,6 +10,7 @@ class Foo < ActiveRecord::Base; end
setup do
@connection = ActiveRecord::Base.connection
Foo.reset_column_information
end
teardown do
@ -20,8 +21,8 @@ def test_time_data_type_with_precision
@connection.create_table(:foos, force: true)
@connection.add_column :foos, :start, :time, precision: 3
@connection.add_column :foos, :finish, :time, precision: 6
assert_equal 3, activerecord_column_option('foos', 'start', 'precision')
assert_equal 6, activerecord_column_option('foos', 'finish', 'precision')
assert_equal 3, Foo.columns_hash['start'].precision
assert_equal 6, Foo.columns_hash['finish'].precision
end
def test_passing_precision_to_time_does_not_set_limit
@ -29,8 +30,8 @@ def test_passing_precision_to_time_does_not_set_limit
t.time :start, precision: 3
t.time :finish, precision: 6
end
assert_nil activerecord_column_option('foos', 'start', 'limit')
assert_nil activerecord_column_option('foos', 'finish', 'limit')
assert_nil Foo.columns_hash['start'].limit
assert_nil Foo.columns_hash['finish'].limit
end
def test_invalid_time_precision_raises_error
@ -42,15 +43,6 @@ def test_invalid_time_precision_raises_error
end
end
def test_database_agrees_with_activerecord_about_precision
@connection.create_table(:foos, force: true) do |t|
t.time :start, precision: 2
t.time :finish, precision: 4
end
assert_equal 2, database_datetime_precision('foos', 'start')
assert_equal 4, database_datetime_precision('foos', 'finish')
end
def test_formatting_time_according_to_precision
@connection.create_table(:foos, force: true) do |t|
t.time :start, precision: 0
@ -88,21 +80,5 @@ def test_time_precision_with_zero_should_be_dumped
end
end
private
def database_datetime_precision(table_name, column_name)
results = @connection.exec_query("SELECT column_name, datetime_precision FROM information_schema.columns WHERE table_name = '#{table_name}'")
result = results.find do |result_hash|
result_hash["column_name"] == column_name
end
result && result["datetime_precision"].to_i
end
def activerecord_column_option(tablename, column_name, option)
result = @connection.columns(tablename).find do |column|
column.name == column_name
end
result && result.send(option)
end
end
end