Spec support activerecord

spec/factories/story_factory.rb

@@ -14,11 +14,11 @@ def source
   def self.build(params = {})
     FakeStory.new(
       id: rand(100),
-      title: params[:title] || Faker::Lorem.sentence, 
-      permalink: params[:permalink] || Faker::Internet.url, 
+      title: params[:title] || Faker::Lorem.sentence,
+      permalink: params[:permalink] || Faker::Internet.url,
       body: params[:body] || Faker::Lorem.paragraph,
       feed: params[:feed] || FeedFactory.build,
       is_read: params[:is_read] || false,
       published: params[:published] ||Time.now)
   end
-end
+end

spec/models/story_spec.rb

@@ -0,0 +1,40 @@
+require "spec_helper"
+require 'support/active_record'
+
+app_require "models/story"
+
+describe "Story" do
+
+  let(:story) do
+    Story.new(
+      title: Faker::Lorem.sentence(50),
+      body: Faker::Lorem.sentence(50)
+    )
+  end
+
+  describe "#headline" do
+    it "truncates to 50 chars" do
+      story.headline.size.should eq(50)
+    end
+
+    it "uses a fallback string if story has no title" do
+      story.title = nil
+      story.headline.should eq(Story::UNTITLED)
+    end
+  end
+
+  describe "#lead" do
+    it "truncates to 100 chars" do
+      story.lead.size.should eq(100)
+    end
+  end
+
+  describe "#source" do
+    let(:feed) { Feed.new(name: 'Superfeed') }
+    before { story.feed = feed }
+
+    it "returns the feeds name" do
+      story.source.should eq(feed.name)
+    end
+  end
+end

spec/spec_helper.rb

@@ -30,5 +30,5 @@ def app
 end
 
 def session
-  last_request.env['rack.session'] 
+  last_request.env['rack.session']
 end

spec/support/active_record.rb

@@ -0,0 +1,26 @@
+require 'active_record'
+
+ActiveRecord::Base.establish_connection adapter: "sqlite3", database: "db/stringer_test.sqlite"
+
+def need_to_migrate?
+  current_migration = ActiveRecord::Migrator.current_version
+  latest_migration = begin
+    ActiveRecord::Migrator.get_all_versions.last
+  rescue ActiveRecord::StatementInvalid
+    nil
+  end
+  current_migration != latest_migration
+end
+
+if need_to_migrate?
+  ActiveRecord::Migrator.up "db/migrate"
+end
+
+RSpec.configure do |config|
+  config.around do |example|
+    ActiveRecord::Base.transaction do
+      example.run
+      raise ActiveRecord::Rollback
+    end
+  end
+end