source: main/waeup.kofa/trunk/src/waeup/kofa/utils/tests/test_utils.py @ 15607

Last change on this file since 15607 was 15476, checked in by Henrik Bettermann, 6 years ago

Use decimal.Decimal to counteract Python floating point limitation.
  • Property svn:keywords set to Id
File size: 6.1 KB
Line 
1# -*- coding: utf-8 -*-
2
3## $Id: test_utils.py 15476 2019-06-27 06:07:52Z henrik $
4##
5## Copyright (C) 2014 Uli Fouquet & Henrik Bettermann
6## This program is free software; you can redistribute it and/or modify
7## it under the terms of the GNU General Public License as published by
8## the Free Software Foundation; either version 2 of the License, or
9## (at your option) any later version.
10##
11## This program is distributed in the hope that it will be useful,
12## but WITHOUT ANY WARRANTY; without even the implied warranty of
13## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14## GNU General Public License for more details.
15##
16## You should have received a copy of the GNU General Public License
17## along with this program; if not, write to the Free Software
18## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19##
20import psutil
21import sys
22import unittest
23from waeup.kofa.interfaces import IKofaUtils
24from waeup.kofa.utils.utils import KofaUtils
25from zope.interface import verify
26
27
28class KofaUtilsTestCase(unittest.TestCase):
29
30    def setUp(self):
31        self.max_vmem = psutil.virtual_memory().total
32        self.max_smem = psutil.swap_memory().total
33
34    def get_cleared_util(self):
35        # Helper: get a `KofUtil` instance with all values of
36        # SYSTEM_MAX_LOAD dict set to ``None``
37        util = KofaUtils()
38        for key, val in util.SYSTEM_MAX_LOAD.items():
39            util.SYSTEM_MAX_LOAD[key] = None
40        return util
41
42    def test_iface(self):
43        # KofaUtils fullfill IKofaUtils expectations
44        utils = KofaUtils()
45        verify.verifyClass(IKofaUtils, KofaUtils)
46        verify.verifyObject(IKofaUtils, utils)
47
48    def test_expensive_actions_allowed_swap_none(self):
49        # unset swap maximum values make KofUtils ignore swap values
50        utils = self.get_cleared_util()
51        utils.SYSTEM_MAX_LOAD['swap-mem'] = None
52        assert utils.expensive_actions_allowed() == True
53        # even not-set values won't block us
54        del utils.SYSTEM_MAX_LOAD['swap-mem']
55        assert utils.expensive_actions_allowed() == True
56
57    @unittest.skipIf(
58        psutil.swap_memory().percent >= 99.0,
59        reason="System swap use over 99%. Cannot set higher allowed value.")
60    def test_expensive_actions_allowed_swap_ok(self):
61        # We can react to high swap values
62        utils = self.get_cleared_util()
63        utils.SYSTEM_MAX_LOAD['swap-mem'] = 99.0           # positive float
64        assert utils.expensive_actions_allowed() == True
65        utils.SYSTEM_MAX_LOAD['swap-mem'] = -1.0           # negative float
66        assert utils.expensive_actions_allowed() == True
67        utils.SYSTEM_MAX_LOAD['swap-mem'] = sys.maxint     # positive int
68        assert utils.expensive_actions_allowed() == True
69        utils.SYSTEM_MAX_LOAD['swap-mem'] = -1             # negative int
70        assert utils.expensive_actions_allowed() == True
71
72    @unittest.skipIf(
73        not psutil.swap_memory().percent,
74        reason="Can test swapping behavior only if actually swapping")
75    def test_expensive_actions_allowed_swap_too_much(self):
76        # We can react if too much swap is used
77        utils = self.get_cleared_util()
78        utils.SYSTEM_MAX_LOAD['swap-mem'] = 0.0            # positive float
79        assert utils.expensive_actions_allowed() == False
80        utils.SYSTEM_MAX_LOAD['swap-mem'] = -100.0         # negative float
81        assert utils.expensive_actions_allowed() == False
82        utils.SYSTEM_MAX_LOAD['swap-mem'] = 0              # positive int
83        assert utils.expensive_actions_allowed() == False
84        utils.SYSTEM_MAX_LOAD['swap-mem'] = -(sys.maxint)  # negative int
85        assert utils.expensive_actions_allowed() == False
86
87    def test_expensive_actions_allowed_virtmem_none(self):
88        # unset virtmem maximum values make KofUtils ignore virtmem values
89        utils = self.get_cleared_util()
90        utils.SYSTEM_MAX_LOAD['virt-mem'] = None
91        assert utils.expensive_actions_allowed() == True
92        # even not-set values won't block us
93        del utils.SYSTEM_MAX_LOAD['virt-mem']
94        assert utils.expensive_actions_allowed() == True
95
96    @unittest.skipIf(
97        psutil.virtual_memory().percent >= 99.0,
98        reason="System virtmem use over 99%. Cannot set higher allowed value.")
99    def test_expensive_actions_allowed_virtmem_ok(self):
100        # We can react to high virtmem values
101        utils = self.get_cleared_util()
102        utils.SYSTEM_MAX_LOAD['virt-mem'] = 99.0           # positive float
103        assert utils.expensive_actions_allowed() == True
104        utils.SYSTEM_MAX_LOAD['virt-mem'] = -1.0           # negative float
105        assert utils.expensive_actions_allowed() == True
106        utils.SYSTEM_MAX_LOAD['virt-mem'] = sys.maxint     # positive int
107        assert utils.expensive_actions_allowed() == True
108        utils.SYSTEM_MAX_LOAD['virt-mem'] = -1             # negative int
109        assert utils.expensive_actions_allowed() == True
110
111    @unittest.skipIf(
112        not psutil.virtual_memory().percent,
113        reason="Can test virtmem behavior only if actually using some")
114    def test_expensive_actions_allowed_virtmem_too_much(self):
115        # We can react if too much virtmem is used
116        utils = self.get_cleared_util()
117        utils.SYSTEM_MAX_LOAD['virt-mem'] = 0.0            # positive float
118        assert utils.expensive_actions_allowed() == False
119        utils.SYSTEM_MAX_LOAD['virt-mem'] = -100.0         # negative float
120        assert utils.expensive_actions_allowed() == False
121        utils.SYSTEM_MAX_LOAD['virt-mem'] = 0              # positive int
122        assert utils.expensive_actions_allowed() == False
123        utils.SYSTEM_MAX_LOAD['virt-mem'] = -(sys.maxint)  # negative int
124        assert utils.expensive_actions_allowed() == False
125
126    def test_format_float(self):
127        format_float= KofaUtils().format_float
128        # This is the floating point problem ...
129        self.assertEqual(4.6*100, 459.99999999999994)
130        # ... which is solved in format_float
131        self.assertEqual(format_float(4.6 ,1), '4.6')
132        self.assertEqual(format_float(4.6 ,2), '4.60')
133        self.assertEqual(format_float(4.6 ,3), '4.600')
134        self.assertEqual(format_float(4.6 ,4), '4.6000')
Note: See TracBrowser for help on using the repository browser.