require 'rails_helper'

describe SurveyResponsesDataLoader do
  let(:path_to_teacher_responses) { Rails.root.join('spec', 'fixtures', 'test_2020-21_teacher_survey_responses.csv') }
  let(:path_to_student_responses) { Rails.root.join('spec', 'fixtures', 'test_2020-21_student_survey_responses.csv') }
  let(:path_to_butler_student_responses) do
    Rails.root.join('spec', 'fixtures', 'test_2022-23_butler_student_survey_responses.csv')
  end

  let(:ay_2020_21) { create(:academic_year, range: '2020-21') }
  let(:ay_2022_23) { create(:academic_year, range: '2022-23') }

  let(:school) { create(:school, name: 'Lee Elementary School', slug: 'lee-elementary-school', dese_id: 1_500_025) }
  let(:lowell) { create(:district, name: 'Lowell', slug: 'lowell') }
  let(:second_school) do
    create(:school, name: 'Lee Middle High School', slug: 'lee-middle-high-school', dese_id: 1_500_505,
                    district: lowell)
  end
  let(:butler_school) do
    create(:school, name: 'Butler Elementary School', slug: 'butler-elementary-school', dese_id: 1_600_310,
                    district: lowell)
  end

  let(:t_pcom_q3) { create(:survey_item, survey_item_id: 't-pcom-q3') }
  let(:t_pcom_q2) { create(:survey_item, survey_item_id: 't-pcom-q2') }
  let(:t_coll_q1) { create(:survey_item, survey_item_id: 't-coll-q1') }
  let(:t_coll_q2) { create(:survey_item, survey_item_id: 't-coll-q2') }
  let(:t_coll_q3) { create(:survey_item, survey_item_id: 't-coll-q3') }
  let(:t_sach_q1) { create(:survey_item, survey_item_id: 't-sach-q1') }
  let(:t_sach_q2) { create(:survey_item, survey_item_id: 't-sach-q2') }
  let(:t_sach_q3) { create(:survey_item, survey_item_id: 't-sach-q3') }

  let(:s_phys_q1) { create(:survey_item, survey_item_id: 's-phys-q1') }
  let(:s_phys_q2) { create(:survey_item, survey_item_id: 's-phys-q2') }
  let(:s_phys_q3) { create(:survey_item, survey_item_id: 's-phys-q3') }
  let(:s_phys_q4) { create(:survey_item, survey_item_id: 's-phys-q4') }
  let(:s_vale_q1) { create(:survey_item, survey_item_id: 's-phys-q1') }
  let(:s_vale_q2) { create(:survey_item, survey_item_id: 's-phys-q2') }
  let(:s_vale_q3) { create(:survey_item, survey_item_id: 's-phys-q3') }
  let(:s_vale_q4) { create(:survey_item, survey_item_id: 's-phys-q4') }
  let(:s_acst_q1) { create(:survey_item, survey_item_id: 's-acst-q1') }
  let(:s_acst_q2) { create(:survey_item, survey_item_id: 's-acst-q2') }
  let(:s_acst_q3) { create(:survey_item, survey_item_id: 's-acst-q3') }
  let(:s_acst_q4) { create(:survey_item, survey_item_id: 's-acst-q4') }
  let(:s_emsa_q1) { create(:survey_item, survey_item_id: 's-emsa-q1') }
  let(:s_emsa_q2) { create(:survey_item, survey_item_id: 's-emsa-q2') }
  let(:s_emsa_q3) { create(:survey_item, survey_item_id: 's-emsa-q3') }

  let(:female) { create(:gender, qualtrics_code: 1) }
  let(:male) { create(:gender, qualtrics_code: 2) }
  let(:another_gender) { create(:gender, qualtrics_code: 3) }
  let(:non_binary) { create(:gender, qualtrics_code: 4) }
  let(:unknown_gender) { create(:gender, qualtrics_code: 99) }
  let(:low_income) { create(:income, designation: 'Economically Disadvantaged – Y') }
  let(:high_income) { create(:income, designation: 'Economically Disadvantaged – N') }
  let(:unknown_income) { create(:income, designation: 'Unknown') }

  let(:setup) do
    ay_2020_21
    ay_2022_23
    school
    second_school
    butler_school
    t_pcom_q3
    t_pcom_q2
    t_coll_q1
    t_coll_q2
    t_coll_q3
    t_sach_q1
    t_sach_q2
    t_sach_q3
    s_phys_q1
    s_phys_q2
    s_phys_q3
    s_phys_q4
    s_vale_q1
    s_vale_q2
    s_vale_q3
    s_vale_q4
    s_acst_q1
    s_acst_q2
    s_acst_q3
    s_acst_q4
    s_emsa_q1
    s_emsa_q2
    s_emsa_q3
    female
    male
    another_gender
    non_binary
    unknown_gender
    low_income
    high_income
    unknown_income
  end

  before :each do
    setup
  end

  describe 'loading teacher survey responses' do
    before do
      SurveyResponsesDataLoader.load_data filepath: path_to_teacher_responses
    end

    it 'ensures teacher responses load correctly' do
      assigns_academic_year_to_survey_item_responses
      assigns_school_to_the_survey_item_responses
      assigns_recorded_date_to_teacher_responses
      loads_survey_item_responses_for_a_given_survey_response
      loads_all_survey_item_responses_for_a_given_survey_item
      captures_likert_scores_for_survey_item_responses
      is_idempotent
    end
  end

  describe 'student survey responses' do
    before do
      SurveyResponsesDataLoader.load_data filepath: path_to_student_responses
    end

    it 'ensures student responses load correctly' do
      assigns_academic_year_to_student_survey_item_responses
      assigns_school_to_student_survey_item_responses
      assigns_recorded_date_to_student_responses
      loads_student_survey_item_response_values
      student_survey_item_response_count_matches_expected
      captures_likert_scores_for_student_survey_item_responses
      assigns_grade_level_to_responses
      assigns_gender_to_responses
      assigns_income_to_responses
      is_idempotent_for_students
    end

    context 'when updating student survey responses from another csv file' do
      before :each do
        SurveyResponsesDataLoader.load_data filepath: Rails.root.join('spec', 'fixtures',
                                                                      'secondary_test_2020-21_student_survey_responses.csv')
      end
      it 'updates the likert score to the score on the new csv file' do
        s_emsa_q1 = SurveyItem.find_by_survey_item_id 's-emsa-q1'
        expect(SurveyItemResponse.where(response_id: 'student_survey_response_3',
                                        survey_item: s_emsa_q1).first.likert_score).to eq 1
        expect(SurveyItemResponse.where(response_id: 'student_survey_response_4',
                                        survey_item: s_emsa_q1).first.likert_score).to eq 1
        expect(SurveyItemResponse.where(response_id: 'student_survey_response_5',
                                        survey_item: s_emsa_q1).first.likert_score).to eq 1

        expect(SurveyItemResponse.where(response_id: 'student_survey_response_5',
                                        survey_item: s_acst_q3).first.likert_score).to eq 4
      end
    end
  end
  # figure out why this is failing
  describe 'when using Lowell rules to skip rows in the csv file' do
    before :each do
      SurveyResponsesDataLoader.load_data filepath: path_to_student_responses,
                                          rules: [Rule::SkipNonLowellSchools]
    end

    it 'rejects any non-lowell school' do
      expect(SurveyItemResponse.where(response_id: 'student_survey_response_1').count).to eq 0
      expect(SurveyItemResponse.count).to eq 69
    end

    it 'loads the correct number of responses for lowell schools' do
      expect(SurveyItemResponse.where(response_id: 'student_survey_response_2').count).to eq 0
      expect(SurveyItemResponse.where(response_id: 'student_survey_response_3').count).to eq 12
      expect(SurveyItemResponse.where(response_id: 'student_survey_response_4').count).to eq 15
      expect(SurveyItemResponse.where(response_id: 'student_survey_response_5').count).to eq 14
    end

    context 'when loading 22-23 butler survey responses' do
      before :each do
        SurveyResponsesDataLoader.load_data filepath: path_to_butler_student_responses,
                                            rules: [Rule::SkipNonLowellSchools]
      end

      it 'loads all the responses for Butler' do
        expect(SurveyItemResponse.where(school: butler_school).count).to eq 56
      end

      it 'blank entries for grade get loaded as nils, not zero values' do
        expect(SurveyItemResponse.where(response_id: 'butler_student_survey_response_1').first.grade).to eq 7
        expect(SurveyItemResponse.where(response_id: 'butler_student_survey_response_2').first.grade).to eq 7
        expect(SurveyItemResponse.where(response_id: 'butler_student_survey_response_3').first.grade).to eq 7
        expect(SurveyItemResponse.where(response_id: 'butler_student_survey_response_4').first.grade).to eq 5
        expect(SurveyItemResponse.where(response_id: 'butler_student_survey_response_5').first.grade).to eq 7
        expect(SurveyItemResponse.where(response_id: 'butler_student_survey_response_6').first.grade).to eq 6
        expect(SurveyItemResponse.where(response_id: 'butler_student_survey_response_7').first.grade).to eq nil
        expect(SurveyItemResponse.where(response_id: 'butler_student_survey_response_8').first.grade).to eq 0
      end
    end
  end
