Chris@1295: module ObjectHelpers
Chris@1295:   def User.generate!(attributes={})
Chris@1295:     @generated_user_login ||= 'user0'
Chris@1295:     @generated_user_login.succ!
Chris@1295:     user = User.new(attributes)
Chris@1295:     user.login = @generated_user_login if user.login.blank?
Chris@1295:     user.mail = "#{@generated_user_login}@example.com" if user.mail.blank?
Chris@1295:     user.firstname = "Bob" if user.firstname.blank?
Chris@1295:     user.lastname = "Doe" if user.lastname.blank?
Chris@1295:     yield user if block_given?
Chris@1295:     user.save!
Chris@1295:     user
Chris@1295:   end
Chris@1295: 
Chris@1295:   def User.add_to_project(user, project, roles=nil)
Chris@1295:     roles = Role.find(1) if roles.nil?
Chris@1295:     roles = [roles] unless roles.is_a?(Array)
Chris@1295:     Member.create!(:principal => user, :project => project, :roles => roles)
Chris@1295:   end
Chris@1295: 
Chris@1295:   def Group.generate!(attributes={})
Chris@1295:     @generated_group_name ||= 'Group 0'
Chris@1295:     @generated_group_name.succ!
Chris@1295:     group = Group.new(attributes)
Chris@1295:     group.name = @generated_group_name if group.name.blank?
Chris@1295:     yield group if block_given?
Chris@1295:     group.save!
Chris@1295:     group
Chris@1295:   end
Chris@1295: 
Chris@1295:   def Project.generate!(attributes={})
Chris@1295:     @generated_project_identifier ||= 'project-0000'
Chris@1295:     @generated_project_identifier.succ!
Chris@1295:     project = Project.new(attributes)
Chris@1295:     project.name = @generated_project_identifier if project.name.blank?
Chris@1295:     project.identifier = @generated_project_identifier if project.identifier.blank?
Chris@1295:     yield project if block_given?
Chris@1295:     project.save!
Chris@1295:     project
Chris@1295:   end
Chris@1295: 
Chris@1295:   def Tracker.generate!(attributes={})
Chris@1295:     @generated_tracker_name ||= 'Tracker 0'
Chris@1295:     @generated_tracker_name.succ!
Chris@1295:     tracker = Tracker.new(attributes)
Chris@1295:     tracker.name = @generated_tracker_name if tracker.name.blank?
Chris@1295:     yield tracker if block_given?
Chris@1295:     tracker.save!
Chris@1295:     tracker
Chris@1295:   end
Chris@1295: 
Chris@1295:   def Role.generate!(attributes={})
Chris@1295:     @generated_role_name ||= 'Role 0'
Chris@1295:     @generated_role_name.succ!
Chris@1295:     role = Role.new(attributes)
Chris@1295:     role.name = @generated_role_name if role.name.blank?
Chris@1295:     yield role if block_given?
Chris@1295:     role.save!
Chris@1295:     role
Chris@1295:   end
Chris@1295: 
Chris@1295:   def Issue.generate!(attributes={})
Chris@1295:     issue = Issue.new(attributes)
Chris@1295:     issue.project ||= Project.find(1)
Chris@1295:     issue.tracker ||= issue.project.trackers.first
Chris@1295:     issue.subject = 'Generated' if issue.subject.blank?
Chris@1295:     issue.author ||= User.find(2)
Chris@1295:     yield issue if block_given?
Chris@1295:     issue.save!
Chris@1295:     issue
Chris@1295:   end
Chris@1295: 
Chris@1295:   # Generates an issue with 2 children and a grandchild
Chris@1295:   def Issue.generate_with_descendants!(attributes={})
Chris@1295:     issue = Issue.generate!(attributes)
Chris@1295:     child = Issue.generate!(:project => issue.project, :subject => 'Child1', :parent_issue_id => issue.id)
Chris@1295:     Issue.generate!(:project => issue.project, :subject => 'Child2', :parent_issue_id => issue.id)
Chris@1295:     Issue.generate!(:project => issue.project, :subject => 'Child11', :parent_issue_id => child.id)
Chris@1295:     issue.reload
Chris@1295:   end
Chris@1295: 
Chris@1295:   def Journal.generate!(attributes={})
Chris@1295:     journal = Journal.new(attributes)
Chris@1295:     journal.user ||= User.first
Chris@1295:     journal.journalized ||= Issue.first
Chris@1295:     yield journal if block_given?
Chris@1295:     journal.save!
Chris@1295:     journal
Chris@1295:   end
Chris@1295: 
Chris@1295:   def Version.generate!(attributes={})
Chris@1295:     @generated_version_name ||= 'Version 0'
Chris@1295:     @generated_version_name.succ!
Chris@1295:     version = Version.new(attributes)
Chris@1295:     version.name = @generated_version_name if version.name.blank?
Chris@1295:     yield version if block_given?
Chris@1295:     version.save!
Chris@1295:     version
Chris@1295:   end
Chris@1295: 
Chris@1295:   def AuthSource.generate!(attributes={})
Chris@1295:     @generated_auth_source_name ||= 'Auth 0'
Chris@1295:     @generated_auth_source_name.succ!
Chris@1295:     source = AuthSource.new(attributes)
Chris@1295:     source.name = @generated_auth_source_name if source.name.blank?
Chris@1295:     yield source if block_given?
Chris@1295:     source.save!
Chris@1295:     source
Chris@1295:   end
Chris@1295: 
Chris@1295:   def Board.generate!(attributes={})
Chris@1295:     @generated_board_name ||= 'Forum 0'
Chris@1295:     @generated_board_name.succ!
Chris@1295:     board = Board.new(attributes)
Chris@1295:     board.name = @generated_board_name if board.name.blank?
Chris@1295:     board.description = @generated_board_name if board.description.blank?
Chris@1295:     yield board if block_given?
Chris@1295:     board.save!
Chris@1295:     board
Chris@1295:   end
Chris@1295: end