Fix unit tests

Author: HelioGuilherme66

utest/application/test_restartutil.py

@@ -44,8 +44,8 @@ def my_ask(title, message, frame=None,  no_default=False):
                 self.RESULT = True
                 result = restart_dialog()
                 time.sleep(2)
-                assert result is False  # OK when running with invoke test-ci
-                # assert result is True  # OK running in IDE or with invoke on real system
+                # assert result is False  # OK when running with invoke test-ci
+                assert result is True  # OK running in IDE or with invoke on real system
                 self.RESULT = False
                 result = restart_dialog()
                 time.sleep(2)

utest/application/test_updatenotifier.py

@@ -30,9 +30,7 @@
 LASTUPDATECHECK = 'last update check'
 
 import wx
-from wx.lib.agw.aui import AuiManager
 import wx.lib.agw.aui as aui
-from multiprocessing import shared_memory
 from utest.resources import datafilereader, MessageRecordingLoadObserver
 from robotide.ui.mainframe import ActionRegisterer, ToolBar
 from robotide.ui.actiontriggers import MenuBar, ShortcutRegistry
@@ -85,26 +83,22 @@ def OnInit(self):  # Overrides wx method
 
         # tell AuiManager to manage this frame
         self._mgr.SetManagedWindow(self.frame)
-        self.book.SetBackgroundColour((255, 255, 255))
-        self.book.SetForegroundColour((0, 0, 0))
-        self._mgr.AddPane(self.book,
-                          aui.AuiPaneInfo().Name("notebook_editors").
-                          CenterPane().PaneBorder(False))
+        self.book.SetBackgroundColour(wx.Colour(255, 255, 254))
+        self.book.SetForegroundColour(wx.Colour(0, 0, 1))
+        self._mgr.AddPane(self.book, aui.AuiPaneInfo().Name("notebook_editors").CenterPane().PaneBorder(False))
         mb = MenuBar(self.frame)
         self.toolbar = ToolBar(self.frame)
         self.toolbar.SetMinSize(wx.Size(100, 60))
-        self.toolbar.SetBackgroundColour((255, 255, 255))
-        self.toolbar.SetForegroundColour((0, 0, 0))
-        mb.m_frame.SetBackgroundColour((255, 255, 255))
-        mb.m_frame.SetForegroundColour((0, 0, 0))
-        self._mgr.AddPane(self.toolbar, aui.AuiPaneInfo().Name("maintoolbar").
-                          ToolbarPane().Top())
+        self.toolbar.SetBackgroundColour(wx.Colour(254, 255, 255))
+        self.toolbar.SetForegroundColour(wx.Colour(0, 0, 0))
+        mb.m_frame.SetBackgroundColour(wx.Colour(254, 255, 255))
+        mb.m_frame.SetForegroundColour(wx.Colour(0, 0, 0))
+        self._mgr.AddPane(self.toolbar, aui.AuiPaneInfo().Name("maintoolbar").ToolbarPane().Top())
         self.frame.actions = ActionRegisterer(self._mgr, mb, self.toolbar, ShortcutRegistry(self.frame))
         self.tree = Tree(self.frame, self.frame.actions, self.settings)
         self.tree.SetMinSize(wx.Size(275, 250))
         self.frame.SetMinSize(wx.Size(600, 400))
-        self._mgr.AddPane(self.tree,
-                          aui.AuiPaneInfo().Name("tree_content").Caption("Test Suites").CloseButton(False).
+        self._mgr.AddPane(self.tree, aui.AuiPaneInfo().Name("tree_content").Caption("Test Suites").CloseButton(False).
                           LeftDockable())
         mb.take_menu_bar_into_use()
         self._mgr.Update()
@@ -120,7 +114,7 @@ def setUp(self):
         self.app = MyApp()
         settings = self.app.settings
         self.frame = self.app.frame
-        self.frame.actions = ActionRegisterer(AuiManager(self.frame), MenuBar(self.frame), ToolBar(self.frame),
+        self.frame.actions = ActionRegisterer(aui.AuiManager(self.frame), MenuBar(self.frame), ToolBar(self.frame),
                                               ShortcutRegistry(self.frame))
         self.frame.tree = Tree(self.frame, self.frame.actions, settings)
         self.app.project = Project(self.app.namespace, self.app.settings)