end

def assigns_academic_year_to_survey_item_responses
  expect(SurveyItemResponse.find_by_response_id('teacher_survey_response_1').academic_year).to eq ay_2020_21
end

def assigns_school_to_the_survey_item_responses
  expect(SurveyItemResponse.find_by_response_id('teacher_survey_response_1').school).to eq school
end

def loads_survey_item_responses_for_a_given_survey_response
  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_1').count).to eq 5
  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_2').count).to eq 0
  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_3').count).to eq 8
  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_4').count).to eq 8
  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_5').count).to eq 8
end

def loads_all_survey_item_responses_for_a_given_survey_item
  expect(SurveyItemResponse.where(survey_item: t_pcom_q2).count).to eq 3
  expect(SurveyItemResponse.where(survey_item: t_pcom_q3).count).to eq 4
end

def captures_likert_scores_for_survey_item_responses
  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_1').where(survey_item: t_pcom_q2)).to be_empty
  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_1').where(survey_item: t_pcom_q3).first.likert_score).to eq 3

  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_2').where(survey_item: t_pcom_q2)).to be_empty
  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_2').where(survey_item: t_pcom_q3)).to be_empty

  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_3').where(survey_item: t_pcom_q2).first.likert_score).to eq 5
  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_3').where(survey_item: t_pcom_q3).first.likert_score).to eq 5

  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_4').where(survey_item: t_pcom_q2).first.likert_score).to eq 4
  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_4').where(survey_item: t_pcom_q3).first.likert_score).to eq 4

  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_5').where(survey_item: t_pcom_q2).first.likert_score).to eq 2
  expect(SurveyItemResponse.where(response_id: 'teacher_survey_response_5').where(survey_item: t_pcom_q3).first.likert_score).to eq 4
