# Copyright 2016 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Test the --after and --wait flags when dsub jobs fail.

This test is a copy of the e2e_after_fail.sh test with additional checks
on the objects returned by dsub.call().
"""
from __future__ import print_function

import sys

from dsub.lib import dsub_errors

# Because this may be invoked from another directory (treated as a library) or
# invoked localy (treated as a binary) both import styles need to be supported.
# pylint: disable=g-import-not-at-top
try:
  from . import test_setup_e2e as test
except ImportError:
  import test_setup_e2e as test

# (1) Launch a job to test command execution failure
print('Launch a job that should fail (--wait for it)...')
try:
  # pyformat: disable
  bad_job_wait = test.run_dsub([
      '--command', 'exit 1',
      '--wait'])
  # pyformat: enable

  print('Expected to throw dsub_errors.JobExecutionError', file=sys.stderr)
  sys.exit(1)
except dsub_errors.JobExecutionError as e:
  if len(e.error_list) != 1:
    print(
        'Expected 1 error during wait, got: %s' % e.error_list, file=sys.stderr)
    sys.exit(1)

# (2) Launch a bad job to allow the next call to detect its failure
print('Launch a job that should fail (don\'t --wait)')
# pyformat: disable
bad_job_previous = test.run_dsub(['--command', 'sleep 5s && exit 1'])
# pyformat: enable

# (3) This call to dsub should fail before submit
print('Launch a job that should fail (--after the previous)')
try:
  # pyformat: disable
  job_after = test.run_dsub([
      '--command', 'echo "does not matter"',
      '--after', bad_job_previous['job-id']])
  # pyformat: enable

  print('Expected to throw a PredecessorJobFailureError', file=sys.stderr)
  sys.exit(1)
except dsub_errors.PredecessorJobFailureError as e:
  if len(e.error_list) != 1:
    print(
        'Expected 1 error from previous job, got: {}'.format(e.error_list),
        file=sys.stderr)
    sys.exit(1)
