def setUp(self):
super(UpdateTests, self).setUp()
self.add_argv(['update'])
self.updater = Update()
self._init_scanner()
self.gr = GitRepo(self.drupal_gh, self.plugin_name)
os_mock = [
'os.makedirs', 'subprocess.call', 'subprocess.check_output',
'dscan.common.functions.md5_file',
'dscan.common.plugins_util.plugins_base_get'
]
self.patchers = []
for mod in os_mock:
mod_name = mod.split('.')[-1]
ret_val = None
if mod_name == 'call':
ret_val = 0
if mod_name == 'plugins_base_get':
ret_val = [self.controller_get('drupal')]
if ret_val != None:
self.patchers.append(
patch(mod, return_value=ret_val, autospec=True))
else:
self.patchers.append(patch(mod, autospec=True))
attr_name = "mock_%s" % (mod_name)
setattr(self, attr_name, self.patchers[-1].start())
