spec/models/bookmark_spec.rb in blacklight-5.3.0 vs spec/models/bookmark_spec.rb in blacklight-5.4.0.rc1

@@ -1,31 +1,36 @@
-# -*- encoding : utf-8 -*-
-require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+require 'spec_helper'
 
 describe Bookmark do
-  before(:each) do
-    @bookmark = Bookmark.new
+  subject do
+    b = Bookmark.new 
+    b.user_id = 1
+    b.document = SolrDocument.new(id: 'u001')
+    b
   end
   
   it "should be valid" do
-    @bookmark.id = 1
-    @bookmark.user_id = 1
-    @bookmark.document_id = 'u001'
-    expect(@bookmark).to be_valid
+    expect(subject).to be_valid
   end
-   
-  it "should require user_id" do
-    expect(@bookmark).to have(1).error_on(:user_id)
-  end
 
   it "should belong to user" do
     expect(Bookmark.reflect_on_association(:user)).not_to be_nil
   end
 
   it "should be valid after saving" do
-    @bookmark.id = 1
-    @bookmark.user_id = 1
-    @bookmark.document_id = 'u001'
-    @bookmark.save
-    expect(@bookmark).to be_valid
+    subject.save
+    expect(subject).to be_valid
+  end
+  
+  describe "#document_type" do
+    it "should be the class of the solr document" do
+      expect(subject.document_type).to eq SolrDocument
+    end
+  end
+  
+  describe "#document" do
+    it "should be a SolrDocument with just an id field" do
+      expect(subject.document).to be_a_kind_of SolrDocument
+      expect(subject.document.id).to eq 'u001'
+    end
   end
 end