Merge pull request lokka#200 from morygonzalez/months-should-not-cont…

…ain-draft Lokka::Helpers.months should not contain draft entry
hirocaster · Jul 28, 2013 · ee002ce · ee002ce
2 parents 1c1c3f9 + ecad5c9
commit ee002ce
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lib/lokka/helpers/helpers.rb b/lib/lokka/helpers/helpers.rb
@@ -66,7 +66,7 @@ def comment_form
 
     def months
       ms = {}
-      Post.all.each do |post|
+      Post.published.each do |post|
         m = post.created_at.strftime('%Y-%m')
         if ms[m].nil?
           ms[m] = 1

diff --git a/spec/unit/helpers_spec.rb b/spec/unit/helpers_spec.rb
@@ -7,6 +7,7 @@
       gravatar_image_url().should eql('http://www.gravatar.com/avatar/00000000000000000000000000000000')
     end
   end
+
   context 'custom_permalink' do
     before do
       Option.permalink_enabled = true
@@ -50,4 +51,11 @@
       end
     end
   end
+
+  describe 'months' do
+    subject { months }
+    before  { FactoryGirl.create(:post, draft: true) }
+
+    it { subject.count.should == 0 }
+  end
 end