Given(/^the quick survey test pages are installed$/) do
  api.create_page 'MediaWiki:anne-survey-question', 'What is the best drink?'
  api.create_page 'MediaWiki:anne-survey-description', 'Help us nourish you!'
  api.create_page 'MediaWiki:anne-survey-answer-one', 'Milk'
  api.create_page 'MediaWiki:anne-survey-answer-two', 'Beer'
  api.create_page 'MediaWiki:anne-survey-answer-three', 'Water'
  api.create_page 'MediaWiki:anne-survey-answer-four', 'Ginger beer'

  api.create_page 'Quick survey test page without infobox nor image',
                  File.read(File.join(__dir__, '../../../samples/no_infobox_or_image.wikitext'))

  api.create_page 'Quick survey test page without lead section',
                  File.read(File.join(__dir__, '../../../samples/no_lead_section.wikitext'))

  api.create_page 'Quick survey test page with no image infobox',
                  File.read(File.join(__dir__, '../../../samples/infobox_no_image.wikitext'))

  api.create_page 'Quick survey test page stub',
                  File.read(File.join(__dir__, '../../../samples/stub.wikitext'))

  api.create_page 'Quick survey test page with image no infobox',
                  File.read(File.join(__dir__, '../../../samples/image_no_infobox.wikitext'))

  api.create_page 'Quick survey test page with infobox with image',
                  File.read(File.join(__dir__, '../../../samples/infobox_and_image.wikitext'))
end

Given(/^I am on the "(.*?)" page with the quick survey flag enabled$/) do |arg1|
  visit(ArticlePage,
        using_params: { article_name: arg1, query_string: '?quicksurvey=true' })
end

Given(/^I am on the "(.*?)" page and I see the drinks survey$/) do |arg1|
  visit(ArticlePage,
        using_params: {
          article_name: arg1,
          query_string: '?quicksurvey=internal-survey-drink-survey'
        }
  )
end

Given(/^I am on the "(.*?)" page with the internal quick survey flag enabled$/)\
 do |arg1|
   visit(
     ArticlePage,
     using_params: {
       article_name: arg1,
       query_string: '?quicksurvey=internal-survey-internal example survey'
     }
   )
 end

Given(/^I am on the "(.*?)" page with the external quick survey flag enabled$/)\
 do |arg1|
   visit(
     ArticlePage,
     using_params: {
       article_name: arg1,
       query_string: '?quicksurvey=external-survey-external example survey'
     }
   )
 end

Given(/^I am on the "(.*?)" page$/) do |arg1|
  visit(ArticlePage, using_params: { article_name: arg1 })
end

Then(/^I should see the survey$/) do
  expect(on(ArticlePage).survey_element.when_present).to be_visible
end

Then(/^the page has fully loaded$/) do
  on(ArticlePage) do |page|
    page.wait_until do
      # Wait for async JS to hijack standard link
      script = 'return mw && mw.loader && '\
        'mw.loader.getState("ext.quicksurveys.init") === "ready";'
      page.execute_script(script)
    end
  end
end

Then(/^the survey code has fully loaded$/) do
  on(ArticlePage) do |page|
    page.wait_until do
      # Wait for async JS to load and render the survey if applicable
      script = 'return mw && mw.loader && '\
        'mw.loader.getState("ext.quicksurveys.views") === "ready";'
      page.execute_script(script)
    end
  end
end



Then(/^I should not see the survey$/) do
  expect(on(ArticlePage).survey_element).to_not be_visible
end

Given(/^I have dismissed survey "(.*?)"$/)  do |arg1|
  step 'I am on the "Main Page" page'
  browser.execute_script(
    'localStorage.setItem("ext-quicksurvey-' + arg1 + '", "~");')
end

Given(/^I am not bucketed for "(.*?)"$/)  do |arg1|
  step 'I am on the "Main Page" page'
  case arg1
  when 'internal-example-survey'
    key = '63e9d6d760750eaa'
  when 'external-example-survey'
    key = '2c0cdc37f48b1b0e'
  else
    key = ''
  end
  browser.execute_script(
    'localStorage.setItem("ext-quicksurvey-' + arg1 + '", "' + key + '")')
end

Given(/^I am bucketed for "(.*?)"$/) do |arg1|
  step 'I am on the "Main Page" page'
  case arg1
  when 'internal-example-survey'
    key = '2c0cdc37f48b1b0e'
  when 'external-example-survey'
    key = '63e9d6d760750eaa'
  else
    key = ''
  end
  browser.execute_script(
    'localStorage.setItem("ext-quicksurvey-' + arg1 + '", "' + key + '")')
end