end

def is_idempotent
  number_of_survey_item_responses = SurveyItemResponse.count

  SurveyResponsesDataLoader.load_data filepath: path_to_teacher_responses

  expect(SurveyItemResponse.count).to eq number_of_survey_item_responses
end

def assigns_academic_year_to_student_survey_item_responses
  expect(SurveyItemResponse.find_by_response_id('student_survey_response_3').academic_year).to eq ay_2020_21
end

def assigns_school_to_student_survey_item_responses
  expect(SurveyItemResponse.find_by_response_id('student_survey_response_3').school).to eq second_school
end

def loads_student_survey_item_response_values
  expect(SurveyItemResponse.where(response_id: 'student_survey_response_1').count).to eq 3
  expect(SurveyItemResponse.where(response_id: 'student_survey_response_2').count).to eq 0
  expect(SurveyItemResponse.where(response_id: 'student_survey_response_3').count).to eq 12
  expect(SurveyItemResponse.where(response_id: 'student_survey_response_4').count).to eq 15
  expect(SurveyItemResponse.where(response_id: 'student_survey_response_5').count).to eq 14
end

def student_survey_item_response_count_matches_expected
  expect(SurveyItemResponse.where(survey_item: s_phys_q1).count).to eq 6
  expect(SurveyItemResponse.where(survey_item: s_phys_q2).count).to eq 5
end

