def setUp(self): self.wm = WorkspaceManager(self.model_controller, mock(plcore.PluginController)) self.workspace = self.wm.createWorkspace( new_random_workspace_name(), workspaceClass=WorkspaceOnCouch) self.wm.setActiveWorkspace(self.workspace) WorkspacePersister.stopThreads()
def setUp(self): self.couch_uri = CONF.getCouchURI() self.cdm = CouchdbManager(uri=self.couch_uri) wpath = os.path.expanduser("~/.faraday/persistence/") self.fsm = FSManager(wpath) self.wm = WorkspaceManager(mock(ModelController), mock(PluginController)) self._fs_workspaces = [] self._couchdb_workspaces = []
def setUpClass(cls): cls.model_controller = controller.ModelController(mock()) api.setUpAPIs(cls.model_controller) cls.wm = WorkspaceManager(cls.model_controller, mock(plcore.PluginController)) cls.temp_workspace = cls.wm.createWorkspace( test_utils.new_random_workspace_name(), workspaceClass=WorkspaceOnCouch) cls.wm.setActiveWorkspace(cls.temp_workspace) WorkspacePersister.stopThreads()
def setUp(self): self.plugin_repo_path = os.path.join(os.getcwd(), "plugins", "repo") self.plugin_manager = PluginManager(self.plugin_repo_path) controller = ModelController(mock(SecurityManager)) wm = WorkspaceManager(controller, mock(PluginController)) work = wm.createWorkspace('default', workspaceClass=WorkspaceOnCouch) work.setModelController(controller) controller.setWorkspace(work) model.api.setUpAPIs(controller) class WorkspaceStub(): def __init__(self): self.id = "test_space" self.controller = self.plugin_manager.createController(WorkspaceStub())
def test_save_command_in_fs(self): """ Tests if command is saved in couch """ wm = WorkspaceManager(mock(ModelController), mock(PluginController)) command_info = self.getDefaultCommandInfo() command_info['workspace'] = 'new_workspace' exec_command = CommandRunInformation(**command_info) workspace = wm.createWorkspace(exec_command.workspace, workspaceClass=WorkspaceOnFS) wm.setActiveWorkspace(workspace) cm = CommandManager() res = cm.saveCommand(exec_command, workspace) self._manager = PersistenceManagerFactory.getInstance() # saved_doc = self._manager.getDocument(exec_command.workspace, res['id']) # After all wm.removeWorkspace(command_info['workspace'])
def setUpClass(cls): cls.model_controller = mock(controller.ModelController) api.setUpAPIs(cls.model_controller) cls.couch_uri = CONF.getCouchURI() cls.cdm = CouchdbManager(uri=cls.couch_uri) class NotificationTest(NotificationCenter): def __init__(self, ui): self.changes = [] def changeFromInstance(self, change): self.changes.append(change) cls.notifier = NotificationTest(None) model.guiapi.notification_center = cls.notifier cls._couchdb_workspaces = [] cls.wm = WorkspaceManager(cls.model_controller, mock(plcore.PluginController)) cls.workspace = cls.wm.createWorkspace(new_random_workspace_name(), workspaceClass=WorkspaceOnCouch) when(cls.workspace).load().thenReturn(True) cls._couchdb_workspaces.append(cls.workspace.name) cls.wm.setActiveWorkspace(cls.workspace)
def test_save_command_in_couch(self): """ Tests if command is saved in couch """ cm = CommandManager() exec_command = CommandRunInformation(**self.getDefaultCommandInfo()) wm = WorkspaceManager(mock(ModelController), mock(PluginController)) workspace = wm.createWorkspace(exec_command.workspace, workspaceClass=WorkspaceOnCouch) res = cm.saveCommand(exec_command, workspace) self._manager = PersistenceManagerFactory.getInstance() saved_doc = self._manager.getDocument(exec_command.workspace, res['id']) self.assertEquals(exec_command.command, saved_doc['command'], 'Saved command diffier') self.assertEquals(exec_command.parameters, saved_doc['parameters'], 'Saved command diffier') self.assertEquals(exec_command.itime, saved_doc['itime'], 'Saved command diffier') self.assertEquals(exec_command.duration, saved_doc['duration'], 'Saved command diffier')
def setUp(self): self._couchdb_workspaces = [] self.wm = WorkspaceManager(self.model_controller, mock(plcore.PluginController))