gitlab-foss/spec/factories/deploy_tokens.rb

# frozen_string_literal: true

FactoryBot.define do
  factory :deploy_token do
    sequence(:name) { |n| "PDT #{n}" }
    read_repository { true }
    read_registry { true }
    write_registry { false }
    read_package_registry { false }
    write_package_registry { false }
    read_virtual_registry { false }
    revoked { false }
    expires_at { 5.days.from_now.to_datetime }
    deploy_token_type { DeployToken.deploy_token_types[:project_type] }
    projects { [association(:project)] }

    before(:create) do |record|
      record.project_id ||= record.project&.id
      record.group_id ||= record.group&.id
    end

    trait :revoked do
      revoked { true }
    end

    trait :gitlab_deploy_token do
      name { DeployToken::GITLAB_DEPLOY_TOKEN_NAME }
    end

    trait :expired do
      expires_at { Date.current - 1.month }
    end

    trait :group do
      deploy_token_type { DeployToken.deploy_token_types[:group_type] }
      projects { [] }
      groups { [association(:group)] }
    end

    trait :project do
      deploy_token_type { DeployToken.deploy_token_types[:project_type] }
    end

    trait :all_scopes do
      write_registry { true }
      read_package_registry { true }
      write_package_registry { true }
      read_virtual_registry { true }
      write_virtual_registry { true }
    end

    trait :dependency_proxy_scopes do
      write_registry { true }
    end

    trait :virtual_registry_scopes do
      read_virtual_registry { true }
      write_virtual_registry { true }
    end
  end
end