1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

# Copyright (C) 2017: Andrey Skvortsov

     8

#

     9

# See COPYING file for copyrights details.

    10

#

    11

# This program is free software; you can redistribute it and/or

    12

# modify it under the terms of the GNU General Public License

    13

# as published by the Free Software Foundation; either version 2

    14

# of the License, or (at your option) any later version.

    15

#

    16

# This program is distributed in the hope that it will be useful,

    17

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    19

# GNU General Public License for more details.

    20

#

    21

# You should have received a copy of the GNU General Public License

    22

# along with this program; if not, write to the Free Software

    23

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    24

    25

import os

    26

import sys

    27

import unittest

    29

import wx

    30

import time

    31

import traceback

    33

    34

import conftest

    35

import Beremiz

    36

    37

    38

class UserApplicationTest(unittest.TestCase):

    39

    def InstallExceptionHandler(self):

    40

        def handle_exception(e_type, e_value, e_traceback):

    41

            # traceback.print_exception(e_type, e_value, e_traceback)

    42

            self.exc_info = [e_type, e_value, e_traceback]

    43

        self.exc_info = None

    44

        self.old_excepthook = sys.excepthook

    45

        sys.excepthook = handle_exception

    46

    47

    def StartApp(self):

    48

        self.app = None

    49

    50

    def FinishApp(self):

    51

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

    52

        self.app.MainLoop()

    53

        self.app = None

    54

    58

    def tearDown(self):

    59

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

    60

            self.FinishApp()

    61

    62

    def RunUIActions(self, actions):

    63

        for act in actions:

    64

            wx.CallAfter(*act)

    65

            self.ProcessEvents()

    66

    67

    def CheckForErrors(self):

    68

        if self.exc_info is not None:

    69

            # reraise catched previously exception

    70

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

    71

    72

    def ProcessEvents(self):

    73

        for i in range(0, 30):

    74

            self.CheckForErrors()

    75

            wx.Yield()

    76

            time.sleep(0.01)

    77

    78

    80

class BeremizApplicationTest(UserApplicationTest):

    81

    """Test Beremiz as whole application"""

    82

    def StartApp(self):

    83

        self.app = Beremiz.BeremizIDELauncher()

    84

        # disable default exception handler in Beremiz

    85

        self.app.InstallExceptionHandler = lambda: None

    86

        self.InstallExceptionHandler()

    87

        self.app.PreStart()

    88

    89

    def FinishApp(self):

    90

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

    91

        self.app.MainLoop()

    92

        time.sleep(1)

    93

        self.app = None

    94

    95

    def OpenAllProjectElements(self):

    96

        """Open editor for every object in the project tree"""

    97

        self.app.frame.ProjectTree.ExpandAll()

    98

        self.ProcessEvents()

    99

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

   100

        while item is not None:

   101

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

   102

            self.ProcessEvents()

   103

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

   104

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

   105

                wx.lib.agw.customtreectrl.wxEVT_TREE_ITEM_ACTIVATED,

   106

                id, item)

   107

            self.app.frame.OnProjectTreeItemActivated(event)

   108

            self.ProcessEvents()

   109

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

   110

   114

        sys.argv = ["", project]

   115

        self.StartApp()

   116

        self.OpenAllProjectElements()

   117

   118

        user_actions = [

   119

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

   120

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

   121

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

   122

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

   123

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

   124

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

   125

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

   126

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

   127

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

   128

        ]

   129

   130

        self.RunUIActions(user_actions)

   131

        self.FinishApp()

   132

   133

    def GetProjectPath(self, project):

   134

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

   135

   136

    def testStartUp(self):

   137

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

   138

        self.StartApp()

   139

        self.FinishApp()

   140

   154

   155

   156

if __name__ == '__main__':

   157

    unittest.main()