bootstripe-rails 0.2.1 → 0.2.2

data/lib/bootstripe-rails/model_additions.rb

@@ -1,5 +1,7 @@
 module ActiveRecord
   class Base
+    DONT_TRUNCATE = [:schema_migrations]
+
     # Options:
     # :merge_lookup => true/false [defaults to true]
     def self.upsert(lookup, params, options = {})
@@ -36,6 +38,24 @@ module ActiveRecord
       order 'created_at ASC'
     end
 
+    # Truncates all tables in the database
+    #
+    # options:
+    # @param {Array} :exclude tables to exclude from the truncation
+    def self.truncate_models!(options={})
+      if adapter = ActiveRecord::Base.connection.adapter_name
+        exclude = [*options[:exclude]].collect(&:to_sym) if options[:exclude]
+        exclude = ([exclude] + DONT_TRUNCATE).flatten.compact.uniq
+
+        ActiveRecord::Base.connection.tables.each do |table_name|
+          trunc_method = adapter.match(/sqlite/i) ? "DELETE FROM" : "TRUNCATE TABLE"
+          ActiveRecord::Base.connection.execute("#{trunc_method} #{table_name}") if !exclude.include?(table_name.to_sym)
+        end
+      else
+        raise 'No ActiveRecord adapter specified!'
+      end
+    end
+
     def clone!(params = {})
       cl = self.class.new
       time = Time.current

data/lib/bootstripe-rails/version.rb

@@ -1,5 +1,5 @@
 module Bootstripe
   module Rails
-    VERSION = "0.2.1"
+    VERSION = "0.2.2"
   end
 end

data/spec/model_additions_spec.rb

@@ -18,4 +18,12 @@ describe ActiveRecord::Base do
     User.where(:name => 'Mike').first.email.should eq 'mikeypants@mike.com'
   end
 
+  it 'should truncate' do
+    User.upsert({:name => 'Mike'}, {:email => 'mikeypants@mike.com'})
+    ActiveRecord::Base.truncate_models!(:exclude => :users)
+    User.count.should eq 2
+    ActiveRecord::Base.truncate_models!
+    User.count.should eq 0
+  end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bootstripe-rails
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
   prerelease: 
 platform: ruby
 authors:
@@ -10,11 +10,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-12-19 00:00:00.000000000Z
+date: 2012-03-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70104038663440 !ruby/object:Gem::Requirement
+  requirement: &70122628959960 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -22,10 +22,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70104038663440
+  version_requirements: *70122628959960
 - !ruby/object:Gem::Dependency
   name: sqlite3
-  requirement: &70104038662740 !ruby/object:Gem::Requirement
+  requirement: &70122628957060 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -33,10 +33,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70104038662740
+  version_requirements: *70122628957060
 - !ruby/object:Gem::Dependency
   name: rails
-  requirement: &70104038651480 !ruby/object:Gem::Requirement
+  requirement: &70122628955680 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -44,7 +44,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70104038651480
+  version_requirements: *70122628955680
 description: A handful of useful additions for Rails
 email:
 - mike@bytebin.com