1

import os

     2

import sys

     3

import unittest

     4

import wx

     5

import time

     6

import traceback

     7

from xvfbwrapper import Xvfb

     8

     9

vdisplay = None

    10

    11

    12

def setUpModule():

    13

    vdisplay = Xvfb(width=1280, height=720)

    14

    vdisplay.start()

    15

    16

    17

def tearDownModule():

    18

    if vdisplay is not None:

    19

        vdisplay.stop()

    20

    21

    22

class UserApplicationTest(unittest.TestCase):

    23

    def InstallExceptionHandler(self):

    24

        def handle_exception(e_type, e_value, e_traceback):

    25

            # traceback.print_exception(e_type, e_value, e_traceback)

    26

            self.exc_info = [e_type, e_value, e_traceback]

    27

        self.exc_info = None

    28

        self.old_excepthook = sys.excepthook

    29

        sys.excepthook = handle_exception

    30

    31

    def StartApp(self):

    32

        self.app = None

    33

    34

    def FinishApp(self):

    35

        wx.CallAfter(self.app.frame.Close)

    36

        self.app.MainLoop()

    37

        # time.sleep(0.2)

    38

        self.app = None

    39

    40

    def tearDown(self):

    41

        if self.app is not None and self.app.frame is not None:

    42

            self.FinishApp()

    43

    44

    def RunUIActions(self, actions):

    45

        for act in actions:

    46

            wx.CallAfter(*act)

    47

            self.ProcessEvents()

    48

    49

    def CheckForErrors(self):

    50

        if self.exc_info is not None:

    51

            # reraise catched previously exception

    52

            raise self.exc_info[0], self.exc_info[1], self.exc_info[2]

    53

    54

    def ProcessEvents(self):

    55

        for i in range(0, 30):

    56

            self.CheckForErrors()

    57

            wx.Yield()

    58

            time.sleep(0.01)

    59

    60

    61

class BeremizApplicationTest(UserApplicationTest):

    62

    """Test Beremiz as whole application"""

    63

    def StartApp(self):

    64

        self.app = Beremiz.BeremizIDELauncher()

    65

        # disable default exception handler in Beremiz

    66

        self.app.InstallExceptionHandler = lambda: None

    67

        self.InstallExceptionHandler()

    68

        self.app.PreStart()

    69

    70

    def FinishApp(self):

    71

        wx.CallAfter(self.app.frame.Close)

    72

        self.app.MainLoop()

    73

        time.sleep(1)

    74

        self.app = None

    75

    76

    def OpenAllProjectElements(self):

    77

        # open every object in the project tree

    78

        self.app.frame.ProjectTree.ExpandAll()

    79

        self.ProcessEvents()

    80

        item = self.app.frame.ProjectTree.GetRootItem()

    81

        while item is not None:

    82

            self.app.frame.ProjectTree.SelectItem(item, True)

    83

            self.ProcessEvents()

    84

            id = self.app.frame.ProjectTree.GetId()

    85

            event = wx.lib.agw.customtreectrl.TreeEvent(

    86

                wx.lib.agw.customtreectrl.wxEVT_TREE_ITEM_ACTIVATED,

    87

                id, item)

    88

            self.app.frame.OnProjectTreeItemActivated(event)

    89

            self.ProcessEvents()

    90

            item = self.app.frame.ProjectTree.GetNextVisible(item)

    91

    92

    def CheckTestProject(self, project):

    93

        sys.argv = ["", project]

    94

        self.StartApp()

    95

        self.OpenAllProjectElements()

    96

    97

        user_actions = [

    98

            [self.app.frame.SwitchFullScrMode, None],

    99

            [self.app.frame.SwitchFullScrMode, None],

   100

            [self.app.frame.CTR._Clean],

   101

            [self.app.frame.CTR._Build],

   102

            [self.app.frame.CTR._Connect],

   103

            [self.app.frame.CTR._Transfer],

   104

            [self.app.frame.CTR._Run],

   105

            [self.app.frame.CTR._Stop],

   106

            [self.app.frame.CTR._Disconnect],

   107

        ]

   108

   109

        # user_actions.append([self.app.frame.OnCloseProjectMenu, None])

   110

        self.RunUIActions(user_actions)

   111

        self.FinishApp()

   112

   113

    def GetProjectPath(self, project):

   114

        return os.path.abspath(os.path.join(os.path.dirname(__file__), "..", project))

   115

   116

    # @unittest.skip("")

   117

    def testStartUp(self):

   118

        """Checks whether the app starts and finishes correctly"""

   119

        self.StartApp()

   120

        self.FinishApp()

   121

   122

    # @unittest.skip("")

   123

    def testOpenExampleProjects(self):

   124

        """Opens, builds and runs user PLC examples from tests directory"""

   125

        prj = [

   126

            "first_steps",

   127

            "logging",

   128

            "svgui",

   129

            "traffic_lights",

   130

            "wxGlade",

   131

            "python",

   132

            "wiimote",

   133

            "wxHMI",

   134

        ]

   135

        for name in prj:

   136

            project = self.GetProjectPath(name)

   137

            print "Testing example " + name

   138

            self.CheckTestProject(project)

   139

   140

   141

if __name__ == '__main__':

   142

    if __package__ is None:

   143

        sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..')))

   144

        global Beremiz

   145

        import Beremiz

   146

    unittest.main()