bookis-paperclip 2.3.3

Sign up to get free protection for your applications and to get access to all the features.
Files changed (56) hide show
  1. data/LICENSE +26 -0
  2. data/README.rdoc +179 -0
  3. data/Rakefile +76 -0
  4. data/generators/paperclip/USAGE +5 -0
  5. data/generators/paperclip/paperclip_generator.rb +27 -0
  6. data/generators/paperclip/templates/paperclip_migration.rb.erb +19 -0
  7. data/init.rb +1 -0
  8. data/lib/generators/paperclip/USAGE +8 -0
  9. data/lib/generators/paperclip/paperclip_generator.rb +31 -0
  10. data/lib/generators/paperclip/templates/paperclip_migration.rb.erb +19 -0
  11. data/lib/paperclip/attachment.rb +326 -0
  12. data/lib/paperclip/callback_compatability.rb +61 -0
  13. data/lib/paperclip/geometry.rb +115 -0
  14. data/lib/paperclip/interpolations.rb +108 -0
  15. data/lib/paperclip/iostream.rb +59 -0
  16. data/lib/paperclip/matchers/have_attached_file_matcher.rb +57 -0
  17. data/lib/paperclip/matchers/validate_attachment_content_type_matcher.rb +74 -0
  18. data/lib/paperclip/matchers/validate_attachment_presence_matcher.rb +54 -0
  19. data/lib/paperclip/matchers/validate_attachment_size_matcher.rb +95 -0
  20. data/lib/paperclip/matchers.rb +33 -0
  21. data/lib/paperclip/processor.rb +49 -0
  22. data/lib/paperclip/railtie.rb +24 -0
  23. data/lib/paperclip/storage.rb +379 -0
  24. data/lib/paperclip/style.rb +90 -0
  25. data/lib/paperclip/thumbnail.rb +78 -0
  26. data/lib/paperclip/upfile.rb +52 -0
  27. data/lib/paperclip/version.rb +3 -0
  28. data/lib/paperclip.rb +399 -0
  29. data/lib/tasks/paperclip.rake +79 -0
  30. data/rails/init.rb +2 -0
  31. data/shoulda_macros/paperclip.rb +119 -0
  32. data/test/attachment_test.rb +758 -0
  33. data/test/database.yml +4 -0
  34. data/test/fixtures/12k.png +0 -0
  35. data/test/fixtures/50x50.png +0 -0
  36. data/test/fixtures/5k.png +0 -0
  37. data/test/fixtures/bad.png +1 -0
  38. data/test/fixtures/s3.yml +8 -0
  39. data/test/fixtures/text.txt +0 -0
  40. data/test/fixtures/twopage.pdf +0 -0
  41. data/test/geometry_test.rb +177 -0
  42. data/test/helper.rb +148 -0
  43. data/test/integration_test.rb +483 -0
  44. data/test/interpolations_test.rb +124 -0
  45. data/test/iostream_test.rb +78 -0
  46. data/test/matchers/have_attached_file_matcher_test.rb +24 -0
  47. data/test/matchers/validate_attachment_content_type_matcher_test.rb +37 -0
  48. data/test/matchers/validate_attachment_presence_matcher_test.rb +26 -0
  49. data/test/matchers/validate_attachment_size_matcher_test.rb +51 -0
  50. data/test/paperclip_test.rb +318 -0
  51. data/test/processor_test.rb +10 -0
  52. data/test/storage_test.rb +343 -0
  53. data/test/style_test.rb +141 -0
  54. data/test/thumbnail_test.rb +227 -0
  55. data/test/upfile_test.rb +36 -0
  56. metadata +200 -0
