file_blobs_rails 0.1.0

data/lib/file_blobs_rails/blob_model.rb

@@ -0,0 +1,105 @@
+require 'base64'
+require 'digest/sha2'
+
+require 'active_model'
+require 'active_support'
+
+module FileBlobs
+
+# Included in the model that stores file data.
+module BlobModel
+  extend ActiveSupport::Concern
+
+  included do
+    # A cryptographic hash over the file's data.
+    #
+    # This will be an URL-safe string.
+    validates :id, presence: true, length: 1..48, uniqueness: true
+
+    # The raw data in the file.
+    validates :data, presence: true
+    validates_each :data do |record, attr, value|
+      if value && value.bytesize > 128.kilobytes
+        record.errors.add attr, 'exceeds 128 kilobytes'.freeze
+      end
+    end
+  end
+
+  # Class methods on models that include FileBlobsRails::BlobModel.
+  module ClassMethods
+    # The URL-safe base64-encoded SHA-256 of the given data.
+    #
+    # @param [String] blob_contents the file data to be hashed
+    # @return [String] a cryptographically strong hash of the given data
+    def id_for(blob_contents)
+      # This needs to be kept in sync with
+      # active_record_fixture_set_extensions.rb.
+      Base64.urlsafe_encode64 Digest::SHA256.digest(blob_contents)
+    end
+
+    # Declares the model classes that use `has_file_blob`.
+    #
+    # This list is necessary for garbage collection to work correctly. If the
+    # class list is incomplete, a blob model may be garbage-collected
+    # prematurely.
+    #
+    # @param [Enumerable<String>] class_names the names of the model classes
+    #     that include the HasFileBlob concern, and therefore point to the
+    #     file blob model (the class that includes FileBlobsRails::BlobModel)
+    def blob_owner_class_names!(*class_names)
+      # We're using map to create a new array out of the class_list parameter,
+      # which can be any iterable.
+      #
+      # We're not particularly concerned with speed, because this is executed
+      # once in production, while the application boots. Also, the class_names
+      # array is expected to be small.
+      @_blob_owner_class_names ||=
+          class_names.map { |class_name| class_name.to_s }.freeze
+    end
+
+    # The classes that include FileBlobsRails::HasFileBlob.
+    #
+    # @return [Array<Class>] the model classes
+    def blob_owner_classes
+      @_blob_owner_classes ||= @_blob_owner_class_names.map do |class_name|
+        # Trigger ActiveSupport's autoloading behavior.
+        #
+        # ActiveSupport's String#constantize is pretty slow, but we're not
+        # particularly concerned with speed, because this is executed once in
+        # production, while the application boots. Also, the list of class
+        # names is expected to be small.
+        class_name.constantize
+      end.freeze
+    end
+  end
+
+  # Garbage-collects this blob if it is not referenced by any other model.
+  #
+  # @return [Boolean] true if the blob was garbage-collected
+  def maybe_garbage_collect
+    self.class.transaction do
+      if eligible_for_garbage_collection?
+        destroy
+        true
+      else
+        false
+      end
+    end
+  end
+
+  # Checks if this blob can be garbage-collected.
+  #
+  # This check's result can become invalid after another Blob-owning model is
+  # created. To prevent data races, the check and its corresponding garbage
+  # collection must be done in the same database transaction.
+  #
+  # @return [Boolean] true if this blob is not referenced by any Blob-owning
+  #     model, and thus is eligible for garbage collection
+  def eligible_for_garbage_collection?
+    self.class.blob_owner_classes.all? do |klass|
+      klass.file_blob_eligible_for_garbage_collection? self
+    end
+  end
+end  # module FileBlobs::BlobModel
+
+end  # namespace FileBlobs

data/lib/file_blobs_rails/engine.rb

@@ -0,0 +1,10 @@
+module FileBlobs
+
+class Engine < Rails::Engine
+  generators do
+    require_relative 'generators/blob_model_generator.rb'
+    require_relative 'generators/blob_owner_generator.rb'
+  end
+end  # class FileBlobs::Engine
+
+end  # namespace FileBlobs

data/lib/file_blobs_rails/file_blob_proxy.rb

@@ -0,0 +1,7 @@
+module FileBlobs
+
+# Base class for the proxies generated by has_file_blob.
+class FileBlobProxy
+end  # module FileBlobs::FileBlobProxy
+
+end  # namespace FileBlobs

data/lib/file_blobs_rails/generators/blob_model_generator.rb

@@ -0,0 +1,25 @@
+module FileBlobs
+
+class BlobModelGenerator < Rails::Generators::NamedBase
+  source_root File.expand_path('../templates', __FILE__)
+
+  check_class_collision
+
+  def create_file_blob_model
+    template 'file_blob.rb.erb', File.join('app', 'models', "#{file_name}.rb")
+    template 'file_blob_test.rb.erb',
+             File.join('test', 'models', "#{file_name}_test.rb")
+    template '001_create_file_blobs.rb.erb',
+        File.join('db', 'migrate',
+                  "20161029000001_create_#{file_name.tableize}.rb")
+    template 'file_blobs.yml.erb',
+             File.join('test', 'fixtures', "#{file_name.tableize}.yml")
+
+    copy_file File.join('files', 'invoice.pdf'),
+              File.join('test', 'fixtures', 'files', 'invoice.pdf')
+    copy_file File.join('files', 'ruby.png'),
+              File.join('test', 'fixtures', 'files', 'ruby.png')
+  end
+end  # class FileBlobs::BlobModelGenerator
+
+end  # namespace FileBlobs

data/lib/file_blobs_rails/generators/blob_owner_generator.rb

@@ -0,0 +1,28 @@
+module FileBlobs
+
+class BlobOwnerGenerator < Rails::Generators::NamedBase
+  source_root File.expand_path('../templates', __FILE__)
+
+  check_class_collision
+
+  class_option :attr_name, type: :string, default: 'file',
+      desc: 'The name of the attribute pointing to the blob model'
+  class_option :blob_model, type: :string, default: 'FileBlob',
+      desc: 'The name of the model class that stores the file contents'
+  class_option :allow_nil, type: :boolean, default: false,
+      desc: "Support models that don't store files"
+
+  def create_file_blob_model
+    # Set up the template environment.
+    template 'blob_owner.rb.erb', File.join('app', 'models', "#{file_name}.rb")
+    template 'blob_owner_test.rb.erb',
+             File.join('test', 'models', "#{file_name}_test.rb")
+    template '002_create_blob_owners.rb.erb',
+        File.join('db', 'migrate',
+                  "20161029000002_create_#{file_name.tableize}.rb")
+    template 'blob_owners.yml.erb',
+             File.join('test', 'fixtures', "#{file_name.tableize}.yml")
+  end
+end  # class FileBlobs::BlobModelGenerator
+
+end  # namespace FileBlobs

data/lib/file_blobs_rails/generators/templates/001_create_file_blobs.rb.erb

@@ -0,0 +1,7 @@
+class Create<%= file_name.classify.pluralize %> < ActiveRecord::Migration[5.0]
+  def change
+    create_file_blobs_table :<%= file_name.tableize %>, blob_limit: 1.megabyte do |t|
+      # Build any custom table structure here.
+    end
+  end
+end

data/lib/file_blobs_rails/generators/templates/002_create_blob_owners.rb.erb

@@ -0,0 +1,7 @@
+class Create<%= file_name.classify.pluralize %> < ActiveRecord::Migration[5.0]
+  def change
+    create_table :<%= file_name.tableize %> do |t|
+      t.file_blob :<%= options[:attr_name] %>, null: <%= options[:allow_nil] %>, mime_type_limit: 64, file_name_limit: 256
+    end
+  end
+end

data/lib/file_blobs_rails/generators/templates/blob_owner.rb.erb