def captures_likert_scores_for_student_survey_item_responses
  expect(SurveyItemResponse.where(response_id: 'student_survey_response_1').where(survey_item: s_phys_q1).first.likert_score).to eq 3
  expect(SurveyItemResponse.where(response_id: 'student_survey_response_1').where(survey_item: s_phys_q2)).to be_empty

  expect(SurveyItemResponse.where(response_id: 'student_survey_response_2').where(survey_item: s_phys_q1)).to be_empty
  expect(SurveyItemResponse.where(response_id: 'student_survey_response_2').where(survey_item: s_phys_q2)).to be_empty

  expect(SurveyItemResponse.where(response_id: 'student_survey_response_3').where(survey_item: s_phys_q1).first.likert_score).to eq 1
  expect(SurveyItemResponse.where(response_id: 'student_survey_response_3').where(survey_item: s_phys_q2).first.likert_score).to eq 3

  expect(SurveyItemResponse.where(response_id: 'student_survey_response_4').where(survey_item: s_phys_q1).first.likert_score).to eq 1
  expect(SurveyItemResponse.where(response_id: 'student_survey_response_4').where(survey_item: s_phys_q2).first.likert_score).to eq 1

  expect(SurveyItemResponse.where(response_id: 'student_survey_response_5').where(survey_item: s_phys_q1).first.likert_score).to eq 1
  expect(SurveyItemResponse.where(response_id: 'student_survey_response_5').where(survey_item: s_phys_q2).first.likert_score).to eq 2
end

def is_idempotent_for_students
  number_of_survey_item_responses = SurveyItemResponse.count

  SurveyResponsesDataLoader.load_data filepath: path_to_student_responses

  expect(SurveyItemResponse.count).to eq number_of_survey_item_responses
end

def assigns_grade_level_to_responses
  results = { 'student_survey_response_1' => 11,
              'student_survey_response_3' => 8,
              'student_survey_response_4' => 8,
              'student_survey_response_5' => 7,
              'student_survey_response_6' => 3,
              'student_survey_response_7' => 4 }
  results.each do |key, value|
    expect(SurveyItemResponse.where(response_id: key).all? do |response|
      response.grade == value
    end).to eq true
  end
end

def assigns_gender_to_responses
  results = { 'student_survey_response_1' => female,
              'student_survey_response_3' => male,
              'student_survey_response_4' => non_binary,
              'student_survey_response_5' => non_binary,
              'student_survey_response_6' => unknown_gender,
              'student_survey_response_7' => unknown_gender }

  results.each do |key, value|
    expect(SurveyItemResponse.where(response_id: key).first.gender).to eq value
  end
end

def assigns_recorded_date_to_student_responses
  results = { 'student_survey_response_1' => '2020-09-30T18:48:50',
              'student_survey_response_3' => '2021-03-31T09:59:02',
              'student_survey_response_4' => '2021-03-31T10:00:17',
              'student_survey_response_5' => '2021-03-31T10:01:36',
              'student_survey_response_6' => '2021-03-31T10:01:37',
              'student_survey_response_7' => '2021-03-31T10:01:38' }
  results.each do |key, value|
    expect(SurveyItemResponse.find_by_response_id(key).recorded_date).to eq Date.parse(value)
  end
end

def assigns_recorded_date_to_teacher_responses
  results = { 'teacher_survey_response_1' => '2020-10-16 11:09:03',
              'teacher_survey_response_3' => '2020-12-06 8:36:52',
              'teacher_survey_response_4' => '2020-12-06 8:51:25',
              'teacher_survey_response_5' => '2020-12-06 8:55:58' }
  results.each do |key, value|
    expect(SurveyItemResponse.find_by_response_id(key).recorded_date).to eq Date.parse(value)
  end
end

def assigns_income_to_responses
  results = { 'student_survey_response_1' => low_income,
              'student_survey_response_3' => low_income,
              'student_survey_response_4' => unknown_income,
              'student_survey_response_5' => low_income,
              'student_survey_response_6' => high_income,
              'student_survey_response_7' => low_income }

  results.each do |key, value|
    expect(SurveyItemResponse.where(response_id: key).first.income).to eq value
  end
end