data/lib/paperclip.rb ADDED
@@ -0,0 +1,399 @@
1
+ # Paperclip allows file attachments that are stored in the filesystem. All graphical
2
+ # transformations are done using the Graphics/ImageMagick command line utilities and
3
+ # are stored in Tempfiles until the record is saved. Paperclip does not require a
4
+ # separate model for storing the attachment's information, instead adding a few simple
5
+ # columns to your table.
6
+ #
7
+ # Author:: Jon Yurek
8
+ # Copyright:: Copyright (c) 2008-2009 thoughtbot, inc.
9
+ # License:: MIT License (http://www.opensource.org/licenses/mit-license.php)
10
+ #
11
+ # Paperclip defines an attachment as any file, though it makes special considerations
12
+ # for image files. You can declare that a model has an attached file with the
13
+ # +has_attached_file+ method:
14
+ #
15
+ # class User < ActiveRecord::Base
16
+ # has_attached_file :avatar, :styles => { :thumb => "100x100" }
17
+ # end
18
+ #
19
+ # user = User.new
20
+ # user.avatar = params[:user][:avatar]
21
+ # user.avatar.url
22
+ # # => "/users/avatars/4/original_me.jpg"
23
+ # user.avatar.url(:thumb)
24
+ # # => "/users/avatars/4/thumb_me.jpg"
25
+ #
26
+ # See the +has_attached_file+ documentation for more details.
27
+
28
+ require 'erb'
29
+ require 'tempfile'
30
+ require 'paperclip/version'
31
+ require 'paperclip/upfile'
32
+ require 'paperclip/iostream'
33
+ require 'paperclip/geometry'
34
+ require 'paperclip/processor'
35
+ require 'paperclip/thumbnail'
36
+ require 'paperclip/storage'
37
+ require 'paperclip/interpolations'
38
+ require 'paperclip/style'
39
+ require 'paperclip/attachment'
40
+ require 'paperclip/callback_compatability'
41
+ require 'paperclip/railtie'
42
+ if defined?(Rails.root) && Rails.root
43
+ Dir.glob(File.join(File.expand_path(Rails.root), "lib", "paperclip_processors", "*.rb")).each do |processor|
44
+ require processor
45
+ end
46
+ end
47
+
48
+ # The base module that gets included in ActiveRecord::Base. See the
49
+ # documentation for Paperclip::ClassMethods for more useful information.
50
+ module Paperclip
51
+
52
+ class << self
53
+ # Provides configurability to Paperclip. There are a number of options available, such as:
54
+ # * whiny: Will raise an error if Paperclip cannot process thumbnails of
55
+ # an uploaded image. Defaults to true.
56
+ # * log: Logs progress to the Rails log. Uses ActiveRecord's logger, so honors
57
+ # log levels, etc. Defaults to true.
58
+ # * command_path: Defines the path at which to find the command line
59
+ # programs if they are not visible to Rails the system's search path. Defaults to
60
+ # nil, which uses the first executable found in the user's search path.
61
+ # * image_magick_path: Deprecated alias of command_path.
62
+ def options
63
+ @options ||= {
64
+ :whiny => true,
65
+ :image_magick_path => nil,
66
+ :command_path => nil,
67
+ :log => true,
68
+ :log_command => true,
69
+ :swallow_stderr => true
70
+ }
71
+ end
72
+
73
+ def configure
74
+ yield(self) if block_given?
75
+ end
76
+
77
+ def path_for_command command #:nodoc:
78
+ if options[:image_magick_path]
79
+ warn("[DEPRECATION] :image_magick_path is deprecated and will be removed. Use :command_path instead")
80
+ end
81
+ path = [options[:command_path] || options[:image_magick_path], command].compact
82
+ File.join(*path)
83
+ end
84
+
85
+ def interpolates key, &block
86
+ Paperclip::Interpolations[key] = block
87
+ end
88
+
89
+ # The run method takes a command to execute and an array of parameters
90
+ # that get passed to it. The command is prefixed with the :command_path
91
+ # option from Paperclip.options. If you have many commands to run and
92
+ # they are in different paths, the suggested course of action is to
93
+ # symlink them so they are all in the same directory.
94
+ #
95
+ # If the command returns with a result code that is not one of the
96
+ # expected_outcodes, a PaperclipCommandLineError will be raised. Generally
97
+ # a code of 0 is expected, but a list of codes may be passed if necessary.
98
+ # These codes should be passed as a hash as the last argument, like so:
99
+ #
100
+ # Paperclip.run("echo", "something", :expected_outcodes => [0,1,2,3])
101
+ #
102
+ # This method can log the command being run when
103
+ # Paperclip.options[:log_command] is set to true (defaults to false). This
104
+ # will only log if logging in general is set to true as well.
105
+ def run cmd, *params
106
+ options = params.last.is_a?(Hash) ? params.pop : {}
107
+ expected_outcodes = options[:expected_outcodes] || [0]
108
+ params = quote_command_options(*params).join(" ")
109
+
110
+ command = %Q[#{path_for_command(cmd)} #{params}]
111
+ command = "#{command} 2>#{bit_bucket}" if Paperclip.options[:swallow_stderr]
112
+ Paperclip.log(command) if Paperclip.options[:log_command]
113
+
114
+ begin
115
+ output = `#{command}`
116
+
117
+ raise CommandNotFoundError if $?.exitstatus == 127
118
+
119
+ unless expected_outcodes.include?($?.exitstatus)
120
+ raise PaperclipCommandLineError,
121
+ "Error while running #{cmd}. Expected return code to be #{expected_outcodes.join(", ")} but was #{$?.exitstatus}",
122
+ output
123
+ end
124
+ rescue Errno::ENOENT => e
125
+ raise CommandNotFoundError
126
+ end
127
+
128
+ output
129
+ end
130
+
131
+ def quote_command_options(*options)
132
+ options.map do |option|
133
+ option.split("'").map{|m| "'#{m}'" }.join("\\'")
134
+ end
135
+ end
136
+
137
+ def bit_bucket #:nodoc:
138
+ File.exists?("/dev/null") ? "/dev/null" : "NUL"
139
+ end
140
+
141
+ def included base #:nodoc:
142
+ base.extend ClassMethods
143
+ if base.respond_to?("set_callback")
144
+ base.send :include, Paperclip::CallbackCompatability::Rails3
145
+ else
146
+ base.send :include, Paperclip::CallbackCompatability::Rails21
147
+ end
148
+ end
149
+
150
+ def processor name #:nodoc:
151
+ name = name.to_s.camelize
152
+ processor = Paperclip.const_get(name)
153
+ unless processor.ancestors.include?(Paperclip::Processor)
154
+ raise PaperclipError.new("Processor #{name} was not found")
155
+ end
156
+ processor
157
+ end
158
+
159
+ # Log a paperclip-specific line. Uses ActiveRecord::Base.logger
160
+ # by default. Set Paperclip.options[:log] to false to turn off.
161
+ def log message
162
+ logger.info("[paperclip] #{message}") if logging?
163
+ end
164
+
165
+ def logger #:nodoc:
166
+ ActiveRecord::Base.logger
167
+ end
168
+
169
+ def logging? #:nodoc:
170
+ options[:log]
171
+ end
172
+ end
173
+
174
+ class PaperclipError < StandardError #:nodoc:
175
+ end
176
+
177
+ class PaperclipCommandLineError < PaperclipError #:nodoc:
178
+ attr_accessor :output
179
+ def initialize(msg = nil, output = nil)
180
+ super(msg)
181
+ @output = output
182
+ end
183
+ end
184
+
185
+ class CommandNotFoundError < PaperclipError
186
+ end
187
+
188
+ class NotIdentifiedByImageMagickError < PaperclipError #:nodoc:
189
+ end
190
+
191
+ class InfiniteInterpolationError < PaperclipError #:nodoc:
192
+ end
193
+
194
+ module ClassMethods
195
+ # +has_attached_file+ gives the class it is called on an attribute that maps to a file. This
196
+ # is typically a file stored somewhere on the filesystem and has been uploaded by a user.
197
+ # The attribute returns a Paperclip::Attachment object which handles the management of
198
+ # that file. The intent is to make the attachment as much like a normal attribute. The
199
+ # thumbnails will be created when the new file is assigned, but they will *not* be saved
200
+ # until +save+ is called on the record. Likewise, if the attribute is set to +nil+ is
201
+ # called on it, the attachment will *not* be deleted until +save+ is called. See the
202
+ # Paperclip::Attachment documentation for more specifics. There are a number of options
203
+ # you can set to change the behavior of a Paperclip attachment:
204
+ # * +url+: The full URL of where the attachment is publically accessible. This can just
205
+ # as easily point to a directory served directly through Apache as it can to an action
206
+ # that can control permissions. You can specify the full domain and path, but usually
207
+ # just an absolute path is sufficient. The leading slash *must* be included manually for
208
+ # absolute paths. The default value is
209
+ # "/system/:attachment/:id/:style/:filename". See
210
+ # Paperclip::Attachment#interpolate for more information on variable interpolaton.
211
+ # :url => "/:class/:attachment/:id/:style_:filename"
212
+ # :url => "http://some.other.host/stuff/:class/:id_:extension"
213
+ # * +default_url+: The URL that will be returned if there is no attachment assigned.
214
+ # This field is interpolated just as the url is. The default value is
215
+ # "/:attachment/:style/missing.png"
216
+ # has_attached_file :avatar, :default_url => "/images/default_:style_avatar.png"
217
+ # User.new.avatar_url(:small) # => "/images/default_small_avatar.png"
218
+ # * +styles+: A hash of thumbnail styles and their geometries. You can find more about
219
+ # geometry strings at the ImageMagick website
220
+ # (http://www.imagemagick.org/script/command-line-options.php#resize). Paperclip
221
+ # also adds the "#" option (e.g. "50x50#"), which will resize the image to fit maximally
222
+ # inside the dimensions and then crop the rest off (weighted at the center). The
223
+ # default value is to generate no thumbnails.
224
+ # * +default_style+: The thumbnail style that will be used by default URLs.
225
+ # Defaults to +original+.
226
+ # has_attached_file :avatar, :styles => { :normal => "100x100#" },
227
+ # :default_style => :normal
228
+ # user.avatar.url # => "/avatars/23/normal_me.png"
229
+ # * +whiny+: Will raise an error if Paperclip cannot post_process an uploaded file due
230
+ # to a command line error. This will override the global setting for this attachment.
231
+ # Defaults to true. This option used to be called :whiny_thumbanils, but this is
232
+ # deprecated.
233
+ # * +convert_options+: When creating thumbnails, use this free-form options
234
+ # array to pass in various convert command options. Typical options are "-strip" to
235
+ # remove all Exif data from the image (save space for thumbnails and avatars) or
236
+ # "-depth 8" to specify the bit depth of the resulting conversion. See ImageMagick
237
+ # convert documentation for more options: (http://www.imagemagick.org/script/convert.php)
238
+ # Note that this option takes a hash of options, each of which correspond to the style
239
+ # of thumbnail being generated. You can also specify :all as a key, which will apply
240
+ # to all of the thumbnails being generated. If you specify options for the :original,
241
+ # it would be best if you did not specify destructive options, as the intent of keeping
242
+ # the original around is to regenerate all the thumbnails when requirements change.
243
+ # has_attached_file :avatar, :styles => { :large => "300x300", :negative => "100x100" }
244
+ # :convert_options => {
245
+ # :all => "-strip",
246
+ # :negative => "-negate"
247
+ # }
248
+ # NOTE: While not deprecated yet, it is not recommended to specify options this way.
249
+ # It is recommended that :convert_options option be included in the hash passed to each
250
+ # :styles for compatability with future versions.
251
+ # NOTE: Strings supplied to :convert_options are split on space in order to undergo
252
+ # shell quoting for safety. If your options require a space, please pre-split them
253
+ # and pass an array to :convert_options instead.
254
+ # * +storage+: Chooses the storage backend where the files will be stored. The current
255
+ # choices are :filesystem and :s3. The default is :filesystem. Make sure you read the
256
+ # documentation for Paperclip::Storage::Filesystem and Paperclip::Storage::S3
257
+ # for backend-specific options.
258
+ def has_attached_file name, options = {}
259
+ include InstanceMethods
260
+
261
+ write_inheritable_attribute(:attachment_definitions, {}) if attachment_definitions.nil?
262
+ attachment_definitions[name] = {:validations => []}.merge(options)
263
+
264
+ after_save :save_attached_files
265
+ before_destroy :destroy_attached_files
266
+
267
+ define_paperclip_callbacks :post_process, :"#{name}_post_process"
268
+
269
+ define_method name do |*args|
270
+ a = attachment_for(name)
271
+ (args.length > 0) ? a.to_s(args.first) : a
272
+ end
273
+
274
+ define_method "#{name}=" do |file|
275
+ attachment_for(name).assign(file)
276
+ end
277
+
278
+ define_method "#{name}?" do
279
+ attachment_for(name).file?
280
+ end
281
+
282
+ validates_each(name) do |record, attr, value|
283
+ attachment = record.attachment_for(name)
284
+ attachment.send(:flush_errors)
285
+ end
286
+ end
287
+
288
+ # Places ActiveRecord-style validations on the size of the file assigned. The
289
+ # possible options are:
290
+ # * +in+: a Range of bytes (i.e. +1..1.megabyte+),
291
+ # * +less_than+: equivalent to :in => 0..options[:less_than]
292
+ # * +greater_than+: equivalent to :in => options[:greater_than]..Infinity
293
+ # * +message+: error message to display, use :min and :max as replacements
294
+ # * +if+: A lambda or name of a method on the instance. Validation will only
295
+ # be run is this lambda or method returns true.
296
+ # * +unless+: Same as +if+ but validates if lambda or method returns false.
297
+ def validates_attachment_size name, options = {}
298
+ min = options[:greater_than] || (options[:in] && options[:in].first) || 0
299
+ max = options[:less_than] || (options[:in] && options[:in].last) || (1.0/0)
300
+ range = (min..max)
301
+ message = options[:message] || "file size must be between :min and :max bytes."
302
+ message = message.gsub(/:min/, min.to_s).gsub(/:max/, max.to_s)
303
+
304
+ validates_inclusion_of :"#{name}_file_size",
305
+ :in => range,
306
+ :message => message,
307
+ :if => options[:if],
308
+ :unless => options[:unless],
309
+ :allow_nil => true
310
+ end
311
+
312
+ # Adds errors if thumbnail creation fails. The same as specifying :whiny_thumbnails => true.
313
+ def validates_attachment_thumbnails name, options = {}
314
+ warn('[DEPRECATION] validates_attachment_thumbnail is deprecated. ' +
315
+ 'This validation is on by default and will be removed from future versions. ' +
316
+ 'If you wish to turn it off, supply :whiny => false in your definition.')
317
+ attachment_definitions[name][:whiny_thumbnails] = true
318
+ end
319
+
320
+ # Places ActiveRecord-style validations on the presence of a file.
321
+ # Options:
322
+ # * +if+: A lambda or name of a method on the instance. Validation will only
323
+ # be run is this lambda or method returns true.
324
+ # * +unless+: Same as +if+ but validates if lambda or method returns false.
325
+ def validates_attachment_presence name, options = {}
326
+ message = options[:message] || "must be set."
327
+ validates_presence_of :"#{name}_file_name",
328
+ :message => message,
329
+ :if => options[:if],
330
+ :unless => options[:unless]
331
+ end
332
+
333
+ # Places ActiveRecord-style validations on the content type of the file
334
+ # assigned. The possible options are:
335
+ # * +content_type+: Allowed content types. Can be a single content type
336
+ # or an array. Each type can be a String or a Regexp. It should be
337
+ # noted that Internet Explorer upload files with content_types that you
338
+ # may not expect. For example, JPEG images are given image/pjpeg and
339
+ # PNGs are image/x-png, so keep that in mind when determining how you
340
+ # match. Allows all by default.
341
+ # * +message+: The message to display when the uploaded file has an invalid
342
+ # content type.
343
+ # * +if+: A lambda or name of a method on the instance. Validation will only
344
+ # be run is this lambda or method returns true.
345
+ # * +unless+: Same as +if+ but validates if lambda or method returns false.
346
+ # NOTE: If you do not specify an [attachment]_content_type field on your
347
+ # model, content_type validation will work _ONLY upon assignment_ and
348
+ # re-validation after the instance has been reloaded will always succeed.
349
+ def validates_attachment_content_type name, options = {}
350
+ validation_options = options.dup
351
+ allowed_types = [validation_options[:content_type]].flatten
352
+ validates_each(:"#{name}_content_type", validation_options) do |record, attr, value|
353
+ if !allowed_types.any?{|t| t === value } && value.present?
354
+ if record.errors.method(:add).arity == -2
355
+ message = options[:message] || "is not one of #{allowed_types.join(", ")}"
356
+ record.errors.add(:"#{name}_content_type", message)
357
+ else
358
+ record.errors.add(:"#{name}_content_type", :inclusion, :default => options[:message], :value => value)
359
+ end
360
+ end
361
+ end
362
+ end
363
+
364
+ # Returns the attachment definitions defined by each call to
365
+ # has_attached_file.
366
+ def attachment_definitions
367
+ read_inheritable_attribute(:attachment_definitions)
368
+ end
369
+ end
370
+
371
+ module InstanceMethods #:nodoc:
372
+ def attachment_for name
373
+ @_paperclip_attachments ||= {}
374
+ @_paperclip_attachments[name] ||= Attachment.new(name, self, self.class.attachment_definitions[name])
375
+ end
376
+
377
+ def each_attachment
378
+ self.class.attachment_definitions.each do |name, definition|
379
+ yield(name, attachment_for(name))
380
+ end
381
+ end
382
+
383
+ def save_attached_files
384
+ Paperclip.log("Saving attachments.")
385
+ each_attachment do |name, attachment|
386
+ attachment.send(:save)
387
+ end
388
+ end
389
+
390
+ def destroy_attached_files
391
+ Paperclip.log("Deleting attachments.")
392
+ each_attachment do |name, attachment|
393
+ attachment.send(:queue_existing_for_delete)
394
+ attachment.send(:flush_deletes)
395
+ end
396
+ end
397
+ end
398
+
399
+ end
@@ -0,0 +1,79 @@
1
+ def obtain_class
2
+ class_name = ENV['CLASS'] || ENV['class']
3
+ raise "Must specify CLASS" unless class_name
4
+ @klass = Object.const_get(class_name)
5
+ end
6
+
7
+ def obtain_attachments
8
+ name = ENV['ATTACHMENT'] || ENV['attachment']
9
+ raise "Class #{@klass.name} has no attachments specified" unless @klass.respond_to?(:attachment_definitions)
10
+ if !name.blank? && @klass.attachment_definitions.keys.include?(name)
11
+ [ name ]
12
+ else
13
+ @klass.attachment_definitions.keys
14
+ end
15
+ end
16
+
17
+ def for_all_attachments
18
+ klass = obtain_class
19
+ names = obtain_attachments
20
+ ids = klass.connection.select_values(klass.send(:construct_finder_sql, :select => 'id'))
21
+
22
+ ids.each do |id|
23
+ instance = klass.find(id)
24
+ names.each do |name|
25
+ result = if instance.send("#{ name }?")
26
+ yield(instance, name)
27
+ else
28
+ true
29
+ end
30
+ print result ? "." : "x"; $stdout.flush
31
+ end
32
+ end
33
+ puts " Done."
34
+ end
35
+
36
+ namespace :paperclip do
37
+ desc "Refreshes both metadata and thumbnails."
38
+ task :refresh => ["paperclip:refresh:metadata", "paperclip:refresh:thumbnails"]
39
+
40
+ namespace :refresh do
41
+ desc "Regenerates thumbnails for a given CLASS (and optional ATTACHMENT)."
42
+ task :thumbnails => :environment do
43
+ errors = []
44
+ for_all_attachments do |instance, name|
45
+ result = instance.send(name).reprocess!
46
+ errors << [instance.id, instance.errors] unless instance.errors.blank?
47
+ result
48
+ end
49
+ errors.each{|e| puts "#{e.first}: #{e.last.full_messages.inspect}" }
50
+ end
51
+
52
+ desc "Regenerates content_type/size metadata for a given CLASS (and optional ATTACHMENT)."
53
+ task :metadata => :environment do
54
+ for_all_attachments do |instance, name|
55
+ if file = instance.send(name).to_file
56
+ instance.send("#{name}_file_name=", instance.send("#{name}_file_name").strip)
57
+ instance.send("#{name}_content_type=", file.content_type.strip)
58
+ instance.send("#{name}_file_size=", file.size) if instance.respond_to?("#{name}_file_size")
59
+ instance.save(false)
60
+ else
61
+ true
62
+ end
63
+ end
64
+ end
65
+ end
66
+
67
+ desc "Cleans out invalid attachments. Useful after you've added new validations."
68
+ task :clean => :environment do
69
+ for_all_attachments do |instance, name|
70
+ instance.send(name).send(:validate)
71
+ if instance.send(name).valid?
72
+ true
73
+ else
74
+ instance.send("#{name}=", nil)
75
+ instance.save
76
+ end
77
+ end
78
+ end
79
+ end
data/rails/init.rb ADDED
@@ -0,0 +1,2 @@
1
+ require 'paperclip/railtie'
2
+ Paperclip::Railtie.insert
@@ -0,0 +1,119 @@
1
+ require 'paperclip/matchers'
2
+ require 'action_controller'
3
+
4
+ module Paperclip
5
+ # =Paperclip Shoulda Macros
6
+ #
7
+ # These macros are intended for use with shoulda, and will be included into
8
+ # your tests automatically. All of the macros use the standard shoulda
9
+ # assumption that the name of the test is based on the name of the model
10
+ # you're testing (that is, UserTest is the test for the User model), and
11
+ # will load that class for testing purposes.
12
+ module Shoulda
13
+ include Matchers
14
+ # This will test whether you have defined your attachment correctly by
15
+ # checking for all the required fields exist after the definition of the
16
+ # attachment.
17
+ def should_have_attached_file name
18
+ klass = self.name.gsub(/Test$/, '').constantize
19
+ matcher = have_attached_file name
20
+ should matcher.description do
21
+ assert_accepts(matcher, klass)
22
+ end
23
+ end
24
+
25
+ # Tests for validations on the presence of the attachment.
26
+ def should_validate_attachment_presence name
27
+ klass = self.name.gsub(/Test$/, '').constantize
28
+ matcher = validate_attachment_presence name
29
+ should matcher.description do
30
+ assert_accepts(matcher, klass)
31
+ end
32
+ end
33
+
34
+ # Tests that you have content_type validations specified. There are two
35
+ # options, :valid and :invalid. Both accept an array of strings. The
36
+ # strings should be a list of content types which will pass and fail
37
+ # validation, respectively.
38
+ def should_validate_attachment_content_type name, options = {}
39
+ klass = self.name.gsub(/Test$/, '').constantize
40
+ valid = [options[:valid]].flatten
41
+ invalid = [options[:invalid]].flatten
42
+ matcher = validate_attachment_content_type(name).allowing(valid).rejecting(invalid)
43
+ should matcher.description do
44
+ assert_accepts(matcher, klass)
45
+ end
46
+ end
47
+
48
+ # Tests to ensure that you have file size validations turned on. You
49
+ # can pass the same options to this that you can to
50
+ # validate_attachment_file_size - :less_than, :greater_than, and :in.
51
+ # :less_than checks that a file is less than a certain size, :greater_than
52
+ # checks that a file is more than a certain size, and :in takes a Range or
53
+ # Array which specifies the lower and upper limits of the file size.
54
+ def should_validate_attachment_size name, options = {}
55
+ klass = self.name.gsub(/Test$/, '').constantize
56
+ min = options[:greater_than] || (options[:in] && options[:in].first) || 0
57
+ max = options[:less_than] || (options[:in] && options[:in].last) || (1.0/0)
58
+ range = (min..max)
59
+ matcher = validate_attachment_size(name).in(range)
60
+ should matcher.description do
61
+ assert_accepts(matcher, klass)
62
+ end
63
+ end
64
+
65
+ # Stubs the HTTP PUT for an attachment using S3 storage.
66
+ #
67
+ # @example
68
+ # stub_paperclip_s3('user', 'avatar', 'png')
69
+ def stub_paperclip_s3(model, attachment, extension)
70
+ definition = model.gsub(" ", "_").classify.constantize.
71
+ attachment_definitions[attachment.to_sym]
72
+
73
+ path = "http://s3.amazonaws.com/:id/#{definition[:path]}"
74
+ path.gsub!(/:([^\/\.]+)/) do |match|
75
+ "([^\/\.]+)"
76
+ end
77
+
78
+ begin
79
+ FakeWeb.register_uri(:put, Regexp.new(path), :body => "OK")
80
+ rescue NameError
81
+ raise NameError, "the stub_paperclip_s3 shoulda macro requires the fakeweb gem."
82
+ end
83
+ end
84
+
85
+ # Stub S3 and return a file for attachment. Best with Factory Girl.
86
+ # Uses a strict directory convention:
87
+ #
88
+ # features/support/paperclip
89
+ #
90
+ # This method is used by the Paperclip-provided Cucumber step:
91
+ #
92
+ # When I attach a "demo_tape" "mp3" file to a "band" on S3
93
+ #
94
+ # @example
95
+ # Factory.define :band_with_demo_tape, :parent => :band do |band|
96
+ # band.demo_tape { band.paperclip_fixture("band", "demo_tape", "png") }
97
+ # end
98
+ def paperclip_fixture(model, attachment, extension)
99
+ stub_paperclip_s3(model, attachment, extension)
100
+ base_path = File.join(File.dirname(__FILE__), "..", "..",
101
+ "features", "support", "paperclip")
102
+ File.new(File.join(base_path, model, "#{attachment}.#{extension}"))
103
+ end
104
+ end
105
+ end
106
+
107
+ if defined?(ActionController::Integration::Session)
108
+ class ActionController::Integration::Session #:nodoc:
109
+ include Paperclip::Shoulda
110
+ end
111
+ end
112
+
113
+ class Factory
114
+ include Paperclip::Shoulda #:nodoc:
115
+ end
116
+
117
+ class Test::Unit::TestCase #:nodoc:
118
+ extend Paperclip::Shoulda
119
+ end