from __future__ import absolute_import, division, print_function

import sys

import libtbx.phil
from libtbx.program_template import ProgramTemplate
from libtbx.utils import multi_out
from libtbx.version import get_version

# =============================================================================
class TestProgram(ProgramTemplate):

  master_phil = """
parameter_a = None
  .type = str
parameter_b = 0
  .type = int
parameter_c = None
  .type = float
"""

class TestVersionProgram(ProgramTemplate):

  version = 'abc'

  master_phil = """
parameter_a = None
  .type = str
parameter_b = 0
  .type = int
parameter_c = None
  .type = float
"""

working_phil = libtbx.phil.parse("parameter_a = not None\nparameter_b = 5")

# -----------------------------------------------------------------------------
def test_phil():
  master_phil = libtbx.phil.parse(TestProgram.master_phil)
  required_output_phil = libtbx.phil.parse(ProgramTemplate.output_phil_str)
  master_phil.adopt_scope(required_output_phil)

  params = master_phil.fetch(working_phil).extract()
  logger = multi_out()
  logger.register('stdout', sys.stdout)
  test_program = TestProgram(None, params, master_phil, logger)

  full_phil = libtbx.phil.parse(test_program.get_program_phil_str())
  full = master_phil.fetch(full_phil).extract()
  assert full.parameter_a == 'not None'
  assert full.parameter_b == 5
  assert full.parameter_c is None
  assert 'parameter_c' in test_program.get_program_phil_str()
  assert 'parameter_c' not in test_program.get_program_phil_str(True)

  assert test_program.get_default_output_filename() == 'cctbx_program_000'
  assert test_program.get_default_output_filename(prefix='abc') == 'abc_000'
  assert test_program.get_default_output_filename(suffix='abc') == 'cctbx_programabc_000'
  assert test_program.get_default_output_filename(serial=999) == 'cctbx_program_999'
  assert test_program.get_default_output_filename(prefix='abc', suffix='def', serial=123) == 'abcdef_123'

  test_program.params.output.prefix = 'prefix'
  test_program.params.output.suffix = 'suffix'
  test_program.params.output.serial = 7
  assert test_program.get_default_output_filename() == 'prefixsuffix_007'
  assert test_program.get_default_output_filename(prefix='abc') == 'abcsuffix_007'
  assert test_program.get_default_output_filename(suffix='abc') == 'prefixabc_007'
  assert test_program.get_default_output_filename(serial=999) == 'prefixsuffix_999'

  try:
    test_program.get_default_output_filename(serial='abc')
  except ValueError as e:
    if str(e) != 'The serial argument should be an integer.':
      raise

# -----------------------------------------------------------------------------
def test_version():
  assert TestProgram.get_version() == get_version()
  assert TestVersionProgram.get_version() == TestVersionProgram.version

# =============================================================================
if __name__ == '__main__':
  test_phil()