aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/lib/active_record/fixtures.rb
diff options
context:
space:
mode:
Diffstat (limited to 'activerecord/lib/active_record/fixtures.rb')
-rwxr-xr-xactiverecord/lib/active_record/fixtures.rb18
1 files changed, 18 insertions, 0 deletions
diff --git a/activerecord/lib/active_record/fixtures.rb b/activerecord/lib/active_record/fixtures.rb
index 248cc170ab..21d0615dfa 100755
--- a/activerecord/lib/active_record/fixtures.rb
+++ b/activerecord/lib/active_record/fixtures.rb
@@ -166,12 +166,30 @@ class Fixtures < Hash
fixtures.each { |fixture| fixture.insert_fixtures }
end
+ reset_sequences(connection, table_names) if ActiveRecord::ConnectionAdapters::PostgreSQLAdapter === connection
+
return fixtures.size > 1 ? fixtures : fixtures.first
ensure
ActiveRecord::Base.logger.level = old_logger_level
end
end
+ # Work around for PostgreSQL to have new fixtures created from id 1 and running.
+ def self.reset_sequences(connection, table_names)
+ table_names.flatten.each do |table|
+ table_class = Inflector.classify(table.to_s)
+ if Object.const_defined?(table_class)
+ pk = eval("#{table_class}::primary_key")
+ if pk == 'id'
+ connection.execute(
+ "SELECT setval('public.#{table.to_s}_id_seq', (SELECT MAX(id) FROM #{table.to_s}), true)",
+ 'Setting Sequence'
+ )
+ end
+ end
+ end
+ end
+
def initialize(connection, table_name, fixture_path, file_filter = DEFAULT_FILTER_RE)
@connection, @table_name, @fixture_path, @file_filter = connection, table_name, fixture_path, file_filter
@class_name = Inflector.classify(@table_name)