def test_get_hook_base_dir(self, provide_directory_mock): fileCache = FileCache(self.config) # Check missing parameter command = {'commandParams': {}} base = fileCache.get_hook_base_dir(command, "server_url_pref") self.assertEqual(base, None) self.assertFalse(provide_directory_mock.called) # Check existing dir case command = { 'commandParams': { 'hooks_folder': os.path.join('HDP', '2.1.1', 'hooks') } } provide_directory_mock.return_value = "dummy value" fileCache = FileCache(self.config) res = fileCache.get_hook_base_dir(command, "server_url_pref") self.assertEquals( pprint.pformat(provide_directory_mock.call_args_list[0][0]), "('/var/lib/ambari-agent/cache', " "{0}, " "'server_url_pref')".format( pprint.pformat(os.path.join('stacks', 'HDP', '2.1.1', 'hooks')))) self.assertEquals(res, "dummy value")
def test_get_hook_base_dir(self, provide_directory_mock): fileCache = FileCache(self.config) # Check missing parameter command = { 'commandParams' : { } } base = fileCache.get_hook_base_dir(command, "server_url_pref") self.assertEqual(base, None) self.assertFalse(provide_directory_mock.called) # Check existing dir case command = { 'commandParams' : { 'hooks_folder' : os.path.join('HDP', '2.1.1', 'hooks') } } provide_directory_mock.return_value = "dummy value" fileCache = FileCache(self.config) res = fileCache.get_hook_base_dir(command, "server_url_pref") self.assertEquals( pprint.pformat(provide_directory_mock.call_args_list[0][0]), "('/var/lib/ambari-agent/cache', " "{0}, " "'server_url_pref')".format(pprint.pformat(os.path.join('stacks','HDP', '2.1.1', 'hooks')))) self.assertEquals(res, "dummy value")
def test_get_hook_base_dir(self, provide_directory_mock): fileCache = FileCache(self.config) # Check missing parameter command = { 'clusterLevelParams' : { }, 'ambariLevelParams': { 'jdk_location': 'server_url_pref' } } base = fileCache.get_hook_base_dir(command) self.assertEqual(base, None) self.assertFalse(provide_directory_mock.called) # Check existing dir case command = { 'clusterLevelParams' : { 'hooks_folder' : 'stack-hooks' }, 'ambariLevelParams': { 'jdk_location': 'server_url_pref' } } provide_directory_mock.return_value = "dummy value" fileCache = FileCache(self.config) res = fileCache.get_hook_base_dir(command) self.assertEquals( pprint.pformat(provide_directory_mock.call_args_list[0][0]), "('/var/lib/ambari-agent/cache', " "{0}, " "'server_url_pref')".format(pprint.pformat('stack-hooks'))) self.assertEquals(res, "dummy value")