diff options
author | Gannon McGibbon <gannon.mcgibbon@gmail.com> | 2018-10-05 14:28:15 -0400 |
---|---|---|
committer | Gannon McGibbon <gannon.mcgibbon@gmail.com> | 2018-10-05 17:57:55 -0400 |
commit | ee5ab2228be88caa9a1dcdd8841488847cb43c07 (patch) | |
tree | 81d290a0ce663d497f49d77b085b146c4ce5d10f /activerecord/lib/active_record/fixture_set | |
parent | ea4b52ab3cd375d314b0d9d74724cdb49b4fcdd4 (diff) | |
download | rails-ee5ab2228be88caa9a1dcdd8841488847cb43c07.tar.gz rails-ee5ab2228be88caa9a1dcdd8841488847cb43c07.tar.bz2 rails-ee5ab2228be88caa9a1dcdd8841488847cb43c07.zip |
Introduce FixtureSet::ModelMetadata
Diffstat (limited to 'activerecord/lib/active_record/fixture_set')
-rw-r--r-- | activerecord/lib/active_record/fixture_set/model_metadata.rb | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/activerecord/lib/active_record/fixture_set/model_metadata.rb b/activerecord/lib/active_record/fixture_set/model_metadata.rb new file mode 100644 index 0000000000..edc03939c8 --- /dev/null +++ b/activerecord/lib/active_record/fixture_set/model_metadata.rb @@ -0,0 +1,40 @@ +# frozen_string_literal: true + +module ActiveRecord + class FixtureSet + class ModelMetadata # :nodoc: + def initialize(model_class, table_name) + @model_class = model_class + @table_name = table_name + end + + def primary_key_name + @primary_key_name ||= @model_class && @model_class.primary_key + end + + def primary_key_type + @primary_key_type ||= @model_class && @model_class.type_for_attribute(@model_class.primary_key).type + end + + def has_primary_key_column? + @has_primary_key_column ||= primary_key_name && + @model_class.columns.any? { |col| col.name == primary_key_name } + end + + def timestamp_column_names + @timestamp_column_names ||= + %w(created_at created_on updated_at updated_on) & column_names + end + + def inheritance_column_name + @inheritance_column_name ||= @model_class && @model_class.inheritance_column + end + + private + + def column_names + @column_names ||= @model_class.connection.columns(@table_name).collect(&:name) + end + end + end +end |