nairobi/spec/factories/users.rb

FactoryBot.define do
  factory :user do
    sequence(:username) { |n| "Manuela#{n}" }
    sequence(:email)    { |n| "manuela#{n}@consul.dev" }

    password            { "judgmentday" }
    terms_of_service    { "1" }
    confirmed_at        { Time.current }
    date_of_birth       { 20.years.ago }
    public_activity     { true }

    trait :incomplete_verification do
      after :create do |user|
        create(:failed_census_call, user: user)
      end
    end

    trait :level_two do
      residence_verified_at { Time.current }
      unconfirmed_phone { "611111111" }
      confirmed_phone { "611111111" }
      sms_confirmation_code { "1234" }
      document_type { "1" }
      document_number
      date_of_birth { Date.new(1980, 12, 31) }
      gender { "female" }
      geozone
    end

    trait :level_three do
      verified_at { Time.current }
      document_type { "1" }
      document_number
    end

    trait :hidden do
      hidden_at { Time.current }
    end

    trait :with_confirmed_hide do
      confirmed_hide_at { Time.current }
    end

    trait :verified do
      residence_verified_at { Time.current }
      verified_at { Time.current }
    end

    trait :in_census do
      document_number { "12345678Z" }
      document_type { "1" }
      verified_at { Time.current }
    end

    trait :with_proposal do
      after(:create) { |user| create(:proposal, author: user) }
    end

    trait :with_debate do
      after(:create) { |user| create(:debate, author: user) }
    end

    trait :with_comment do
      after(:create) { |user| create(:comment, author: user) }
    end

    transient do
      votables { [] }
      followables { [] }
      ballot_lines { [] }
    end

    after(:create) do |user, evaluator|
      evaluator.votables.each { |votable| create(:vote, votable: votable, voter: user) }
      evaluator.followables.each { |followable| create(:follow, followable: followable, user: user) }

      if evaluator.ballot_lines.any?
        ballot = create(:budget_ballot, budget: evaluator.ballot_lines.first.budget.reload, user: user)

        evaluator.ballot_lines.each do |investment|
          create(:budget_ballot_line, investment: investment, ballot: ballot)
        end
      end
    end
  end

  factory :identity do
    user { nil }
    provider { "Twitter" }
    uid { "MyString" }
  end

  factory :administrator do
    user
  end

  factory :moderator do
    user
  end

  factory :valuator do
    user
  end

  factory :manager do
    user
  end

  factory :sdg_manager, class: "SDG::Manager" do
    user
  end

  factory :poll_officer, class: "Poll::Officer" do
    user { association(:user, username: name) }

    transient do
      sequence(:name) { |n| "Officer #{n}" }
      polls { [] }
    end

    after(:create) do |officer, evaluator|
      evaluator.polls.each do |poll|
        create(:poll_officer_assignment, poll: poll, officer: officer)
      end
    end
  end

  factory :follow do
    user

    trait :followed_proposal do
      followable factory: :proposal
    end

    trait :followed_investment do
      followable factory: :budget_investment
    end
  end

  factory :direct_message do
    title    { "Hey" }
    body     { "How are You doing?" }
    sender factory: :user
    receiver factory: :user
  end
end