BigW Consortium Gitlab

notes.rb 2.3 KB
Newer Older
1 2
require_relative '../support/repo_helpers'

3 4
include ActionDispatch::TestProcess

5 6 7 8 9
FactoryGirl.define do
  factory :note do
    project
    note "Note"
    author
Grzegorz Bizon committed
10
    on_issue
11

12 13 14 15 16
    factory :note_on_commit,             traits: [:on_commit]
    factory :note_on_issue,              traits: [:on_issue], aliases: [:votable_note]
    factory :note_on_merge_request,      traits: [:on_merge_request]
    factory :note_on_project_snippet,    traits: [:on_project_snippet]
    factory :system_note,                traits: [:system]
17

Douwe Maan committed
18 19 20 21 22 23 24 25 26 27 28 29 30
    factory :legacy_diff_note_on_commit, traits: [:on_commit, :legacy_diff_note], class: LegacyDiffNote
    factory :legacy_diff_note_on_merge_request, traits: [:on_merge_request, :legacy_diff_note], class: LegacyDiffNote

    factory :diff_note_on_merge_request, traits: [:on_merge_request], class: DiffNote do
      position do
        Gitlab::Diff::Position.new(
          old_path: "files/ruby/popen.rb",
          new_path: "files/ruby/popen.rb",
          old_line: nil,
          new_line: 14,
          diff_refs: noteable.diff_refs
        )
      end
31 32 33 34 35

      trait :resolved do
        resolved_at { Time.now }
        resolved_by { create(:user) }
      end
Douwe Maan committed
36 37 38 39 40 41 42 43 44 45 46 47 48 49
    end

    factory :diff_note_on_commit, traits: [:on_commit], class: DiffNote do
      position do
        Gitlab::Diff::Position.new(
          old_path: "files/ruby/popen.rb",
          new_path: "files/ruby/popen.rb",
          old_line: nil,
          new_line: 14,
          diff_refs: project.commit(commit_id).try(:diff_refs)
        )
      end
    end

50
    trait :on_commit do
51
      noteable nil
Grzegorz Bizon committed
52
      noteable_type 'Commit'
Douwe Maan committed
53
      noteable_id nil
54 55 56
      commit_id RepoHelpers.sample_commit.id
    end

Douwe Maan committed
57
    trait :legacy_diff_note do
58 59 60
      line_code "0_184_184"
    end

61 62
    trait :on_issue do
      noteable { create(:issue, project: project) }
63 64 65
    end

    trait :on_merge_request do
Grzegorz Bizon committed
66
      noteable { create(:merge_request, source_project: project) }
67 68
    end

69
    trait :on_project_snippet do
70
      noteable { create(:snippet, project: project) }
71 72
    end

73 74 75 76
    trait :system do
      system true
    end

77 78 79 80 81 82 83 84
    trait :downvote do
      note "thumbsdown"
    end

    trait :upvote do
      note "thumbsup"
    end

85 86 87 88 89
    trait :with_attachment do
      attachment { fixture_file_upload(Rails.root + "spec/fixtures/dk.png", "`/png") }
    end
  end
end