def test_run_manifest(self):
self.mock(bot_main, 'post_error_task', lambda *args: self.fail(args))
def call_hook(botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(self.attributes['dimensions'],
botobj.dimensions)
self.assertEqual(False, failure)
self.assertEqual(False, internal_failure)
self.assertEqual({'os': 'Amiga'}, dimensions)
self.assertEqual({}, summary)
self.mock(bot_main, 'call_hook', call_hook)
self._mock_popen(0, url='https://localhost:3')
manifest = {
'dimensions': {
'os': 'Amiga'
},
'hard_timeout': 60,
'host': 'https://localhost:3',
'task_id': '24',
}
bot_main.run_manifest(self.bot, manifest, time.time())
def test_run_manifest_exception(self):
posted = []
def post_error_task(botobj, msg, task_id):
posted.append((botobj, msg.splitlines()[0], task_id))
self.mock(bot_main, "post_error_task", post_error_task)
def call_hook(_botobj, name, *args):
if name == "on_after_task":
failure, internal_failure, dimensions, summary = args
self.assertEqual(False, failure)
self.assertEqual(True, internal_failure)
self.assertEqual({}, dimensions)
self.assertEqual({}, summary)
self.mock(bot_main, "call_hook", call_hook)
def raiseOSError(*_a, **_k):
raise OSError("Dang")
self.mock(subprocess42, "Popen", raiseOSError)
manifest = {
"command": ["echo", "hi"],
"dimensions": {},
"grace_period": 30,
"hard_timeout": 60,
"task_id": "24",
}
bot_main.run_manifest(self.bot, manifest, time.time())
expected = [(self.bot, "Internal exception occured: Dang", "24")]
self.assertEqual(expected, posted)
def test_run_manifest_task_failure(self):
self.mock(bot_main, 'post_error_task', lambda *args: self.fail(args))
def call_hook(_botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(True, failure)
self.assertEqual(False, internal_failure)
self.assertEqual({'pool': 'default'}, dimensions)
self.assertEqual(result, summary)
self.mock(bot_main, 'call_hook', call_hook)
result = self._mock_popen(exit_code=1)
manifest = {
'command': ['echo', 'hi'],
'dimensions': {
'pool': 'default'
},
'grace_period': 30,
'hard_timeout': 60,
'io_timeout': 60,
'task_id': '24',
}
bot_main.run_manifest(self.bot, manifest, time.time())
def test_run_manifest_exception(self):
posted = []
def post_error_task(botobj, msg, task_id):
posted.append((botobj, msg.splitlines()[0], task_id))
self.mock(bot_main, 'post_error_task', post_error_task)
def call_hook(_botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(False, failure)
self.assertEqual(True, internal_failure)
self.assertEqual({}, dimensions)
self.assertEqual({}, summary)
self.mock(bot_main, 'call_hook', call_hook)
def raiseOSError(*_a, **_k):
raise OSError('Dang')
self.mock(subprocess42, 'Popen', raiseOSError)
manifest = {
'command': ['echo', 'hi'],
'dimensions': {},
'grace_period': 30,
'hard_timeout': 60,
'task_id': '24',
}
bot_main.run_manifest(self.bot, manifest, time.time())
expected = [(self.bot, 'Internal exception occured: Dang', '24')]
self.assertEqual(expected, posted)
def test_run_manifest_internal_failure(self):
posted = []
self.mock(bot_main, "post_error_task", lambda *args: posted.append(args))
def call_hook(_botobj, name, *args):
if name == "on_after_task":
failure, internal_failure, dimensions, summary = args
self.assertEqual(False, failure)
self.assertEqual(True, internal_failure)
self.assertEqual({}, dimensions)
self.assertEqual(result, summary)
self.mock(bot_main, "call_hook", call_hook)
result = self._mock_popen(returncode=1)
manifest = {
"command": ["echo", "hi"],
"dimensions": {},
"grace_period": 30,
"hard_timeout": 60,
"io_timeout": 60,
"task_id": "24",
}
bot_main.run_manifest(self.bot, manifest, time.time())
expected = [(self.bot, "Execution failed: internal error (1).", "24")]
self.assertEqual(expected, posted)
def test_run_manifest(self):
self.mock(bot_main, "post_error_task", lambda *args: self.fail(args))
def call_hook(botobj, name, *args):
if name == "on_after_task":
failure, internal_failure, dimensions, summary = args
self.assertEqual(self.attributes["dimensions"], botobj.dimensions)
self.assertEqual(False, failure)
self.assertEqual(False, internal_failure)
self.assertEqual({"os": "Amiga"}, dimensions)
self.assertEqual(result, summary)
self.mock(bot_main, "call_hook", call_hook)
result = self._mock_popen(url="https://localhost:3")
manifest = {
"command": ["echo", "hi"],
"dimensions": {"os": "Amiga"},
"grace_period": 30,
"hard_timeout": 60,
"host": "https://localhost:3",
"task_id": "24",
}
self.assertEqual(self.root_dir, self.bot.base_dir)
bot_main.run_manifest(self.bot, manifest, time.time())
def test_run_manifest(self):
self.mock(bot_main, 'post_error_task', lambda *args: self.fail(args))
def call_hook(botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(self.attributes['dimensions'],
botobj.dimensions)
self.assertEqual(False, failure)
self.assertEqual(False, internal_failure)
self.assertEqual({'os': 'Amiga'}, dimensions)
self.assertEqual(result, summary)
self.mock(bot_main, 'call_hook', call_hook)
result = self._mock_popen(url='https://localhost:3')
manifest = {
'command': ['echo', 'hi'],
'dimensions': {
'os': 'Amiga'
},
'grace_period': 30,
'hard_timeout': 60,
'host': 'https://localhost:3',
'task_id': '24',
}
self.assertEqual(self.root_dir, self.bot.base_dir)
bot_main.run_manifest(self.bot, manifest, time.time())
def test_run_manifest_internal_failure(self):
posted = []
self.mock(bot_main, 'post_error_task',
lambda *args: posted.append(args))
def call_hook(_botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(False, failure)
self.assertEqual(True, internal_failure)
self.assertEqual({}, dimensions)
self.assertEqual(result, summary)
self.mock(bot_main, 'call_hook', call_hook)
result = self._mock_popen(returncode=1)
manifest = {
'command': ['echo', 'hi'],
'dimensions': {},
'grace_period': 30,
'hard_timeout': 60,
'io_timeout': 60,
'task_id': '24',
}
bot_main.run_manifest(self.bot, manifest, time.time())
expected = [(self.bot, 'Execution failed: internal error (1).', '24')]
self.assertEqual(expected, posted)
def test_run_manifest_with_auth_headers(self):
self.bot = self.make_bot(
auth_headers_cb=lambda: ({'A': 'a'}, time.time() + 3600))
self.mock(bot_main, 'post_error_task', self.print_err_and_fail)
def call_hook(botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(self.attributes['dimensions'], botobj.dimensions)
self.assertEqual(False, failure)
self.assertEqual(False, internal_failure)
self.assertEqual({'os': 'Amiga', 'pool': 'default'}, dimensions)
self.assertEqual(result, summary)
self.mock(bot_main, 'call_hook', call_hook)
result = self._mock_popen(
url='https://localhost:3',
expected_auth_params_json={
'swarming_http_headers': {'A': 'a'},
'task_service_account': 'bot', # as in task manifest
})
manifest = {
'command': ['echo', 'hi'],
'dimensions': {'os': 'Amiga', 'pool': 'default'},
'grace_period': 30,
'hard_timeout': 60,
'host': 'https://localhost:3',
'service_account': 'bot',
'task_id': '24',
}
self.assertEqual(self.root_dir, self.bot.base_dir)
bot_main.run_manifest(self.bot, manifest, time.time())
def test_run_manifest_exception(self):
posted = []
def post_error_task(botobj, msg, task_id):
posted.append((botobj, msg.splitlines()[0], task_id))
self.mock(bot_main, 'post_error_task', post_error_task)
def call_hook(_botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(False, failure)
self.assertEqual(True, internal_failure)
self.assertEqual({}, dimensions)
self.assertEqual({}, summary)
self.mock(bot_main, 'call_hook', call_hook)
def raiseOSError(*_a, **_k):
raise OSError('Dang')
self.mock(subprocess42, 'Popen', raiseOSError)
manifest = {
'command': ['echo', 'hi'],
'dimensions': {},
'grace_period': 30,
'hard_timeout': 60,
'task_id': '24',
}
bot_main.run_manifest(self.bot, manifest, time.time())
expected = [(self.bot, 'Internal exception occured: Dang', '24')]
self.assertEqual(expected, posted)
def test_run_manifest_task_failure(self):
self.mock(bot_main, 'post_error_task', self.fail)
def call_hook(_botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure = args
self.assertEqual(True, failure)
self.assertEqual(False, internal_failure)
self.mock(bot_main, 'call_hook', call_hook)
self._mock_popen(bot_main.TASK_FAILED)
manifest = {'hard_timeout': 60, 'task_id': '24'}
bot_main.run_manifest(self.bot, manifest, time.time())
def test_run_manifest_internal_failure(self):
posted = []
self.mock(bot_main, 'post_error_task', lambda *args: posted.append(args))
def call_hook(_botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure = args
self.assertEqual(False, failure)
self.assertEqual(True, internal_failure)
self.mock(bot_main, 'call_hook', call_hook)
self._mock_popen(1)
manifest = {'hard_timeout': 60, 'task_id': '24'}
bot_main.run_manifest(self.bot, manifest, time.time())
expected = [(self.bot, 'Execution failed, internal error.', '24')]
self.assertEqual(expected, posted)
def test_run_manifest_task_failure(self):
self.mock(bot_main, 'post_error_task', self.fail)
def call_hook(_botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(True, failure)
self.assertEqual(False, internal_failure)
self.assertEqual({}, dimensions)
self.assertEqual({}, summary)
self.mock(bot_main, 'call_hook', call_hook)
self._mock_popen(bot_main.TASK_FAILED)
manifest = {'dimensions': {}, 'hard_timeout': 60, 'task_id': '24'}
bot_main.run_manifest(self.bot, manifest, time.time())
def test_run_manifest(self):
self.mock(bot_main, 'post_error_task', lambda *args: self.fail(args))
def call_hook(botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure = args
self.assertEqual(self.attributes['dimensions'], botobj.dimensions)
self.assertEqual(False, failure)
self.assertEqual(False, internal_failure)
self.mock(bot_main, 'call_hook', call_hook)
self._mock_popen(0, url='https://localhost:3')
manifest = {
'hard_timeout': 60,
'host': 'https://localhost:3',
'task_id': '24',
}
bot_main.run_manifest(self.bot, manifest, time.time())
def test_run_manifest_internal_failure(self):
posted = []
self.mock(bot_main, 'post_error_task',
lambda *args: posted.append(args))
def call_hook(_botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(False, failure)
self.assertEqual(True, internal_failure)
self.assertEqual({}, dimensions)
self.assertEqual({}, summary)
self.mock(bot_main, 'call_hook', call_hook)
self._mock_popen(1)
manifest = {'dimensions': {}, 'hard_timeout': 60, 'task_id': '24'}
bot_main.run_manifest(self.bot, manifest, time.time())
expected = [(self.bot, 'Execution failed, internal error.', '24')]
self.assertEqual(expected, posted)
def test_run_manifest_task_failure(self):
self.mock(bot_main, 'post_error_task', lambda *args: self.fail(args))
def call_hook(_botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(True, failure)
self.assertEqual(False, internal_failure)
self.assertEqual({}, dimensions)
self.assertEqual(result, summary)
self.mock(bot_main, 'call_hook', call_hook)
result = self._mock_popen(exit_code=1)
manifest = {
'command': ['echo', 'hi'],
'dimensions': {},
'grace_period': 30,
'hard_timeout': 60,
'io_timeout': 60,
'task_id': '24',
}
bot_main.run_manifest(self.bot, manifest, time.time())
def test_run_manifest_task_failure(self):
self.mock(bot_main, "post_error_task", lambda *args: self.fail(args))
def call_hook(_botobj, name, *args):
if name == "on_after_task":
failure, internal_failure, dimensions, summary = args
self.assertEqual(True, failure)
self.assertEqual(False, internal_failure)
self.assertEqual({}, dimensions)
self.assertEqual(result, summary)
self.mock(bot_main, "call_hook", call_hook)
result = self._mock_popen(exit_code=1)
manifest = {
"command": ["echo", "hi"],
"dimensions": {},
"grace_period": 30,
"hard_timeout": 60,
"io_timeout": 60,
"task_id": "24",
}
bot_main.run_manifest(self.bot, manifest, time.time())
def test_run_manifest(self):
self.mock(bot_main, 'post_error_task', lambda *args: self.fail(args))
def call_hook(botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(self.attributes['dimensions'], botobj.dimensions)
self.assertEqual(False, failure)
self.assertEqual(False, internal_failure)
self.assertEqual({'os': 'Amiga'}, dimensions)
self.assertEqual(result, summary)
self.mock(bot_main, 'call_hook', call_hook)
result = self._mock_popen(url='https://localhost:3')
manifest = {
'command': ['echo', 'hi'],
'dimensions': {'os': 'Amiga'},
'grace_period': 30,
'hard_timeout': 60,
'host': 'https://localhost:3',
'task_id': '24',
}
self.assertEqual(self.root_dir, self.bot.base_dir)
bot_main.run_manifest(self.bot, manifest, time.time())
def test_run_manifest_internal_failure(self):
posted = []
self.mock(bot_main, 'post_error_task', lambda *args: posted.append(args))
def call_hook(_botobj, name, *args):
if name == 'on_after_task':
failure, internal_failure, dimensions, summary = args
self.assertEqual(False, failure)
self.assertEqual(True, internal_failure)
self.assertEqual({}, dimensions)
self.assertEqual(result, summary)
self.mock(bot_main, 'call_hook', call_hook)
result = self._mock_popen(returncode=1)
manifest = {
'command': ['echo', 'hi'],
'dimensions': {},
'grace_period': 30,
'hard_timeout': 60,
'io_timeout': 60,
'task_id': '24',
}
bot_main.run_manifest(self.bot, manifest, time.time())
expected = [(self.bot, 'Execution failed: internal error (1).', '24')]
self.assertEqual(expected, posted)
