Qubes/core-agent-linux
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add some checks for QubesDB /qubes-firewall_handled/[ip]

Browse Source
This commit is contained in:
3hhh 2021-05-16 07:32:10 +02:00
parent 3067e469d3
commit 0993115bdc
No known key found for this signature in database
GPG Key ID: EB03A691DB2F0833
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
qubesagent/test_firewall.py
View File

@ -675,11 +675,17 @@ class TestFirewallWorker(TestCase):
def test_handle_addr(self): def test_handle_addr(self):
self.obj.handle_addr('10.137.0.2') self.obj.handle_addr('10.137.0.2')
self.assertEqual(self.obj.rules['10.137.0.2'], [{'action': 'accept'}]) self.assertEqual(self.obj.rules['10.137.0.2'], [{'action': 'accept'}])
self.assertEqual(self.obj.qdb.entries['/qubes-firewall_handled/10.137.0.2'], '1')
self.obj.handle_addr('10.137.0.2')
self.assertEqual(self.obj.rules['10.137.0.2'], [{'action': 'accept'}])
self.assertEqual(self.obj.qdb.entries['/qubes-firewall_handled/10.137.0.2'], '2')
# fallback to block all # fallback to block all
self.obj.handle_addr('10.137.0.3') self.obj.handle_addr('10.137.0.3')
self.assertEqual(self.obj.rules['10.137.0.3'], [{'action': 'drop'}]) self.assertEqual(self.obj.rules['10.137.0.3'], [{'action': 'drop'}])
self.assertEqual(self.obj.qdb.entries['/qubes-firewall_handled/10.137.0.3'], '1')
self.obj.handle_addr('10.137.0.4') self.obj.handle_addr('10.137.0.4')
self.assertEqual(self.obj.rules['10.137.0.4'], [{'action': 'drop'}]) self.assertEqual(self.obj.rules['10.137.0.4'], [{'action': 'drop'}])
self.assertEqual(self.obj.qdb.entries['/qubes-firewall_handled/10.137.0.4'], '1')
@patch('os.path.isfile') @patch('os.path.isfile')
@patch('os.access') @patch('os.access')