require 'minitest/autorun'
require 'capybara'
require 'capybara-webkit'

ENV['DISPLAY'] ="localhost:1.0"

Capybara.register_driver :webkit do |app|
  Capybara::Webkit::Driver.new(app).tap{|d| d.browser.ignore_ssl_errors}
end
Capybara.default_driver = :webkit
Capybara.default_max_wait_time = 20
Capybara.javascript_driver = :webkit
Capybara.run_server = false
Capybara.app_host = "http://localhost:8088" 

begin
  puts "Service URI is: http://localhost:8088"
rescue
  puts "Unable to start service."
  exit
end

class LazarWebTest < MiniTest::Test
  
  def self.test_order
    :sorted
  end
  
  include Capybara::DSL

  def test_00_xsetup
    `Xvfb :1 -screen 0 1024x768x16 2>/dev/null &`
    sleep 2
  end

  def test_01_visit
    visit('/predict')
    assert page.has_content?('Lazar Toxicity Predictions')
    assert page.has_content?("Draw a chemical structure")
    assert page.has_content?("enter")
    assert page.has_content?("upload")
    assert page.has_content?("Select one or more endpoints")
    assert page.has_content?("Acute toxicity")
    assert page.has_content?("Fathead minnow")
    assert page.has_content?("Carcinogenicity")
    assert page.has_content?("Rat")
    assert page.has_content?("Rodents (multiple species/sites)")
    assert page.has_content?("Mouse")
    assert page.has_content?("Maximum Recommended Daily Dose")
    assert page.has_content?("Human")
    assert page.has_content?("Predict")
  end

  def test_02_insert_wrong_smiles
    visit('/')
    page.fill_in 'identifier', :with => "blahblah"
    check('selection[Rat]')
    first(:button, '>>').click
    assert page.has_content?('Attention')
  end

  def test_03_check_all_links_exists
    visit('/')
    links = ["Details | Validation", "SMILES", "toxicology gmbh 2004 - #{Time.now.year.to_s}"]
    links.each{|l| assert page.has_link?(l), "true"}
  end

  def test_04_model_details
    visit("/")
    details = page.all('a', :text => 'Details | Validation')
    details[0].click
    assert page.has_content?('Model:')
    assert page.has_content?('Source: http://www.epa.gov/comptox/dsstox/sdf_epafhm.html')
    assert page.has_content?('Algorithm: LAZAR')
    assert page.has_content?('Type: regression')
    assert page.has_content?('Training dataset: EPAFHM.csv')
    assert page.has_content?('Training compounds: 617')
    assert page.has_content?('Validation:')
    assert page.has_content?('Num folds: 10')
    details[0].click
    #
    details[1].click
    assert page.has_content?('Model:')
    assert page.has_content?('Source: http://www.epa.gov/ncct/dsstox/sdf_cpdbas.html')
    assert page.has_content?('Algorithm: LAZAR')
    assert page.has_content?('Type: classification')
    assert page.has_content?('Training dataset: DSSTox_Carcinogenic_Potency_DBS_Rat.csv')
    assert page.has_content?('Training compounds: 1195')
    assert page.has_content?('Validation:')
    assert page.has_content?('Num folds: 10')
    details[1].click
    #
    details[2].click
    assert page.has_content?('Model:')
    assert page.has_content?('Source: http://www.epa.gov/ncct/dsstox/sdf_cpdbas.html')
    assert page.has_content?('Algorithm: LAZAR')
    assert page.has_content?('Type: classification')
    assert page.has_content?('Training dataset: DSSTox_Carcinogenic_Potency_DBS_MultiCellCall.csv')
    assert page.has_content?('Training compounds: 1116')
    assert page.has_content?('Validation:')
    assert page.has_content?('Num folds: 10')
    details[2].click
    #
    details[3].click
    assert page.has_content?('Model:')
    assert page.has_content?('Source: http://www.epa.gov/ncct/dsstox/sdf_cpdbas.html')
    assert page.has_content?('Algorithm: LAZAR')
    assert page.has_content?('Type: classification')
    assert page.has_content?('Training dataset: DSSTox_Carcinogenic_Potency_DBS_Mouse.csv')
    assert page.has_content?('Training compounds: 973')
    assert page.has_content?('Validation:')
    assert page.has_content?('Num folds: 10')
    details[3].click
    #
    details[4].click
    assert page.has_content?('Model:')
    assert page.has_content?('Source: http://www.epa.gov/comptox/dsstox/sdf_fdamdd.html')
    assert page.has_content?('Algorithm: LAZAR')
    assert page.has_content?('Type: regression')
    assert page.has_content?('Training dataset: FDA_v3b_Maximum_Recommended_Daily_Dose_mmol.csv')
    assert page.has_content?('Training compounds: 1216')
    assert page.has_content?('Validation:')
    assert page.has_content?('Num folds: 10')
    details[4].click
  end 

  def test_05_predict
    visit('/')
    page.fill_in('identifier', :with => "NNc1ccccc1")
    check('selection[Rat]')
    first(:button, '>>').click
    assert page.has_content?('Carcinogenicity (Rat)'), "true"
    assert page.has_content?('Type: Classification'), "true"
    assert page.has_content?('Prediction: active'), "true"
    assert page.has_content?('Confidence: 0.019'), "true"
    assert page.has_content?('Neighbors'), "true"
    assert page.has_content?('Compound'), "true"
    assert page.has_content?('Measured Activity'), "true"
    assert page.has_content?('Similarity'), "true"
