def test__ip_binding(self):
"""Testing method or function named _ip_binding."""
# Test with Hostname
ip_listen_address = 'localhost'
ip_bind_port = '1000'
agent_api_variable = AgentAPIVariable(
ip_bind_port=ip_bind_port, ip_listen_address=ip_listen_address)
result = agent._ip_binding(agent_api_variable)
expected = 'localhost:1000'
self.assertEqual(result, expected)
# Test with IPv4
ip_listen_address = '1.2.3.4'
ip_bind_port = '5678'
agent_api_variable = AgentAPIVariable(
ip_bind_port=ip_bind_port, ip_listen_address=ip_listen_address)
result = agent._ip_binding(agent_api_variable)
expected = '1.2.3.4:5678'
self.assertEqual(result, expected)
# Test with IPv6
ip_listen_address = '1::1'
ip_bind_port = '91011'
agent_api_variable = AgentAPIVariable(
ip_bind_port=ip_bind_port, ip_listen_address=ip_listen_address)
result = agent._ip_binding(agent_api_variable)
expected = '[1::1]:91011'
self.assertEqual(result, expected)
def test___repr__(self):
"""Testing function __repr__."""
# Test defaults
aav = AgentAPIVariable()
expected = ('''\
<AgentAPIVariable ip_bind_port=20201, ip_listen_address='0.0.0.0'>''')
result = aav.__repr__()
self.assertEqual(expected, result)
def __init__(self, parent, child, app, config=None):
"""Initialize the class.
Args:
parent: Name of parent daemon
child: Name of child daemon
app: Flask App
config: Config object
Returns:
None
"""
# Initialize key variables
if config is None:
_config = Config()
else:
_config = config
# Apply inheritance
Agent.__init__(self, parent, child=child, config=_config)
self._app = app
self._agent_api_variable = AgentAPIVariable(
ip_bind_port=_config.ip_bind_port(),
ip_listen_address=_config.ip_listen_address())
def test___init__(self):
"""Testing function __init__."""
# Setup AgentAPIVariable
ip_bind_port = 1234
ip_listen_address = '1.2.3.4'
# Test defaults
aav = AgentAPIVariable()
self.assertEqual(aav.ip_bind_port, 20201)
self.assertEqual(aav.ip_listen_address, '0.0.0.0')
# Test non-defaults
aav = AgentAPIVariable(ip_bind_port=ip_bind_port,
ip_listen_address=ip_listen_address)
self.assertEqual(aav.ip_bind_port, ip_bind_port)
self.assertEqual(aav.ip_listen_address, ip_listen_address)