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)