Fix get 'Is DVM Template' widget

marmarek · Jul 23, 2020 · b155e05 · b155e05
1 parent 61d7a6d
commit b155e05
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/qubesmanager/tests/test_qube_manager.py b/qubesmanager/tests/test_qube_manager.py
@@ -256,7 +256,9 @@ def test_100_sorting(self):
     @unittest.mock.patch('qubesmanager.qube_manager.QSettings.setValue')
     @unittest.mock.patch('qubesmanager.qube_manager.QSettings.sync')
     def test_101_hide_column(self, mock_sync, mock_settings):
-        self.dialog.action_is_dvm_template.trigger()
+        model = self.dialog.qubes_model
+        action_no = model.columns_indices.index('Is DVM Template')
+        self.dialog.menu_view.actions()[action_no].trigger()
         mock_settings.assert_called_with('columns/Is DVM Template', False)
         self.assertEqual(mock_sync.call_count, 1, "Hidden column not synced")