@@ -0,0 +1,3 @@
+class <%= file_name.classify %> < ActiveRecord::Base
+  has_file_blob :<%= options[:attr_name] %>, allows_nil: <%= options[:allow_nil] %>, blob_model: <%= options[:blob_model].classify.inspect %>
+end

data/lib/file_blobs_rails/generators/templates/blob_owner_test.rb.erb

@@ -0,0 +1,13 @@
+require 'test_helper'
+
+class <%= file_name.classify %>Test < ActiveSupport::TestCase
+  test '<%= options[:attr_name] %> is wired correctly' do
+    <%= file_name %> = <%= file_name.tableize %>(:ruby)
+
+    assert_equal 'image/png', <%= file_name %>.<%= options[:attr_name] %>.mime_type
+    assert_equal 'ruby.png', <%= file_name %>.<%= options[:attr_name] %>.original_name
+    assert_equal file_blob_id('files/ruby.png'), <%= file_name %>.<%= options[:attr_name] %>.blob.id
+    assert_equal file_blob_size('files/ruby.png'), <%= file_name %>.<%= options[:attr_name] %>.size
+    assert_equal file_blob_data('files/ruby.png'), <%= file_name %>.<%= options[:attr_name] %>.data
+  end
+end

data/lib/file_blobs_rails/generators/templates/blob_owners.yml.erb

@@ -0,0 +1,11 @@
+ruby:
+  <%= options[:attr_name] %>_blob_id: <%%= file_blob_id 'files/ruby.png' %>
+  <%= options[:attr_name] %>_size: <%%= file_blob_size 'files/ruby.png' %>
+  <%= options[:attr_name] %>_mime_type: image/png
+  <%= options[:attr_name] %>_original_name: ruby.png
+
+invoice:
+  <%= options[:attr_name] %>_blob_id: <%%= file_blob_id 'files/invoice.pdf' %>
+  <%= options[:attr_name] %>_size: <%%= file_blob_size 'files/invoice.pdf' %>
+  <%= options[:attr_name] %>_mime_type: application/pdf
+  <%= options[:attr_name] %>_original_name: invoice.pdf

data/lib/file_blobs_rails/generators/templates/file_blob.rb.erb

@@ -0,0 +1,11 @@
+# Stores the contents of database-backed files for this application.
+class <%= file_name.classify %> < ActiveRecord::Base
+  include FileBlobs::BlobModel
+
+  # The list below must include all the models that that use has_file_blob and
+  # store data using this model class. Omitting a module will lead to data
+  # loss, as content blobs will get garbage-collected prematurely.
+  blob_owner_class_names! 'BlobOwner'
+
+  # Place any extensions to the file contents blob class below.
+end

data/lib/file_blobs_rails/generators/templates/file_blob_test.rb.erb

@@ -0,0 +1,9 @@
+require 'test_helper'
+
+class <%= file_name.classify %>Test < ActiveSupport::TestCase
+  test 'blob_owner_classes all use has_file_blob' do
+    <%= file_name.camelize %>.blob_owner_classes.each do |klass|
+      assert klass.respond_to?(:file_blob_eligible_for_garbage_collection?), klass.name
+    end
+  end
+end

data/lib/file_blobs_rails/generators/templates/file_blobs.yml.erb

@@ -0,0 +1,7 @@
+ruby_png:
+  id: <%%= file_blob_id 'files/ruby.png' %>
+  data: <%%= file_blob_data 'files/ruby.png' %>
+
+invoice_pdf:
+  id: <%%= file_blob_id 'files/invoice.pdf' %>
+  data: <%%= file_blob_data 'files/invoice.pdf' %>

data/lib/file_blobs_rails/generators/templates/files/invoice.pdf

