use AR.sequence_name instead of sql

razum2um · May 28, 2014 · a00a8d4 · a00a8d4
1 parent 2bf53e4
commit a00a8d4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/templates/generate_stuff.rb b/templates/generate_stuff.rb
@@ -310,8 +310,8 @@ class Api::V2::ReposController < Api::V1::ReposController; end
     class ActionDispatch::IntegrationTest
       def setup
         super if defined? super
-        %w[repos_id_seq users_id_seq].each do |id|
-          ActiveRecord::Base.connection.execute "ALTER SEQUENCE \#{id} RESTART WITH 1"
+        [User, Repo].each do |klass|
+          ActiveRecord::Base.connection.execute "ALTER SEQUENCE \#{klass.sequence_name} RESTART WITH 1"
         end
         DatabaseCleaner.start
       end
@@ -354,8 +354,8 @@ def teardown
       ]
 
       c.before do
-        %w[repos_id_seq users_id_seq].each do |id|
-          ActiveRecord::Base.connection.execute "ALTER SEQUENCE \#{id} RESTART WITH 1"
+        [User, Repo].each do |klass|
+          ActiveRecord::Base.connection.execute "ALTER SEQUENCE \#{klass.sequence_name} RESTART WITH 1"
         end
         DatabaseCleaner.start
       end