root/branches/rails2.1/test/unit/attachment_test.rb

require File.dirname(__FILE__) + '/../test_helper'

class AttachmentTest < Test::Unit::TestCase

  def setup
    setup_settings

    user = create_user('test')
    user_group = UserGroup.create!(:name => 'Test', :info => 'test')
    user_group.add_user(user)
    UserAccount.current_user = user
    @root_asset = create_root_asset
    @root_for_user = add_user_group_to_asset(user_group, @root_asset, true, true, true, true)
  end

  def test_Attachment_find_without_data
    assert_equal 0, Attachment.count
    attachment = create_attachment(@root_asset, true)

    x = Attachment.find_without(:data, attachment.id)
    Attachment.column_names.each { |c|
      if c == 'data'
        assert_raise(ActiveRecord::MissingAttributeError) { x.send(c) }
      else
        assert_equal true, x.respond_to?(c)
        assert_equal attachment.send(c).to_s, x.send(c).to_s
      end
    }

    # Just check that there is no typos in other finders
    assert [x], Attachment.find_authorized_to_read_without(:data, :all)
    assert Attachment.find_authorized_to_edit_without(:data, :all).empty?
  end

  # Testing type casting and insertion of binary data to database.
  def test_binary_column
    dir = File.dirname(__FILE__)
    filenames = []
    ['/../fixtures/test_files',
     '/../fixtures/open_document_files'
    ].each {|rel_dir|
      Dir.foreach(dir + rel_dir) { |f|
        filenames << dir + rel_dir + '/' + f
      }
    }

    data_hash = {}
    data_attribute_hash = {}

    filenames.map {|path| File.new(path, 'rb') }.each { |file|
      name = File.basename(file.path)
      next unless name == 'README' or ['.png', '.jpg', '.odt', '.pdf'].include?(File.extname(file.path).downcase)
      data = file.read
      file.close
      x = Attachment.new
      x.asset = @root_asset
      x.filename = name
      x.content_type = 'something'
      x.data = data
      #### DEBUG
      puts "\n" + 'File: ' + name
      #x.save!
      #f=File.new('tmp/%s_before' % name, 'wb')
      #f.write(data)
      #f.close
      #f=File.new('tmp/%s_after' % name, 'wb')
      #f.write(x.data)
      #f.close
      ####
      # Test that the binary data was not corrupted when it was
      # type casted or when it was saved/inserted to database or
      # when it is fetched from database.
      x.save!
      assert_equal data.size, x.data.size
      assert x.data == data
      #x.save!
      read_data = x.class.find(x.id).data
      assert_equal data.size, read_data.size
      assert read_data == data
    }
  end

end