notes.rb 2.48 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's avatar
Grzegorz Bizon committed
10
    on_issue
11

12 13 14 15
    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]
16
    factory :note_on_personal_snippet,   traits: [:on_personal_snippet]
17
    factory :system_note,                traits: [:system]
18

Douwe Maan's avatar
Douwe Maan committed
19 20 21 22 23 24 25 26 27 28 29 30 31
    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
32 33 34 35 36

      trait :resolved do
        resolved_at { Time.now }
        resolved_by { create(:user) }
      end
Douwe Maan's avatar
Douwe Maan committed
37 38 39 40 41 42 43 44 45 46 47 48 49 50
    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

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

Douwe Maan's avatar
Douwe Maan committed
58
    trait :legacy_diff_note do
59 60 61
      line_code "0_184_184"
    end

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

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

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

74 75 76 77 78
    trait :on_personal_snippet do
      noteable { create(:personal_snippet) }
      project nil
    end

79 80 81 82
    trait :system do
      system true
    end

83 84 85 86 87 88 89 90
    trait :downvote do
      note "thumbsdown"
    end

    trait :upvote do
      note "thumbsup"
    end

91 92 93 94 95
    trait :with_attachment do
      attachment { fixture_file_upload(Rails.root + "spec/fixtures/dk.png", "`/png") }
    end
  end
end