PyBitmessage-2024-12-25/src/tests/test_config.py

# pylint: disable=no-member
"""
Various tests for config
"""
import unittest

from six import StringIO
from pybitmessage.bmconfigparser import BMConfigParser

test_config = """[bitmessagesettings]
maxaddrperstreamsend = 100
maxbootstrapconnections = 10
maxdownloadrate = 0
maxoutboundconnections = 8
maxtotalconnections = 100
maxuploadrate = 0
apiinterface = 127.0.0.1
apiport = 8442
udp = True

[threads]
receive = 3

[network]
bind = None
dandelion = 90

[inventory]
storage = sqlite
acceptmismatch = False

[knownnodes]
maxnodes = 15000

[zlib]
maxsize = 1048576"""


class TestConfig(unittest.TestCase):
    """A test case for bmconfigparser"""
    configfile = StringIO('')

    def setUp(self):
        """creates a backup of BMConfigparser current state"""
        BMConfigParser().write(self.configfile)
        self.configfile.seek(0)

    def tearDown(self):
        """restore to the backup of BMConfigparser"""
        # pylint: disable=protected-access
        BMConfigParser()._reset()
        BMConfigParser().readfp(self.configfile)

    def test_safeGet(self):
        """safeGet retuns provided default for nonexistent option or None"""
        self.assertIs(
            BMConfigParser().safeGet('nonexistent', 'nonexistent'), None)
        self.assertEqual(
            BMConfigParser().safeGet('nonexistent', 'nonexistent', 42), 42)

    def test_safeGetBoolean(self):
        """safeGetBoolean returns False for nonexistent option, no default"""
        self.assertIs(
            BMConfigParser().safeGetBoolean('nonexistent', 'nonexistent'),
            False
        )
        # no arg for default
        # pylint: disable=too-many-function-args
        with self.assertRaises(TypeError):
            BMConfigParser().safeGetBoolean(
                'nonexistent', 'nonexistent', True)

    def test_safeGetInt(self):
        """safeGetInt retuns provided default for nonexistent option or 0"""
        self.assertEqual(
            BMConfigParser().safeGetInt('nonexistent', 'nonexistent'), 0)
        self.assertEqual(
            BMConfigParser().safeGetInt('nonexistent', 'nonexistent', 42), 42)

    def test_safeGetFloat(self):
        """safeGetFloat retuns provided default for nonexistent option or 0.0"""
        self.assertEqual(
            BMConfigParser().safeGetFloat('nonexistent', 'nonexistent'), 0.0)
        self.assertEqual(
            BMConfigParser().safeGetFloat('nonexistent', 'nonexistent', 42.0), 42.0)

    def test_reset(self):
        """safeGetInt retuns provided default for bitmessagesettings option or 0"""
        test_config_object = StringIO(test_config)
        BMConfigParser().readfp(test_config_object)

        self.assertEqual(
            BMConfigParser().safeGetInt('bitmessagesettings', 'maxaddrperstreamsend'), 100)
        # pylint: disable=protected-access
        BMConfigParser()._reset()
        self.assertEqual(
            BMConfigParser().safeGetInt('bitmessagesettings', 'maxaddrperstreamsend'), 500)