class BaseTestCase(unittest.TestCase): def setUp(self): super(BaseTestCase, self).setUp() self.root_controller = RootController() self.keypair_controller = self.root_controller.keypair # Make sure the lock doesnt exist self.removeLock() # Mock puppet check for tests RootController.puppet_check = mock.Mock() RootController.puppet_check.return_value = False # Travis fix self.root_controller.path = 'tmp/hieradata/' self.root_controller.command = '/bin/echo testing' def tearDown(self): super(BaseTestCase, self).tearDown() set_config({}, overwrite=True) self.removeLock() def setLock(self): self.root_controller.lock() def removeLock(self): if os.path.isfile(self.root_controller.lockfile): os.remove(self.root_controller.lockfile)
def setUp(self): super(BaseTestCase, self).setUp() self.root_controller = RootController() self.keypair_controller = self.root_controller.keypair # Make sure the lock doesnt exist self.removeLock() # Mock puppet check for tests RootController.puppet_check = mock.Mock() RootController.puppet_check.return_value = False # Travis fix self.root_controller.path = 'tmp/hieradata/' self.root_controller.command = '/bin/echo testing'