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