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

    26

    27

    28

import unittest

    29

import time

    30

    31

import wx

    32

import conftest

    33

import controls.CustomIntCtrl

    34

    35

    36

wxApp = None

    37

    38

class TestCustomIntCtrl(unittest.TestCase):

    39

    def setUp(self):

    40

        global wxApp

    41

        if wxApp is None:

    42

            wxApp = wx.App()

    43

        self.frame = wx.Frame(None)

    44

    45

    def tearDown(self):

    46

        global wxApp

    47

        self.frame.Destroy()

    48

        if wxApp is not None:

    49

            wx.CallAfter(wx.Exit)

    50

            wxApp.MainLoop()

    51

            wxApp = None

    52

    53

    def testMaxLimit(self):

    54

        """Test working upper bound"""

    55

        self.AddControls()

    56

        self.int_ctrl.SetValue(self.max_val + 100)

    57

        self.ProcessEvents()

    58

    59

        self.txt_ctrl.SetFocus()

    60

        self.ProcessEvents()

    61

        self.assertEqual(self.int_ctrl.GetValue(), self.max_val)

    62

    63

    def testMinLimit(self):

    64

        """Test working lower bound"""

    65

        self.AddControls()

    66

        self.int_ctrl.SetValue(self.min_val - 100)

    67

        self.ProcessEvents()

    68

    69

        self.txt_ctrl.SetFocus()

    70

        self.ProcessEvents()

    71

    72

        self.assertEqual(self.int_ctrl.GetValue(), self.min_val)

    73

    74

    def testCorrectValue(self):

    75

        """Test case if no limiting is necessary"""

    76

        self.AddControls()

    77

        val = (self.max_val + self.min_val) // 2

    78

        self.int_ctrl.SetValue(val)

    79

        self.ProcessEvents()

    80

    81

        self.txt_ctrl.SetFocus()

    82

        self.ProcessEvents()

    83

    84

        self.assertEqual(self.int_ctrl.GetValue(), val)

    85

    86

    def testEventBinding(self):

    87

        """Test event sending after edit and bound checks are done"""

    88

        self.AddControls()

    89

        self.event_happend = False

    90

    91

        def EventHandler(event):

    92

            self.event_happend = True

    93

            event.Skip()

    94

    95

        self.int_ctrl.Bind(controls.CustomIntCtrl.EVT_CUSTOM_INT, EventHandler)

    96

    97

        val = (self.max_val + self.min_val) // 2

    98

    99

        self.int_ctrl.SetValue(val)

   100

        self.ProcessEvents()

   101

        self.txt_ctrl.SetFocus()

   102

   103

        self.ProcessEvents()

   104

        self.txt_ctrl.SetFocus()

   105

        self.ProcessEvents()

   106

   107

        self.assertEqual(self.int_ctrl.GetValue(), val)

   108

        self.assertTrue(self.event_happend)

   109

   110

    def testLongNumbers(self):

   111

        """Test support of long integer"""

   112

        self.AddControls()

   113

        val = 40000000000

   114

        self.int_ctrl.SetMax(val)

   115

        self.int_ctrl.SetValue(val)

   116

        self.ProcessEvents()

   117

   118

        self.txt_ctrl.SetFocus()

   119

        self.ProcessEvents()

   120

   121

        self.assertEqual(val, val)

   122

   123

    def ProcessEvents(self):

   124

        for dummy in range(0, 10):

   125

            wx.Yield()

   126

            time.sleep(0.01)

   127

   128

    def AddControls(self):

   129

        vs = wx.BoxSizer(wx.VERTICAL)

   130

        self.int_ctrl = controls.CustomIntCtrl(self.frame)

   131

        self.txt_ctrl = wx.TextCtrl(self.frame)

   132

        vs.Add(self.int_ctrl, 0, wx.ALIGN_CENTRE | wx.ALL, 5)

   133

        vs.Add(self.txt_ctrl, 0, wx.ALIGN_CENTRE | wx.ALL, 5)

   134

        self.frame.SetSizer(vs)

   135

        vs.Fit(self.frame)

   136

        self.frame.Show()

   137

        self.frame.Raise()

   138

   139

        self.min_val = 50

   140

        self.max_val = 100

   141

        self.int_ctrl.SetBounds(self.min_val, self.max_val)

   142

        self.ProcessEvents()

   143

   144

   145

if __name__ == '__main__':

   146

    conftest.init_environment()

   147

    unittest.main()