require "rails_helper"
require "rake"

RSpec.describe "correct_addresses" do
  around do |example|
    Timecop.freeze(Time.zone.local(2023, 10, 10)) do
      Singleton.__init__(FormHandler)
      example.run
    end
  end

  describe ":send_missing_addresses_lettings_csv", type: :task do
    subject(:task) { Rake::Task["correct_addresses:send_missing_addresses_lettings_csv"] }

    before do
      organisation.users.destroy_all
      Rake.application.rake_require("tasks/send_missing_addresses_csv")
      Rake::Task.define_task(:environment)
      task.reenable
    end

    context "when the rake task is run" do
      let(:organisation) { create(:organisation, name: "test organisation") }

      before do
        stub_const("EmailMissingAddressesCsvJob::MISSING_ADDRESSES_THRESHOLD", 5)
      end

      context "when org has more than 5 missing addresses and data coordinators" do
        let!(:data_coordinator) { create(:user, :data_coordinator, organisation:, email: "data_coordinator1@example.com") }
        let!(:data_coordinator2) { create(:user, :data_coordinator, organisation:, email: "data_coordinator2@example.com") }

        before do
          create(:user, :data_provider, organisation:, email: "data_provider1@example.com")
          create_list(:lettings_log, 7, :imported, startdate: Time.zone.local(2023, 9, 9), address_line1: nil, town_or_city: nil, needstype: 1, old_form_id: "form_1", owning_organisation: organisation, managing_organisation: organisation, created_by: data_coordinator)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_coordinator.id, data_coordinator2.id), organisation, "lettings", %w[missing_address], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing lettings addresses CSV for test organisation to (?:data_coordinator1@example\.com, data_coordinator2@example\.com|data_coordinator2@example\.com, data_coordinator1@example\.com)$/))
          task.invoke
        end
      end

      context "when org has 5 missing addresses and data providers only" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }
        let!(:data_provider2) { create(:user, :data_provider, organisation:, email: "data_provider4@example.com") }

        before do
          create_list(:lettings_log, 5, :imported, startdate: Time.zone.local(2023, 9, 9), address_line1: nil, town_or_city: nil, needstype: 1, old_form_id: "form_2", owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_provider.id, data_provider2.id), organisation, "lettings", %w[missing_address], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing lettings addresses CSV for test organisation to (?:data_provider3@example\.com, data_provider4@example\.com|data_provider4@example\.com, data_provider3@example\.com)$/))
          task.invoke
        end
      end

      context "when org has less than 5 missing addresses" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }

        before do
          create_list(:lettings_log, 3, :imported, startdate: Time.zone.local(2023, 9, 9), address_line1: nil, town_or_city: nil, needstype: 1, old_form_id: "form_2", owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
          create_list(:lettings_log, 2, :imported, startdate: Time.zone.local(2023, 9, 9), address_line1: nil, needstype: 1, owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
        end

        it "does not enqueue the job with organisations that is missing less addresses than threshold amount" do
          expect { task.invoke }.not_to enqueue_job(EmailMissingAddressesCsvJob)
        end
      end

      context "when org has more than 5 missing town_or_city and data coordinators" do
        let!(:data_coordinator) { create(:user, :data_coordinator, organisation:, email: "data_coordinator1@example.com") }
        let!(:data_coordinator2) { create(:user, :data_coordinator, organisation:, email: "data_coordinator2@example.com") }

        before do
          create(:user, :data_provider, organisation:, email: "data_provider1@example.com")
          create_list(:lettings_log, 7, :imported, startdate: Time.zone.local(2023, 9, 9), address_line1: "exists", town_or_city: nil, needstype: 1, old_form_id: "form_1", owning_organisation: organisation, managing_organisation: organisation, created_by: data_coordinator)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_coordinator.id, data_coordinator2.id), organisation, "lettings", %w[missing_town], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing lettings addresses CSV for test organisation to (?:data_coordinator1@example\.com, data_coordinator2@example\.com|data_coordinator2@example\.com, data_coordinator1@example\.com)$/))
          task.invoke
        end
      end

      context "when org has 5 missing town or city and data providers only" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }
        let!(:data_provider2) { create(:user, :data_provider, organisation:, email: "data_provider4@example.com") }

        before do
          create_list(:lettings_log, 5, :imported, startdate: Time.zone.local(2023, 9, 9), address_line1: "exists", town_or_city: nil, needstype: 1, old_form_id: "form_2", owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_provider.id, data_provider2.id), organisation, "lettings", %w[missing_town], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing lettings addresses CSV for test organisation to (?:data_provider3@example\.com, data_provider4@example\.com|data_provider4@example\.com, data_provider3@example\.com)$/))
          task.invoke
        end
      end

      context "when org has less than 5 missing town or city" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }

        before do
          create_list(:lettings_log, 3, :imported, startdate: Time.zone.local(2023, 9, 9), address_line1: "address", town_or_city: nil, needstype: 1, old_form_id: "form_2", owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
          create_list(:lettings_log, 2, :imported, startdate: Time.zone.local(2023, 9, 9), address_line1: "address", needstype: 1, owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
        end

        it "does not enqueue the job with organisations that is missing less town or city data than threshold amount" do
          expect { task.invoke }.not_to enqueue_job(EmailMissingAddressesCsvJob)
        end
      end

      context "when org has more than 5 wrong uprn and data coordinators" do
        let!(:data_coordinator) { create(:user, :data_coordinator, organisation:, email: "data_coordinator1@example.com") }
        let!(:data_coordinator2) { create(:user, :data_coordinator, organisation:, email: "data_coordinator2@example.com") }

        before do
          create(:user, :data_provider, organisation:, email: "data_provider1@example.com")
          create_list(:lettings_log, 7, :imported, :setup_completed, startdate: Time.zone.local(2023, 9, 9), uprn_known: 1, uprn: "123", town_or_city: "Bristol", needstype: 1, owning_organisation: organisation, managing_organisation: organisation, created_by: data_coordinator)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_coordinator.id, data_coordinator2.id), organisation, "lettings", %w[bristol_uprn], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing lettings addresses CSV for test organisation to (?:data_coordinator1@example\.com, data_coordinator2@example\.com|data_coordinator2@example\.com, data_coordinator1@example\.com)$/))
          task.invoke
        end
      end

      context "when org has 5 wrong uprn and data providers only" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }
        let!(:data_provider2) { create(:user, :data_provider, organisation:, email: "data_provider4@example.com") }

        before do
          create_list(:lettings_log, 5, :imported, :setup_completed, startdate: Time.zone.local(2023, 9, 9), uprn: "12", propcode: "12", needstype: 1, owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_provider.id, data_provider2.id), organisation, "lettings", %w[wrong_uprn], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing lettings addresses CSV for test organisation to (?:data_provider3@example\.com, data_provider4@example\.com|data_provider4@example\.com, data_provider3@example\.com)$/))
          task.invoke
        end
      end

      context "when org has less than 5 wrong uprn" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }
        let!(:data_provider2) { create(:user, :data_provider, organisation:, email: "data_provider4@example.com") }

        before do
          create_list(:lettings_log, 3, :imported, :setup_completed, startdate: Time.zone.local(2023, 9, 9), uprn: "123", town_or_city: "Bristol", needstype: 1, owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
          create_list(:lettings_log, 2, :imported, :setup_completed, startdate: Time.zone.local(2023, 9, 9), uprn: "12", tenancycode: "12", needstype: 1, owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_provider.id, data_provider2.id), organisation, "lettings", %w[wrong_uprn bristol_uprn], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing lettings addresses CSV for test organisation to (?:data_provider3@example\.com, data_provider4@example\.com|data_provider4@example\.com, data_provider3@example\.com)$/))
          task.invoke
        end
      end

      context "when org is included in skip_uprn_issue_organisations list" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }

        before do
          create_list(:lettings_log, 5, :imported, startdate: Time.zone.local(2023, 9, 9), uprn: "12", propcode: "12", needstype: 1, owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
        end

        it "does not enqueue the job" do
          expect { task.invoke(organisation.id.to_s) }.not_to enqueue_job(EmailMissingAddressesCsvJob)
        end
      end

      context "when org is included in skip_uprn_issue_organisations list but faces a different issue" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }
        let!(:data_provider2) { create(:user, :data_provider, organisation:, email: "data_provider4@example.com") }

        before do
          create_list(:lettings_log, 5, :imported, startdate: Time.zone.local(2023, 9, 9), address_line1: nil, town_or_city: nil, needstype: 1, old_form_id: "form_2", owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
          create_list(:lettings_log, 5, :imported, startdate: Time.zone.local(2023, 9, 9), uprn: "12", propcode: "12", needstype: 1, owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
        end

        it "does not enqueue the job" do
          expect { task.invoke(organisation.id.to_s) }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_provider.id, data_provider2.id), organisation, "lettings", %w[missing_address], [organisation.id])
        end
      end

      context "when skip_uprn_issue_organisations list is provided" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }
        let!(:data_provider2) { create(:user, :data_provider, organisation:, email: "data_provider4@example.com") }

        before do
          create_list(:lettings_log, 5, :imported, :setup_completed, startdate: Time.zone.local(2023, 9, 9), uprn: "12", propcode: "12", needstype: 1, owning_organisation: organisation, managing_organisation: organisation, created_by: data_provider)
        end

        it "does enqueues the job with correct skip_uprn_issue_organisations" do
          expect { task.invoke("100 400") }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_provider.id, data_provider2.id), organisation, "lettings", %w[wrong_uprn], [100, 400])
        end
      end
    end
  end

  describe ":send_missing_addresses_sales_csv", type: :task do
    subject(:task) { Rake::Task["correct_addresses:send_missing_addresses_sales_csv"] }

    before do
      organisation.users.destroy_all
      Rake.application.rake_require("tasks/send_missing_addresses_csv")
      Rake::Task.define_task(:environment)
      task.reenable
    end

    context "when the rake task is run" do
      let(:organisation) { create(:organisation, name: "test organisation") }

      before do
        stub_const("EmailMissingAddressesCsvJob::MISSING_ADDRESSES_THRESHOLD", 5)
      end

      context "when org has more than 5 missing addresses and data coordinators" do
        let!(:data_coordinator) { create(:user, :data_coordinator, organisation:, email: "data_coordinator1@example.com") }
        let!(:data_coordinator2) { create(:user, :data_coordinator, organisation:, email: "data_coordinator2@example.com") }

        before do
          create(:user, :data_provider, organisation:, email: "data_provider1@example.com")
          create_list(:sales_log, 7, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), address_line1: nil, town_or_city: nil, old_form_id: "form_1", owning_organisation: organisation, created_by: data_coordinator)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke("70 90") }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_coordinator.id, data_coordinator2.id), organisation, "sales", %w[missing_address], [70, 90])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing sales addresses CSV for test organisation to (?:data_coordinator1@example\.com, data_coordinator2@example\.com|data_coordinator2@example\.com, data_coordinator1@example\.com)$/))
          task.invoke
        end
      end

      context "when org has 5 missing addresses and data providers only" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }
        let!(:data_provider2) { create(:user, :data_provider, organisation:, email: "data_provider4@example.com") }

        before do
          create_list(:sales_log, 5, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), address_line1: nil, town_or_city: nil, old_form_id: "form_2", owning_organisation: organisation, created_by: data_provider)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_provider.id, data_provider2.id), organisation, "sales", %w[missing_address], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing sales addresses CSV for test organisation to (?:data_provider3@example\.com, data_provider4@example\.com|data_provider4@example\.com, data_provider3@example\.com)$/))
          task.invoke
        end
      end

      context "when org has less than 5 missing addresses" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }

        before do
          create_list(:sales_log, 3, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), address_line1: nil, town_or_city: nil, old_form_id: "form_2", owning_organisation: organisation, created_by: data_provider)
          create_list(:sales_log, 2, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), address_line1: nil, owning_organisation: organisation, created_by: data_provider)
        end

        it "does not enqueue the job with organisations that is missing less addresses than threshold amount" do
          expect { task.invoke }.not_to enqueue_job(EmailMissingAddressesCsvJob)
        end
      end

      context "when org has more than 5 missing town_or_city and data coordinators" do
        let!(:data_coordinator) { create(:user, :data_coordinator, organisation:, email: "data_coordinator1@example.com") }
        let!(:data_coordinator2) { create(:user, :data_coordinator, organisation:, email: "data_coordinator2@example.com") }

        before do
          create(:user, :data_provider, organisation:, email: "data_provider1@example.com")
          create_list(:sales_log, 7, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), address_line1: "exists", town_or_city: nil, old_form_id: "form_1", owning_organisation: organisation, created_by: data_coordinator)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_coordinator.id, data_coordinator2.id), organisation, "sales", %w[missing_town], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing sales addresses CSV for test organisation to (?:data_coordinator1@example\.com, data_coordinator2@example\.com|data_coordinator2@example\.com, data_coordinator1@example\.com)$/))
          task.invoke
        end
      end

      context "when org has 5 missing town or city and data providers only" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }
        let!(:data_provider2) { create(:user, :data_provider, organisation:, email: "data_provider4@example.com") }

        before do
          create_list(:sales_log, 5, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), address_line1: "exists", town_or_city: nil, old_form_id: "form_2", owning_organisation: organisation, created_by: data_provider)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_provider.id, data_provider2.id), organisation, "sales", %w[missing_town], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing sales addresses CSV for test organisation to (?:data_provider3@example\.com, data_provider4@example\.com|data_provider4@example\.com, data_provider3@example\.com)$/))
          task.invoke
        end
      end

      context "when org has less than 5 missing town or city" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }

        before do
          create_list(:sales_log, 3, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), address_line1: "address", town_or_city: nil, old_form_id: "form_2", owning_organisation: organisation, created_by: data_provider)
          create_list(:sales_log, 2, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), address_line1: "address", owning_organisation: organisation, created_by: data_provider)
        end

        it "does not enqueue the job with organisations that is missing less town or city data than threshold amount" do
          expect { task.invoke }.not_to enqueue_job(EmailMissingAddressesCsvJob)
        end
      end

      context "when org has more than 5 wrong uprn and data coordinators" do
        let!(:data_coordinator) { create(:user, :data_coordinator, organisation:, email: "data_coordinator1@example.com") }
        let!(:data_coordinator2) { create(:user, :data_coordinator, organisation:, email: "data_coordinator2@example.com") }

        before do
          create(:user, :data_provider, organisation:, email: "data_provider1@example.com")
          create_list(:sales_log, 7, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), uprn_known: 1, uprn: "123", town_or_city: "Bristol", owning_organisation: organisation, created_by: data_coordinator)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_coordinator.id, data_coordinator2.id), organisation, "sales", %w[wrong_uprn], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing sales addresses CSV for test organisation to (?:data_coordinator1@example\.com, data_coordinator2@example\.com|data_coordinator2@example\.com, data_coordinator1@example\.com)$/))
          task.invoke
        end
      end

      context "when org has 5 wrong uprn and data providers only" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }
        let!(:data_provider2) { create(:user, :data_provider, organisation:, email: "data_provider4@example.com") }

        before do
          create_list(:sales_log, 5, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), uprn_known: 1, uprn: "12", purchid: "12", owning_organisation: organisation, created_by: data_provider)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_provider.id, data_provider2.id), organisation, "sales", %w[wrong_uprn], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing sales addresses CSV for test organisation to (?:data_provider3@example\.com, data_provider4@example\.com|data_provider4@example\.com, data_provider3@example\.com)$/))
          task.invoke
        end
      end

      context "when org has less than 5 wrong uprn" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }
        let!(:data_provider2) { create(:user, :data_provider, organisation:, email: "data_provider4@example.com") }

        before do
          create_list(:sales_log, 3, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), uprn_known: 1, uprn: "123", town_or_city: "Bristol", owning_organisation: organisation, created_by: data_provider)
          create_list(:sales_log, 2, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), uprn_known: 1, uprn: "12", purchid: "12", owning_organisation: organisation, created_by: data_provider)
        end

        it "enqueues the job with correct organisations" do
          expect { task.invoke }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_provider.id, data_provider2.id), organisation, "sales", %w[wrong_uprn], [])
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with(match(/^Sending missing sales addresses CSV for test organisation to (?:data_provider3@example\.com, data_provider4@example\.com|data_provider4@example\.com, data_provider3@example\.com)$/))
          task.invoke
        end
      end

      context "when org is included in skip_uprn_issue_organisations list" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }

        before do
          create_list(:sales_log, 5, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), uprn_known: 1, uprn: "12", purchid: "12", owning_organisation: organisation, created_by: data_provider)
        end

        it "does not enqueue the job" do
          expect { task.invoke("#{organisation.id} 4") }.not_to enqueue_job(EmailMissingAddressesCsvJob)
        end
      end

      context "when skip_uprn_issue_organisations list is provided" do
        let!(:data_provider) { create(:user, :data_provider, organisation:, email: "data_provider3@example.com") }
        let!(:data_provider2) { create(:user, :data_provider, organisation:, email: "data_provider4@example.com") }

        before do
          create_list(:sales_log, 5, :completed, :imported, saledate: Time.zone.local(2023, 9, 9), uprn_known: 1, uprn: "12", purchid: "12", owning_organisation: organisation, created_by: data_provider)
        end

        it "does enqueues the job with correct skip_uprn_issue_organisations" do
          expect { task.invoke("100 400") }.to enqueue_job(EmailMissingAddressesCsvJob).with(include(data_provider.id, data_provider2.id), organisation, "sales", %w[wrong_uprn], [100, 400])
        end
      end
    end
  end

  describe ":create_lettings_addresses_csv", type: :task do
    subject(:task) { Rake::Task["correct_addresses:create_lettings_addresses_csv"] }

    before do
      organisation.users.destroy_all
      Rake.application.rake_require("tasks/send_missing_addresses_csv")
      Rake::Task.define_task(:environment)
      task.reenable
    end

    context "when the rake task is run" do
      let(:organisation) { create(:organisation, name: "test organisation") }

      context "and organisation ID is provided" do
        it "enqueues the job with correct organisation" do
          expect { task.invoke(organisation.id) }.to enqueue_job(CreateAddressesCsvJob).with(organisation, "lettings")
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with("Creating lettings addresses CSV for test organisation")
          task.invoke(organisation.id)
        end
      end

      context "when organisation with given ID cannot be found" do
        it "prints out error" do
          expect(Rails.logger).to receive(:error).with("Organisation with ID fake not found")
          task.invoke("fake")
        end
      end

      context "when organisation ID is not given" do
        it "raises an error" do
          expect { task.invoke }.to raise_error(RuntimeError, "Usage: rake correct_addresses:create_lettings_addresses_csv['organisation_id']")
        end
      end
    end
  end

  describe ":create_sales_addresses_csv", type: :task do
    subject(:task) { Rake::Task["correct_addresses:create_sales_addresses_csv"] }

    before do
      organisation.users.destroy_all
      Rake.application.rake_require("tasks/send_missing_addresses_csv")
      Rake::Task.define_task(:environment)
      task.reenable
    end

    context "when the rake task is run" do
      let(:organisation) { create(:organisation, name: "test organisation") }

      context "and organisation ID is provided" do
        it "enqueues the job with correct organisation" do
          expect { task.invoke(organisation.id) }.to enqueue_job(CreateAddressesCsvJob).with(organisation, "sales")
        end

        it "prints out the jobs enqueued" do
          expect(Rails.logger).to receive(:info).with(nil)
          expect(Rails.logger).to receive(:info).with("Creating sales addresses CSV for test organisation")
          task.invoke(organisation.id)
        end
      end

      context "when organisation with given ID cannot be found" do
        it "prints out error" do
          expect(Rails.logger).to receive(:error).with("Organisation with ID fake not found")
          task.invoke("fake")
        end
      end

      context "when organisation ID is not given" do
        it "raises an error" do
          expect { task.invoke }.to raise_error(RuntimeError, "Usage: rake correct_addresses:create_sales_addresses_csv['organisation_id']")
        end
      end
    end
  end
end