RailsMug 0.0.1 → 0.0.2

data/lib/RailsMug/images.rb

@@ -26,6 +26,7 @@ module RailsMug
 
 			raise RailsMug::ImagesError, "SmugMug images.get Error: #{json_response['code']} - #{json_response['message']}" if json_response['stat'] == "fail"
 
+			count = 0
 			json_response['Album']['Images'].each do |img|
 				img = RailsMug::Model::Image.new(
 					:id => img['id'],
@@ -33,6 +34,9 @@ module RailsMug
 				)
 				img.get_photo_info @access_token unless opts[:no_info]
 				self.images << img
+
+				count = count + 1
+				break if !opts[:limit].nil? && count >= opts[:limit].to_i
 			end
 
 			self.images

data/lib/RailsMug/version.rb

@@ -1,3 +1,3 @@
 module RailsMug
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/spec/images_spec.rb

@@ -2,16 +2,34 @@ require 'spec_helper'
 
 describe RailsMug do
 	session = RailsMug::Session.new('<API KEY>','<API SECRET>','<ACCESS TOKEN>','<ACCESS TOKEN SECRET>')
+	album_id = 28536068
+	album_key = '4b4dGb'
+
+	it "should require album_id" do
+		rails_mug = RailsMug::Images.new(session)
+		expect { rails_mug.get_images }.to raise_error
+	end
+
+	it "should require album_key" do
+		rails_mug = RailsMug::Images.new(session)
+		expect { rails_mug.get_images album_id }.to raise_error
+	end
 
 	it "should get photos without detailed info" do
 		rails_mug = RailsMug::Images.new(session)
-		images = rails_mug.get_images 28536068, '4b4dGb', :no_info => true
+		images = rails_mug.get_images album_id, album_key, :no_info => true
 		images.should_not be_nil
 	end
 
 	it "should get photos with detailed info" do
 		rails_mug = RailsMug::Images.new(session)
-		rails_mug.get_images 28536068, '4b4dGb'
+		rails_mug.get_images album_id, album_key
 		rails_mug.images.should_not be_nil
 	end
+
+	it "should stop after 3 images from an album" do
+		rails_mug = RailsMug::Images.new(session)
+		rails_mug.get_images album_id, album_key, :limit => 3
+		rails_mug.images.count.should be(3)
+	end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: RailsMug
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-03-21 00:00:00.000000000 Z
+date: 2013-03-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec