require_relative '../../../lib/commands'

module Unipept
  class UnipeptPept2protTestCase < Unipept::TestCase
    def test_default_batch_size
      command = Cri::Command.define { name 'pept2prot' }
      pept2prot = Commands::Pept2prot.new({ host: 'http://api.unipept.ugent.be' }, [], command)
      assert_equal(10, pept2prot.default_batch_size)
      pept2prot.options[:all] = true
      assert_equal(5, pept2prot.default_batch_size)
    end

    def test_required_fields
      command = Cri::Command.define { name 'pept2prot' }
      pept2prot = Commands::Pept2prot.new({ host: 'http://api.unipept.ugent.be' }, [], command)
      assert_equal(['peptide'], pept2prot.required_fields)
    end

    def test_meganize_options
      command = Cri::Command.define { name 'pept2prot' }
      pept2prot = Commands::Pept2prot.new({ meganize: true, host: 'http://api.unipept.ugent.be' }, [], command)
      assert(pept2prot.options[:all])
      assert_equal(['peptide,refseq_protein_ids'], pept2prot.options[:select])
      assert_equal('blast', pept2prot.options[:format])
    end

    def test_meganize_options_overridecommand
      command = Cri::Command.define { name 'pept2prot' }
      pept2prot = Commands::Pept2prot.new({ meganize: true, format: 'xml', all: false, select: ['something'], host: 'http://api.unipept.ugent.be' }, [], command)
      assert(pept2prot.options[:all])
      assert_equal(['peptide,refseq_protein_ids'], pept2prot.options[:select])
      assert_equal('blast', pept2prot.options[:format])
    end

    def test_help
      out, _err = capture_io_while do
        assert_raises SystemExit do
          Commands::Unipept.run(%w[pept2prot -h])
        end
      end
      assert(out.include?('show help for this command'))

      out, _err = capture_io_while do
        assert_raises SystemExit do
          Commands::Unipept.run(%w[pept2prot --help])
        end
      end
      assert(out.include?('show help for this command'))
    end

    def test_run
      out, err = capture_io_while do
        Commands::Unipept.run(%w[pept2prot --host http://api.unipept.ugent.be ENFVYIAK])
      end
      lines = out.each_line
      assert_equal('', err)
      assert(lines.next.start_with?('peptide,uniprot_id,protein_name,taxon_id'))
      assert(lines.next.start_with?('ENFVYIAK,'))
    end

    def test_run_with_fasta_multiple_batches
      out, err = capture_io_while do
        Commands::Unipept.run(%w[pept2prot --host http://api.unipept.ugent.be --batch 2 >test EGGAGSSTGQR ENFVYIAK >tost EGGAGSSTGQR])
      end
      lines = out.each_line
      assert_equal('', err)
      assert(lines.next.start_with?('fasta_header,peptide,uniprot_id,protein_name,taxon_id'))
      assert(lines.count { |line| line.start_with? '>test,EGGAGSSTGQR,' } >= 1)
      assert(lines.count { |line| line.start_with? '>test,ENFVYIAK,' } >= 1)
      assert(lines.count { |line| line.start_with? '>tost,EGGAGSSTGQR,' } >= 1)
    end

    def test_run_with_fasta_multiple_batches_and_select
      out, err = capture_io_while do
        Commands::Unipept.run(%w[pept2prot --host http://api.unipept.ugent.be --batch 2 --select uniprot_id >test EGGAGSSTGQR ENFVYIAK >tost EGGAGSSTGQR])
      end
      lines = out.each_line
      assert_equal('', err)
      assert(lines.next.start_with?('fasta_header,peptide,uniprot_id'))
      assert(lines.count { |line| line.start_with? '>test,EGGAGSSTGQR,' } >= 1)
      assert(lines.count { |line| line.start_with? '>test,ENFVYIAK,' } >= 1)
      assert(lines.count { |line| line.start_with? '>tost,EGGAGSSTGQR,' } >= 1)
    end

    def test_run_with_fasta_multiple_batches_json
      out, err = capture_io_while do
        Commands::Unipept.run(%w[pept2prot --host http://api.unipept.ugent.be --batch 2 --format json >test EGGAGSSTGQR ENFVYIAK >tost EGGAGSSTGQR])
      end
      lines = out.each_line
      assert_equal('', err)
      output = lines.to_a.join.chomp
      assert(output.start_with?('['))
      assert(output.end_with?(']'))
      assert(!output.include?('}{'))
      assert(output.include?('fasta_header'))
    end

    def test_run_with_fasta_multiple_batches_xml
      out, err = capture_io_while do
        Commands::Unipept.run(%w[pept2prot --host http://api.unipept.ugent.be --batch 2 --format xml >test EGGAGSSTGQR ENFVYIAK >tost EGGAGSSTGQR])
      end
      lines = out.each_line
      assert_equal('', err)
      output = lines.to_a.join.chomp
      assert(output.start_with?('<results>'))
      assert(output.end_with?('</results>'))
      assert(output.include?('<fasta_header>'))
    end
  end
end