def test_ipset_restore(self):
"""Test the state restore functionality of IPSet"""
# Disable protected-access: Test access protected members .
# pylint: disable=protected-access
iptables.ipset_restore('Initial IPSet state')
treadmill.iptables._ipset.assert_called_with(
'-exist', 'restore', cmd_input='Initial IPSet state')
def test_ipset_restore(self):
"""Test the state restore functionality of IPSet"""
# Disable W0212: Test access protected members of admin module.
# pylint: disable=W0212
iptables.ipset_restore('Initial IPSet state')
treadmill.iptables._ipset.assert_called_with(
'restore', cmd_input='Initial IPSet state')
