specs/spec_helper.rb in pdf-wrapper-0.0.6 vs specs/spec_helper.rb in pdf-wrapper-0.0.7

@@ -12,33 +12,48 @@
 
 # make some private methods of PDF::Wrapper public for testing
 class PDF::Wrapper
   public :build_pango_layout
   public :calc_image_dimensions
-  public :load_librsvg
-  public :load_libpixbuf
-  public :load_libpango
-  public :load_libpoppler
   public :default_text_options
   public :detect_image_type
   public :draw_pdf
   public :draw_pixbuf
   public :draw_png
   public :draw_svg
+  public :image_dimensions
+  public :load_librsvg
+  public :load_libpixbuf
+  public :load_libpango
+  public :load_libpoppler
   public :validate_color
 end
 
-# a helper class for couting the number of pages in a PDF
+# a helper class for counting the number of pages in a PDF
 class PageReceiver
   attr_accessor :page_count
 
   def initialize
     @page_count = 0
   end
 
   # Called when page parsing ends
   def end_page
     @page_count += 1
+  end
+end
+
+# a helper class for recording the dimensions of pages in a PDF
+class PageSizeReceiver
+  attr_accessor :pages
+
+  def initialize
+    @pages = []
+  end
+
+  # Called when page parsing ends
+  def begin_page(args)
+    pages << args["MediaBox"] || args[:MediaBox]
   end
 end
 
 class PageTextReceiver
   attr_accessor :content