mirror of
				https://github.com/redmine/redmine.git
				synced 2025-10-31 10:25:55 +01:00 
			
		
		
		
	git-svn-id: http://svn.redmine.org/redmine/trunk@13535 e93f8b46-1217-0410-a6f0-8f06a7374b81
		
			
				
	
	
		
			234 lines
		
	
	
		
			6.9 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			234 lines
		
	
	
		
			6.9 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| module ObjectHelpers
 | |
|   def User.generate!(attributes={})
 | |
|     @generated_user_login ||= 'user0'
 | |
|     @generated_user_login.succ!
 | |
|     user = User.new(attributes)
 | |
|     user.login = @generated_user_login.dup if user.login.blank?
 | |
|     user.mail = "#{@generated_user_login}@example.com" if user.mail.blank?
 | |
|     user.firstname = "Bob" if user.firstname.blank?
 | |
|     user.lastname = "Doe" if user.lastname.blank?
 | |
|     yield user if block_given?
 | |
|     user.save!
 | |
|     user
 | |
|   end
 | |
| 
 | |
|   def User.add_to_project(user, project, roles=nil)
 | |
|     roles = Role.find(1) if roles.nil?
 | |
|     roles = [roles] if roles.is_a?(Role)
 | |
|     Member.create!(:principal => user, :project => project, :roles => roles)
 | |
|   end
 | |
| 
 | |
|   def Group.generate!(attributes={})
 | |
|     @generated_group_name ||= 'Group 0'
 | |
|     @generated_group_name.succ!
 | |
|     group = Group.new(attributes)
 | |
|     group.name = @generated_group_name.dup if group.name.blank?
 | |
|     yield group if block_given?
 | |
|     group.save!
 | |
|     group
 | |
|   end
 | |
| 
 | |
|   def Project.generate!(attributes={})
 | |
|     @generated_project_identifier ||= 'project-0000'
 | |
|     @generated_project_identifier.succ!
 | |
|     project = Project.new(attributes)
 | |
|     project.name = @generated_project_identifier.dup if project.name.blank?
 | |
|     project.identifier = @generated_project_identifier.dup if project.identifier.blank?
 | |
|     yield project if block_given?
 | |
|     project.save!
 | |
|     project
 | |
|   end
 | |
| 
 | |
|   def Project.generate_with_parent!(parent, attributes={})
 | |
|     project = Project.generate!(attributes)
 | |
|     project.set_parent!(parent)
 | |
|     project
 | |
|   end
 | |
| 
 | |
|   def IssueStatus.generate!(attributes={})
 | |
|     @generated_status_name ||= 'Status 0'
 | |
|     @generated_status_name.succ!
 | |
|     status = IssueStatus.new(attributes)
 | |
|     status.name = @generated_status_name.dup if status.name.blank?
 | |
|     yield status if block_given?
 | |
|     status.save!
 | |
|     status
 | |
|   end
 | |
| 
 | |
|   def Tracker.generate!(attributes={})
 | |
|     @generated_tracker_name ||= 'Tracker 0'
 | |
|     @generated_tracker_name.succ!
 | |
|     tracker = Tracker.new(attributes)
 | |
|     tracker.name = @generated_tracker_name.dup if tracker.name.blank?
 | |
|     tracker.default_status ||= IssueStatus.order('position').first || IssueStatus.generate!
 | |
|     yield tracker if block_given?
 | |
|     tracker.save!
 | |
|     tracker
 | |
|   end
 | |
| 
 | |
|   def Role.generate!(attributes={})
 | |
|     @generated_role_name ||= 'Role 0'
 | |
|     @generated_role_name.succ!
 | |
|     role = Role.new(attributes)
 | |
|     role.name = @generated_role_name.dup if role.name.blank?
 | |
|     yield role if block_given?
 | |
|     role.save!
 | |
|     role
 | |
|   end
 | |
| 
 | |
|   # Generates an unsaved Issue
 | |
|   def Issue.generate(attributes={})
 | |
|     issue = Issue.new(attributes)
 | |
|     issue.project ||= Project.find(1)
 | |
|     issue.tracker ||= issue.project.trackers.first
 | |
|     issue.subject = 'Generated' if issue.subject.blank?
 | |
|     issue.author ||= User.find(2)
 | |
|     yield issue if block_given?
 | |
|     issue
 | |
|   end
 | |
| 
 | |
|   # Generates a saved Issue
 | |
|   def Issue.generate!(attributes={}, &block)
 | |
|     issue = Issue.generate(attributes, &block)
 | |
|     issue.save!
 | |
|     issue
 | |
|   end
 | |
| 
 | |
|   # Generates an issue with 2 children and a grandchild
 | |
|   def Issue.generate_with_descendants!(attributes={})
 | |
|     issue = Issue.generate!(attributes)
 | |
|     child = Issue.generate!(:project => issue.project, :subject => 'Child1', :parent_issue_id => issue.id)
 | |
|     Issue.generate!(:project => issue.project, :subject => 'Child2', :parent_issue_id => issue.id)
 | |
|     Issue.generate!(:project => issue.project, :subject => 'Child11', :parent_issue_id => child.id)
 | |
|     issue.reload
 | |
|   end
 | |
| 
 | |
|   def Journal.generate!(attributes={})
 | |
