def test_base_title_none(self, wpe, ii):
wpe.return_value = False
ii.return_value = None
title.constants.FLAGS.config_root_path = None
self.assertEqual(title._base_title(), '')
def test_base_title_wpe(self, wpe, ii):
wpe.return_value = True
title.constants.FLAGS.config_root_path = None
ii.return_value = None
self.assertEqual(title._base_title(), 'WinPE')
def test_base_title_path(self, wpe, ii):
wpe.return_value = False
title.constants.FLAGS.config_root_path = '/some/directory'
ii.return_value = None
self.assertEqual(title._base_title(), 'some/directory')
def test_base_title_all(self, wpe, ii):
wpe.return_value = True
ii.return_value = TEST_ID
title.constants.FLAGS.config_root_path = '/some/directory'
self.assertEqual(title._base_title(),
'WinPE - some/directory - {}'.format(TEST_ID))