def test_enable_all(self): '''Test that all clients are enabled''' clients = render_manager.get_all_clients(self.database_path) render_manager.enable_all(self.database_path) for i, row in enumerate(clients): status = render_manager.get_status(self.database_path, i + 1) self.assertEqual(status, "Available")
def create_tree_list(self): '''Create the QTreeWidgetItem for every row and store it in a dictionary''' clients = render_manager.get_all_clients(self.database_path) for id in range(1, len(clients) + 1): row = render_manager.get_row(self.database_path, id) # Create tree list tree_list = QtGui.QTreeWidgetItem(self.ui.render_list) self.render_list_ids.append(row[0]) self.render_list_items[row[0]] = tree_list # Set every coulmns text tree_list.setText(0, self.format_tree_items(row[1])) tree_list.setText(1, self.format_tree_items(row[2])) tree_list.setText(2, self.format_tree_items(row[3])) tree_list.setText(3, self.format_tree_items(row[4])) tree_list.setText(4, self.format_tree_items(row[5])) tree_list.setText(5, self.format_tree_items(row[6])) tree_list.setCheckState(6, QtCore.Qt.Unchecked) # Create stop button stop_button = QtGui.QPushButton("Stop", self) # Store row ID into the stop button stop_button.row_id = row[0] # Group all buttons in a dictionary self.stop_buttons[row[0]] = stop_button # Button size stop_button.setMinimumSize(QtCore.QSize(80, 20)) stop_button.setMaximumSize(QtCore.QSize(80, 20)) self.ui.render_list.setItemWidget(tree_list, 7, stop_button) # Connect button to function stop_button.clicked.connect(self.stop_render) # Fromat text self.tree_color_formatting(tree_list, row[0])
def test_all_get_ifd(self): '''Test that all IFDs are a path or none''' clients = render_manager.get_all_clients(self.database_path) for i, row in enumerate(clients): ifd = render_manager.get_ifd(self.database_path, i + 1) if ifd == "None": self.assertEqual(ifd, "None") else: self.assertTrue(os.path.isfile(ifd))
def update_tree_list(self): '''Update the database by accessing the QTreeWidgetItem objects and editing the text''' clients = render_manager.get_all_clients(self.database_path) for id in range(1, len(clients) + 1): row = render_manager.get_row(self.database_path, id) tree_list = self.render_list_items[row[0]] tree_list.setText(0, self.format_tree_items(row[1])) tree_list.setText(1, self.format_tree_items(row[2])) tree_list.setText(2, self.format_tree_items(row[3])) tree_list.setText(3, self.format_tree_items(row[4])) tree_list.setText(4, self.format_tree_items(row[5])) tree_list.setText(5, self.format_tree_items(row[6])) status = tree_list.text(1) current_host = str(tree_list.text(2)) if (status == "Disabled" or status == "Rendering" and user + "@" + hostname != current_host): tree_list.setCheckState(6, QtCore.Qt.Unchecked) self.tree_color_formatting(tree_list, row[0])
def test_all_status(self): '''Test that all the status of the clients are valid''' clients = render_manager.get_all_clients(self.database_path) for i, row in enumerate(clients): status = render_manager.get_status(self.database_path, i + 1) self.assertIn(status, self.valid_status)
def test_all_clients_are_valid(self): '''Test that all clients are in the all_clients db''' clients_from_config = self.settings.clients clients_from_db = render_manager.get_all_clients(self.database_path) self.assertListEqual(clients_from_db, clients_from_config)