@@ -0,0 +1,137 @@
+%PDF-1.4
+%����
+5 0 obj
+<<
+/Type /Font
+/Subtype /TrueType
+/BaseFont /Arial
+/Encoding /WinAnsiEncoding
+>>
+endobj
+7 0 obj
+<<
+/Type /FontDescriptor
+/FontName /BookAntiqua,Bold
+/Flags 16418
+/FontBBox [-250 -260 1236 930]
+/MissingWidth 750
+/StemV 146
+/StemH 146
+/ItalicAngle 0
+/CapHeight 930
+/XHeight 651
+/Ascent 930
+/Descent 260
+/Leading 210
+/MaxWidth 1030
+/AvgWidth 460
+>>
+endobj
+6 0 obj
+<<
+/Type /Font
+/Subtype /TrueType
+/BaseFont /BookAntiqua,Bold
+/FirstChar 31
+/LastChar 255
+/Widths [750 250 278 402 606 500 889 833 227 333 333 444 606 250 333 250 296 500 500 500 500 500 500 500 500 500 500 250 250 606 606 606 444 747 778 667 722 833 611 556 833 833 389 389 778 611 1000 833 833 611 833 722 611 667 778 778 1000 667 667 667 333 606 333 606 500 333 500 611 444 611 500 389 556 611 333 333 611 333 889 611 556 611 611 389 444 333 611 556 833 500 556 500 310 606 310 606 750 500 750 333 500 500 1000 500 500 333 1000 611 389 1000 750 750 750 750 278 278 500 500 606 500 1000 333 998 444 389 833 750 750 667 250 278 500 500 606 500 606 500 333 747 438 500 606 333 747 500 400 549 361 361 333 576 641 250 333 361 488 500 889 890 889 444 778 778 778 778 778 778 1000 722 611 611 611 611 389 389 389 389 833 833 833 833 833 833 833 606 833 778 778 778 778 667 611 611 500 500 500 500 500 500 778 444 500 500 500 500 333 333 333 333 556 611 556 556 556 556 556 549 556 611 611 611 611 556 611 556]
+/Encoding /WinAnsiEncoding
+/FontDescriptor 7 0 R
+>>
+endobj
+8 0 obj
+[/PDF /Text]
+endobj
+10 0 obj
+246
+endobj
+9 0 obj
+<<
+/Length 246
+/Filter /FlateDecode
+>>
+stream
+H�͐�J�0�� ��{��f�$M��n�-���[&je���ۤ�~�$���}�Ʌ�Ij���s����~�X�-],��$Y���)�'N�u�1!���V�?��?
+�b1Rbb�҉�H�[��TD:#�&ح��X���i�$qnf�����]������a��{��أ���q|J�Ls]�Q�I��j�%��9��`�঺��U�ite�z�$����OeB�Ēү�R��@zܗ���g���<���
+endstream
+endobj
+4 0 obj
+<<
+/Type /Page
+/MediaBox [0 0 612 792]
+/Resources <<
+/Font <<
+/F0 5 0 R
+/F1 6 0 R
+>>
+/ProcSet 8 0 R
+>>
+/Contents 9 0 R
+/Parent 2 0 R
+>>
+endobj
+11 0 obj
+[/CalGray <<
+/WhitePoint [0.9505 1 1.0891]
+/Gamma 0.2468
+>>]
+endobj
+12 0 obj
+[/CalRGB <<
+/WhitePoint [0.9505 1 1.0891]
+/Gamma [0.2468 0.2468 0.2468]
+/Matrix [0.4361 0.2225 0.0139 0.3851 0.7169 0.0971 0.1431 0.0606 0.7141]
+>>]
+endobj
+2 0 obj
+<<
+/Type /Pages
+/Kids [4 0 R]
+/Count 1
+>>
+endobj
+1 0 obj
+<<
+/Type /Catalog
+/Pages 2 0 R
+/DefaultGray 11 0 R
+/DefaultRGB 12 0 R
+>>
+endobj
+3 0 obj
+<<
+/Creator (1725.fm)
+/CreationDate (1-Jan-3 18:15PM)
+/Title (1725.PDF)
+/Author (Unknown)
+/Keywords <>
+/Subject <>
+/ModDate (D:20131124155828Z)
+/Producer (3-Heights\(TM\) PDF Producer 4.2.26.0 \(http://www.pdf-tools.com\))
+>>
+endobj
+xref
+0 13
+0000000000 65535 f
+0000002266 00000 n
+0000002209 00000 n
+0000002354 00000 n
+0000001816 00000 n
+0000000015 00000 n
+0000000376 00000 n
+0000000111 00000 n
+0000001450 00000 n
+0000001498 00000 n
+0000001478 00000 n
+0000001967 00000 n
+0000002044 00000 n
+trailer
+<<
+/Size 13
+/Root 1 0 R
+/Info 3 0 R
+/ID [<F2B22CD0880D25E8E743A676B47199CD> <5B1A04D8F6660EB95B25A6F7A746258B>]
+>>
+startxref
+2595
+%%EOF

data/lib/file_blobs_rails.rb

@@ -0,0 +1,21 @@
+require 'active_support/dependencies'
+
+module FileBlobs
+  extend ActiveSupport::Autoload
+
+  autoload :BlobModel, 'file_blobs_rails/blob_model.rb'
+  autoload :FileBlobProxy, 'file_blobs_rails/file_blob_proxy.rb'
+end  # module FileBlobs
+
+require_relative(
+    'file_blobs_rails/action_controller_data_streaming_extensions.rb')
+require_relative 'file_blobs_rails/active_record_extensions.rb'
+require_relative 'file_blobs_rails/active_record_fixture_set_extensions.rb'
+require_relative 'file_blobs_rails/active_record_migration_extensions.rb'
+require_relative(
+    'file_blobs_rails/active_record_table_definition_extensions.rb')
+require_relative 'file_blobs_rails/active_support_test_extensions.rb'
+
+if defined?(Rails)
+  require_relative 'file_blobs_rails/engine.rb'
+end

data/test/blob_model_test.rb

@@ -0,0 +1,23 @@
+require_relative 'test_helper'
+
+class BlobModelTest < ActiveSupport::TestCase
+  setup do
+    blob_data = 'Hello blob world!'
+    @blob = GemTestBlob.new data: blob_data,
+                            id: GemTestBlob.id_for('blob_data')
+  end
+
+  test 'setup' do
+    assert @blob.valid?, @blob.errors
+  end
+
+  test 'requires an id' do
+    @blob.id = nil
+    assert !@blob.valid?
+  end
+
+  test 'requires data' do
+    @blob.data = nil
+    assert !@blob.valid?
+  end
+end

data/test/blob_owner_test.rb

@@ -0,0 +1,9 @@
+require_relative 'test_helper'
+
+# Run the tests in the generator, to make sure they pass.
+template_path = File.expand_path(
+    '../../lib/file_blobs_rails/generators/templates/blob_owner_test.rb.erb',
+    __FILE__)
+eval Erubis::Eruby.new(File.read(template_path)).result(
+    options: { attr_name: 'file', allow_nil: false, blob_model: 'FileBlob' },
+    file_name: 'blob_owner'), nil, template_path

data/test/controller_extensions_test.rb

@@ -0,0 +1,80 @@
+require_relative 'test_helper'
+
+class FileBlobGemTestController < ActionController::Base
+  before_action :set_blob_owner, only: [:show]
+
+  # POST /file_blob_gem_test
+  def create
+    blob_owner = BlobOwner.create! blob_owner_params
+    render json: blob_owner
+  end
+
+  # GET /file_blob_gem_test
+  def show
+    send_file_blob @blob_owner.file
+  end
+
+  # Use callbacks to share common setup or constraints between actions.
+  def set_blob_owner
+    @blob_owner = BlobOwner.find params[:id]
+  end
+  private :set_blob_owner
+
+  # Only allow a trusted parameter "white list" through.
+  def blob_owner_params
+    params.require(:blob_owner).permit(:file)
+  end
+  private :blob_owner_params
+end
+
+class ControllerExtensionsTest < ActionController::TestCase
+  tests FileBlobGemTestController
+
+  # Workaround the issue that is well-documented but incorrectly addressed in
+  # https://github.com/rails/rails/pull/25796
+  # include ActionDispatch::TestProcess
+
+  test 'can read files from ActionDispatch::Http::UploadedFile' do
+    BlobOwner.destroy_all
+    FileBlob.destroy_all
+
+    post :create, params: { blob_owner: {
+        file: fixture_file_upload('files/ruby.png', 'image/png', :binary) } }
+
+    assert_response :success
+
+    assert_equal 1, BlobOwner.count
+
+    blob_owner = BlobOwner.last
+    assert_equal 'image/png', blob_owner.file.mime_type
+    assert_equal 'ruby.png', blob_owner.file.original_name
+    assert_equal file_blob_size('files/ruby.png'), blob_owner.file.size
+    assert_equal file_blob_data('files/ruby.png'), blob_owner.file.data
+  end
+
+  test '#send_file_blob generates a fresh response correctly' do
+    get :show, params: { id: blob_owners(:ruby).id }
+
+    assert_response :success
+    assert_equal 'image/png', response.headers['Content-Type']
+    assert_equal file_blob_id('files/ruby.png'), response.headers['ETag']
+    assert_equal file_blob_data('files/ruby.png'), response.body
+  end
+
+  test '#send_file_blob sends 200 for requests with non-matching e-tags' do
+    request.set_header 'HTTP_IF_NONE_MATCH', '1234'
+    get :show, params: { id: blob_owners(:ruby).id }
+
+    assert_response :success
+    assert_equal 'image/png', response.headers['Content-Type']
+    assert_equal file_blob_id('files/ruby.png'), response.headers['ETag']
+    assert_equal file_blob_data('files/ruby.png'), response.body
+  end
+
+  test '#send_file_blob sends 304 for requests with matching e-tags' do
+    request.set_header 'HTTP_IF_NONE_MATCH', file_blob_id('files/ruby.png')
+    get :show, params: { id: blob_owners(:ruby).id }
+
+    assert_response :not_modified
+  end
+end

data/test/file_blob_proxy_test.rb

@@ -0,0 +1,100 @@
+require_relative 'test_helper'
+
+class FileBlobProxyTest < ActiveSupport::TestCase
+  setup do
+    @blob_owner = blob_owners(:ruby)
+    @message = GemTestMessage.new
+  end
+
+  test '#blob_class' do
+    assert_equal FileBlob, @blob_owner.file.blob_class
+    assert_equal GemTestBlob, @message.attachment.blob_class
+  end
+
+  test '#allows_nil?' do
+    assert_equal false, @blob_owner.file.allows_nil?
+    assert_equal true, @message.attachment.allows_nil?
+  end
+
+  test '#owner' do
+    assert_equal @blob_owner, @blob_owner.file.owner
+    assert_equal @message, @message.attachment.owner
+  end
+
+  test 'proxy getter on the owner model' do
+    proxy = @blob_owner.file
+
+    assert_kind_of FileBlobs::FileBlobProxy, proxy
+  end
+
+  test 'getter proxies' do
+    proxy = @blob_owner.file
+
+    assert_equal 'image/png', proxy.mime_type
+    assert_equal 'ruby.png', proxy.original_name
+    assert_equal file_blob_id('files/ruby.png'), proxy.blob_id
+    assert_equal file_blob_size('files/ruby.png'), proxy.size
+    assert_equal file_blob_data('files/ruby.png'), proxy.data
+    assert_equal file_blobs(:ruby_png), proxy.blob
+  end
+
+  test '#mime_type=' do
+    proxy = @blob_owner.file
+    proxy.mime_type = 'x-test/x-setter'
+
+    assert_equal 'x-test/x-setter', proxy.mime_type
+    assert_equal 'ruby.png', proxy.original_name
+    assert_equal file_blob_id('files/ruby.png'), proxy.blob_id
+    assert_equal file_blob_size('files/ruby.png'), proxy.size
+    assert_equal file_blob_data('files/ruby.png'), proxy.data
+    assert_equal file_blobs(:ruby_png), proxy.blob
+  end
+
+  test '#original_name=' do
+    proxy = @blob_owner.file
+    proxy.original_name = 'setter-test.png'
+
+    assert_equal 'image/png', proxy.mime_type
+    assert_equal 'setter-test.png', proxy.original_name
+    assert_equal file_blob_id('files/ruby.png'), proxy.blob_id
+    assert_equal file_blob_size('files/ruby.png'), proxy.size
+    assert_equal file_blob_data('files/ruby.png'), proxy.data
+    assert_equal file_blobs(:ruby_png), proxy.blob
+  end
+
+  test '#data=' do
+    proxy = @blob_owner.file
+    proxy.data = file_blob_data('files/invoice.pdf')
+
+    assert_equal 'image/png', proxy.mime_type
+    assert_equal 'ruby.png', proxy.original_name
+    assert_equal file_blob_id('files/invoice.pdf'), proxy.blob_id
+    assert_equal file_blob_size('files/invoice.pdf'), proxy.size
+    assert_equal file_blob_data('files/invoice.pdf'), proxy.data
+    assert_equal file_blobs(:invoice_pdf), proxy.blob
+  end
+
+  test 'proxy setter on the owner model with same-table blobs' do
+    new_blob_owner = BlobOwner.new file: @blob_owner.file
+
+    assert_equal 'ruby.png', new_blob_owner.file_original_name
+    assert_equal file_blob_id('files/ruby.png'), new_blob_owner.file_blob_id
+    assert_equal file_blob_size('files/ruby.png'), new_blob_owner.file_size
+    assert_equal file_blob_data('files/ruby.png'), new_blob_owner.file_data
+    assert_equal file_blobs(:ruby_png), new_blob_owner.file_blob
+  end
+
+  test 'proxy setter on the owner model with different-table blobs' do
+    @message.attachment = @blob_owner.file
+
+    assert_equal 'ruby.png', @message.attachment_original_name
+    assert_equal file_blob_id('files/ruby.png'), @message.attachment_blob_id
+    assert_equal file_blob_size('files/ruby.png'), @message.attachment_size
+
+    assert_instance_of GemTestBlob, @message.attachment_blob
+    assert_equal file_blob_id('files/ruby.png'), @message.attachment_blob.id
+    assert_equal file_blob_data('files/ruby.png'),
+                 @message.attachment_blob.data
+    assert_equal true, @message.attachment_blob.new_record?
+  end
+end

data/test/file_blob_test.rb

@@ -0,0 +1,8 @@
+require_relative 'test_helper'
+
+# Run the tests in the generator, to make sure they pass.
+template_path = File.expand_path(
+    '../../lib/file_blobs_rails/generators/templates/file_blob_test.rb.erb',
+    __FILE__)
+eval Erubis::Eruby.new(File.read(template_path)).result(
+    file_name: 'file_blob'), nil, template_path

data/test/file_blobs_fixture_test.rb

@@ -0,0 +1,21 @@
+require_relative 'test_helper'
+
+class FileBlobsFixtureTest < ActiveSupport::TestCase
+  setup do
+    @ruby = FileBlob.where(id: file_blob_id('files/ruby.png')).first
+    @invoice = FileBlob.where(id: file_blob_id('files/invoice.pdf')).first
+
+    @ruby_path = File.expand_path '../fixtures/files/ruby.png', __FILE__
+    @invoice_path = File.expand_path '../fixtures/files/invoice.pdf', __FILE__
+  end
+
+  test 'ruby fixture' do
+    assert @ruby
+    assert_equal file_blob_data('files/ruby.png'), @ruby.data
+  end
+
+  test 'invoice fixture' do
+    assert @invoice
+    assert_equal file_blob_data('files/invoice.pdf'), @invoice.data
+  end
+end