gitlab-org--gitlab-foss/spec/factories/deployments.rb

# frozen_string_literal: true

FactoryBot.define do
  factory :deployment, class: Deployment do
    sha { 'b83d6e391c22777fca1ed3012fce84f633d7fed0' }
    ref { 'master' }
    tag { false }
    user { nil }
    project { nil }
    deployable factory: :ci_build
    environment factory: :environment

    after(:build) do |deployment, evaluator|
      deployment.project ||= deployment.environment.project
      deployment.user ||= deployment.project.creator

      unless deployment.project.repository_exists?
        allow(deployment.project.repository).to receive(:create_ref)
      end

      if deployment.cluster && deployment.cluster.project_type? && deployment.cluster.project.nil?
        deployment.cluster.projects << deployment.project
      end
    end

    trait :review_app do
      sha { TestEnv::BRANCH_SHA['pages-deploy'] }
      ref { 'pages-deploy' }
    end

    trait :on_cluster do
      cluster factory: %i(cluster provided_by_gcp)
    end

    trait :running do
      status { :running }
    end

    trait :success do
      status { :success }
      finished_at { Time.now }
    end

    trait :failed do
      status { :failed }
      finished_at { Time.now }
    end

    trait :canceled do
      status { :canceled }
      finished_at { Time.now }
    end

    trait :created do
      status { :created }
    end

    # This trait hooks the state maechine's events
    trait :succeed do
      after(:create) do |deployment, evaluator|
        deployment.succeed!
      end
    end
  end
end