|     journal = Journal.new(attributes)
 | |
|     journal.user ||= User.first
 | |
|     journal.journalized ||= Issue.first
 | |
|     yield journal if block_given?
 | |
|     journal.save!
 | |
|     journal
 | |
|   end
 | |
| 
 | |
|   def Version.generate!(attributes={})
 | |
|     @generated_version_name ||= 'Version 0'
 | |
|     @generated_version_name.succ!
 | |
|     version = Version.new(attributes)
 | |
|     version.name = @generated_version_name.dup if version.name.blank?
 | |
|     yield version if block_given?
 | |
|     version.save!
 | |
|     version
 | |
|   end
 | |
| 
 | |
|   def TimeEntry.generate!(attributes={})
 | |
|     entry = TimeEntry.new(attributes)
 | |
|     entry.user ||= User.find(2)
 | |
|     entry.issue ||= Issue.find(1) unless entry.project
 | |
|     entry.project ||= entry.issue.project
 | |
|     entry.activity ||= TimeEntryActivity.first
 | |
|     entry.spent_on ||= Date.today
 | |
|     entry.hours ||= 1.0
 | |
|     entry.save!
 | |
|     entry
 | |
|   end
 | |
| 
 | |
|   def AuthSource.generate!(attributes={})
 | |
|     @generated_auth_source_name ||= 'Auth 0'
 | |
|     @generated_auth_source_name.succ!
 | |
|     source = AuthSource.new(attributes)
 | |
|     source.name = @generated_auth_source_name.dup if source.name.blank?
 | |
|     yield source if block_given?
 | |
|     source.save!
 | |
|     source
 | |
|   end
 | |
| 
 | |
|   def Board.generate!(attributes={})
 | |
|     @generated_board_name ||= 'Forum 0'
 | |
|     @generated_board_name.succ!
 | |
|     board = Board.new(attributes)
 | |
|     board.name = @generated_board_name.dup if board.name.blank?
 | |
|     board.description = @generated_board_name.dup if board.description.blank?
 | |
|     yield board if block_given?
 | |
|     board.save!
 | |
|     board
 | |
|   end
 | |
| 
 | |
|   def Attachment.generate!(attributes={})
 | |
|     @generated_filename ||= 'testfile0'
 | |
|     @generated_filename.succ!
 | |
|     attributes = attributes.dup
 | |
|     attachment = Attachment.new(attributes)
 | |
|     attachment.container ||= Issue.find(1)
 | |
|     attachment.author ||= User.find(2)
 | |
|     attachment.filename = @generated_filename.dup if attachment.filename.blank?
 | |
|     attachment.save!
 | |
|     attachment
 | |
|   end
 | |
| 
 | |
|   def CustomField.generate!(attributes={})
 | |
|     @generated_custom_field_name ||= 'Custom field 0'
 | |
|     @generated_custom_field_name.succ!
 | |
|     field = new(attributes)
 | |
|     field.name = @generated_custom_field_name.dup if field.name.blank?
 | |
|     field.field_format = 'string' if field.field_format.blank?
 | |
|     yield field if block_given?
 | |
|     field.save!
 | |
|     field
 | |
|   end
 | |
| 
 | |
|   def Changeset.generate!(attributes={})
 | |
|     @generated_changeset_rev ||= '123456'
 | |
|     @generated_changeset_rev.succ!
 | |
|     changeset = new(attributes)
 | |
|     changeset.repository ||= Project.find(1).repository
 | |
|     changeset.revision ||= @generated_changeset_rev
 | |
|     changeset.committed_on ||= Time.now
 | |
|     yield changeset if block_given?
 | |
|     changeset.save!
 | |
|     changeset
 | |
|   end
 | |
| 
 | |
|   def Query.generate!(attributes={})
 | |
|     query = new(attributes)
 | |
|     query.name = "Generated query" if query.name.blank?
 | |
|     query.user ||= User.find(1)
 | |
|     query.save!
 | |
|     query
 | |
|   end
 | |
| end
 | |
| 
 | |
| module TrackerObjectHelpers
 | |
|   def generate_transitions!(*args)
 | |
|     options = args.last.is_a?(Hash) ? args.pop : {}
 | |
|     if args.size == 1
 | |
|       args << args.first
 | |
|     end
 | |
|     if options[:clear]
 | |
|       WorkflowTransition.where(:tracker_id => id).delete_all
 | |
|     end
 | |
|     args.each_cons(2) do |old_status_id, new_status_id|
 | |
|       WorkflowTransition.create!(
 | |
|         :tracker => self,
 | |
|         :role_id => (options[:role_id] || 1),
 | |
|         :old_status_id => old_status_id,
 | |
|         :new_status_id => new_status_id
 | |
|       )
 | |
|     end
 | |
|   end
 | |
| end
 | |
| Tracker.send :include, TrackerObjectHelpers
 | |
| 
 | |
| module IssueObjectHelpers
 | |
|   def close!
 | |
|     self.status = IssueStatus.where(:is_closed => true).first
 | |
|     save!
 | |
|   end
 | |
| 
 | |
|   def generate_child!(attributes={})
 | |
|     Issue.generate!(attributes.merge(:parent_issue_id => self.id))
 | |
|   end
 | |
| end
 | |
| Issue.send :include, IssueObjectHelpers
 |