object_daddy_helpers.rb
40 lines
| 1.1 KiB
| text/x-ruby
|
RubyLexer
/ test / object_daddy_helpers.rb
Eric Davis
|
r2824 | module ObjectDaddyHelpers | ||
|
Eric Davis
|
r3284 | # TODO: Remove these three once everyone has ported their code to use the | ||
| # new object_daddy version with protected attribute support | ||||
|
Eric Davis
|
r3223 | def User.generate_with_protected(attributes={}) | ||
|
Eric Davis
|
r3284 | User.generate(attributes) | ||
|
Eric Davis
|
r3223 | end | ||
|
Eric Davis
|
r2824 | def User.generate_with_protected!(attributes={}) | ||
|
Eric Davis
|
r3284 | User.generate!(attributes) | ||
|
Eric Davis
|
r3223 | end | ||
| def User.spawn_with_protected(attributes={}) | ||||
|
Eric Davis
|
r3284 | User.spawn(attributes) | ||
|
Eric Davis
|
r2824 | end | ||
|
Eric Davis
|
r3964 | def User.add_to_project(user, project, roles) | ||
| roles = [roles] unless roles.is_a?(Array) | ||||
| Member.generate!(:principal => user, :project => project, :roles => roles) | ||||
| end | ||||
|
Eric Davis
|
r2824 | # Generate the default Query | ||
| def Query.generate_default!(attributes={}) | ||||
| query = Query.spawn(attributes) | ||||
| query.name ||= '_' | ||||
| query.save! | ||||
| query | ||||
| end | ||||
| # Generate an issue for a project, using it's trackers | ||||
| def Issue.generate_for_project!(project, attributes={}) | ||||
| issue = Issue.spawn(attributes) do |issue| | ||||
| issue.project = project | ||||
|
Eric Davis
|
r3958 | issue.tracker = project.trackers.first unless project.trackers.empty? | ||
| yield issue if block_given? | ||||
|
Eric Davis
|
r2824 | end | ||
| issue.save! | ||||
| issue | ||||
| end | ||||
| end | ||||