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

FactoryGirl.define do
  factory :environment, class: Environment do
    sequence(:name) { |n| "environment#{n}" }

    project factory: :empty_project
    sequence(:external_url) { |n| "https://env#{n}.example.gitlab.com" }

    trait :with_review_app do |environment|
      project

      # At this point `review app` is an ephemeral concept related to
      # deployments being deployed for given environment. There is no
      # first-class `review app` available so we need to create set of
      # interconnected objects to simulate a review app.
      #
      after(:create) do |environment|
        deployment = create(:deployment,
                            environment: environment,
                            project: environment.project,
                            sha: environment.project.commit.id)

        teardown_build = create(:ci_build, :manual,
                                name: "#{deployment.environment.name}:teardown",
                                pipeline: deployment.deployable.pipeline)

        deployment.update_column(:on_stop, teardown_build.name)
        environment.update_attribute(:deployments, [deployment])
      end
    end
  end
end