added attachment_fu and made the set of changes

[selectricity-live] / vendor / plugins / attachment_fu / test / test_helper.rb

diff --git a/vendor/plugins/attachment_fu/test/test_helper.rb b/vendor/plugins/attachment_fu/test/test_helper.rb
new file mode 100644 (file)
index 0000000..66a0b72
--- /dev/null
+++ b/vendor/plugins/attachment_fu/test/test_helper.rb
@@ -0,0 +1,142 @@
+$:.unshift(File.dirname(__FILE__) + '/../lib')
+
+ENV['RAILS_ENV'] = 'test'
+
+require 'test/unit'
+require File.expand_path(File.join(File.dirname(__FILE__), '../../../../config/environment.rb'))
+require 'breakpoint'
+require 'active_record/fixtures'
+require 'action_controller/test_process'
+
+config = YAML::load(IO.read(File.dirname(__FILE__) + '/database.yml'))
+ActiveRecord::Base.logger = Logger.new(File.dirname(__FILE__) + "/debug.log")
+
+db_adapter = ENV['DB']
+
+# no db passed, try one of these fine config-free DBs before bombing.
+db_adapter ||= 
+  begin
+    require 'rubygems'
+    require 'sqlite'
+    'sqlite'
+  rescue MissingSourceFile
+    begin
+      require 'sqlite3'
+      'sqlite3'
+    rescue MissingSourceFile
+    end
+  end
+
+if db_adapter.nil?
+  raise "No DB Adapter selected.  Pass the DB= option to pick one, or install Sqlite or Sqlite3."
+end
+
+ActiveRecord::Base.establish_connection(config[db_adapter])
+
+load(File.dirname(__FILE__) + "/schema.rb")
+
+Test::Unit::TestCase.fixture_path = File.dirname(__FILE__) + "/fixtures"
+$LOAD_PATH.unshift(Test::Unit::TestCase.fixture_path)
+
+class Test::Unit::TestCase #:nodoc:
+  include ActionController::TestProcess
+  def create_fixtures(*table_names)
+    if block_given?
+      Fixtures.create_fixtures(Test::Unit::TestCase.fixture_path, table_names) { yield }
+    else
+      Fixtures.create_fixtures(Test::Unit::TestCase.fixture_path, table_names)
+    end
+  end
+
+  def setup
+    Attachment.saves = 0
+    DbFile.transaction { [Attachment, FileAttachment, OrphanAttachment, MinimalAttachment, DbFile].each { |klass| klass.delete_all } }
+    attachment_model self.class.attachment_model
+  end
+  
+  def teardown
+    FileUtils.rm_rf File.join(File.dirname(__FILE__), 'files')
+  end
+
+  self.use_transactional_fixtures = true
+  self.use_instantiated_fixtures  = false
+
+  def self.attachment_model(klass = nil)
+    @attachment_model = klass if klass 
+    @attachment_model
+  end
+
+  def self.test_against_class(test_method, klass, subclass = false)
+    define_method("#{test_method}_on_#{:sub if subclass}class") do
+      klass = Class.new(klass) if subclass
+      attachment_model klass
+      send test_method, klass
+    end
+  end
+
+  def self.test_against_subclass(test_method, klass)
+    test_against_class test_method, klass, true
+  end
+
+  protected
+    def upload_file(options = {})
+      use_temp_file options[:filename] do |file|
+        att = attachment_model.create :uploaded_data => fixture_file_upload(file, options[:content_type] || 'image/png')
+        att.reload unless att.new_record?
+        return att
+      end
+    end
+
+    def use_temp_file(fixture_filename)
+      temp_path = File.join('/tmp', File.basename(fixture_filename))
+      FileUtils.mkdir_p File.join(fixture_path, 'tmp')
+      FileUtils.cp File.join(fixture_path, fixture_filename), File.join(fixture_path, temp_path)
+      yield temp_path
+    ensure
+      FileUtils.rm_rf File.join(fixture_path, 'tmp')
+    end
+
+    def assert_created(num = 1)
+      assert_difference attachment_model.base_class, :count, num do
+        if attachment_model.included_modules.include? DbFile
+          assert_difference DbFile, :count, num do
+            yield
+          end
+        else
+          yield
+        end
+      end
+    end
+    
+    def assert_not_created
+      assert_created(0) { yield }
+    end
+    
+    def should_reject_by_size_with(klass)
+      attachment_model klass
+      assert_not_created do
+        attachment = upload_file :filename => '/files/rails.png'
+        assert attachment.new_record?
+        assert attachment.errors.on(:size)
+        assert_nil attachment.db_file if attachment.respond_to?(:db_file)
+      end
+    end
+    
+    def assert_difference(object, method = nil, difference = 1)
+      initial_value = object.send(method)
+      yield
+      assert_equal initial_value + difference, object.send(method)
+    end
+    
+    def assert_no_difference(object, method, &block)
+      assert_difference object, method, 0, &block
+    end
+    
+    def attachment_model(klass = nil)
+      @attachment_model = klass if klass 
+      @attachment_model
+    end
+end
+
+require File.join(File.dirname(__FILE__), 'fixtures/attachment')
+require File.join(File.dirname(__FILE__), 'base_attachment_tests')
\ No newline at end of file