=begin
    assert page.has_link?('Significant fragments'), "true"
    assert page.has_link?('v'), "true"
     open 'significant fragments' view
    find_link('linkPredictionSf').click
    sleep 5
    within_frame('details_overview') do
      assert page.has_content?('Predominantly in compounds with activity "inactive"'), "true"
      assert page.has_content?('Predominantly in compounds with activity "active"'), "true"
      assert page.has_content?('p value'), "true"
      # inactive
      assert page.has_content?('[#6&a]:[#6&a]:[#6&a]:[#6&a]:[#6&a]-[#7&A]'), "true"
      assert page.has_content?('0.98674'), "true"
      assert page.has_content?('[#6&a]:[#6&a](-[#7&A])(:[#6&a]:[#6&a]:[#6&a])'), "true"
      assert page.has_content?('0.97699'), "true"
      assert page.has_content?('[#6&a]:[#6&a](-[#7&A])(:[#6&a]:[#6&a])'), "true"
      assert page.has_content?('0.97699'), "true"
      assert page.has_content?('[#6&a]:[#6&a](-[#7&A])(:[#6&a])'), "true"
      assert page.has_content?('0.97699'), "true"
      assert page.has_content?('[#6&a]:[#6&a]'), "true"
      assert page.has_content?('0.99605'), "true"
      assert page.has_content?('[#6&a]:[#6&a]:[#6&a]:[#6&a]'), "true"
      assert page.has_content?('0.99791'), "true"
      assert page.has_content?('[#6&a]:[#6&a]:[#6&a]:[#6&a]:[#6&a]'), "true"
      assert page.has_content?('0.99985'), "true"
      # active
      assert page.has_content?('[#7&A]-[#7&A]'), "true"
      assert page.has_content?('0.99993'), "true"
      # close 'significant fragments' view
      find_button('closebutton').click
    end
    find_link('link0').click
    sleep 2
    assert page.has_content?('Supporting information'), "true"
    first(:link, 'linkCompound').click
    sleep 5
    within_frame('details_overview') do
      assert page.has_content?('SMILES:'), "true"
      assert page.has_content?('c1ccc(cc1)NN'), "true"
      assert page.has_content?('InChI:'), "true"
      assert page.has_content?('1S/C6H8N2/c7-8-6-4-2-1-3-5-6/h1-5,8H,7H2'), "true"
      assert page.has_content?('Names:'), "true"
      assert page.has_content?('Phenylhydrazine'), "true"
      assert page.has_link?('PubChem read across'), "true"
    end
=end
  end

  def test_99_kill
    `pidof Xvfb|xargs kill`
  end

end