diff --git a/tests/test_arp.py b/tests/test_arp.py index 84060bc5..b9fadf81 100644 --- a/tests/test_arp.py +++ b/tests/test_arp.py @@ -1,5 +1,6 @@ import os import unittest +import json import jc.parsers.arp THIS_DIR = os.path.dirname(os.path.abspath(__file__)) @@ -8,6 +9,7 @@ THIS_DIR = os.path.dirname(os.path.abspath(__file__)) class MyTests(unittest.TestCase): def setUp(self): + # input with open(os.path.join(THIS_DIR, os.pardir, 'tests/fixtures/centos-7.7/arp.out'), 'r') as f: self.centos_7_7_arp = f.read() @@ -26,105 +28,60 @@ class MyTests(unittest.TestCase): with open(os.path.join(THIS_DIR, os.pardir, 'tests/fixtures/ubuntu-18.04/arp-v.out'), 'r') as f: self.ubuntu_18_4_arp_v = f.read() + # output + with open(os.path.join(THIS_DIR, os.pardir, 'tests/fixtures/centos-7.7/arp.json'), 'r') as f: + self.centos_7_7_arp_json = json.loads(f.read()) + + with open(os.path.join(THIS_DIR, os.pardir, 'tests/fixtures/ubuntu-18.04/arp.json'), 'r') as f: + self.ubuntu_18_4_arp_json = json.loads(f.read()) + + with open(os.path.join(THIS_DIR, os.pardir, 'tests/fixtures/centos-7.7/arp-a.json'), 'r') as f: + self.centos_7_7_arp_a_json = json.loads(f.read()) + + with open(os.path.join(THIS_DIR, os.pardir, 'tests/fixtures/ubuntu-18.04/arp-a.json'), 'r') as f: + self.ubuntu_18_4_arp_a_json = json.loads(f.read()) + + with open(os.path.join(THIS_DIR, os.pardir, 'tests/fixtures/centos-7.7/arp-v.json'), 'r') as f: + self.centos_7_7_arp_v_json = json.loads(f.read()) + + with open(os.path.join(THIS_DIR, os.pardir, 'tests/fixtures/ubuntu-18.04/arp-v.json'), 'r') as f: + self.ubuntu_18_4_arp_v_json = json.loads(f.read()) + def test_arp_centos_7_7(self): """ Test 'arp' on Centos 7.7 """ - self.assertEqual(jc.parsers.arp.parse(self.centos_7_7_arp), [{'address': 'gateway', - 'flags_mask': 'C', - 'hwaddress': '00:50:56:f7:4a:fc', - 'hwtype': 'ether', - 'iface': 'ens33'}, - {'address': '192.168.71.254', - 'flags_mask': 'C', - 'hwaddress': '00:50:56:fe:7a:b4', - 'hwtype': 'ether', - 'iface': 'ens33'}]) + self.assertEqual(jc.parsers.arp.parse(self.centos_7_7_arp, quiet=True), self.centos_7_7_arp_json) def test_arp_ubuntu_18_4(self): """ Test 'arp' on Ubuntu 18.4 """ - self.assertEqual(jc.parsers.arp.parse(self.ubuntu_18_4_arp), [{'address': '192.168.71.254', - 'flags_mask': 'C', - 'hwaddress': '00:50:56:fe:7a:b4', - 'hwtype': 'ether', - 'iface': 'ens33'}, - {'address': '_gateway', - 'flags_mask': 'C', - 'hwaddress': '00:50:56:f7:4a:fc', - 'hwtype': 'ether', - 'iface': 'ens33'}]) + self.assertEqual(jc.parsers.arp.parse(self.ubuntu_18_4_arp, quiet=True), self.ubuntu_18_4_arp_json) def test_arp_a_centos_7_7(self): """ Test 'arp -a' on Centos 7.7 """ - self.assertEqual(jc.parsers.arp.parse(self.centos_7_7_arp_a), [{'address': '192.168.71.2', - 'hwaddress': '00:50:56:f7:4a:fc', - 'hwtype': 'ether', - 'iface': 'ens33', - 'name': 'gateway'}, - {'address': '192.168.71.1', - 'hwaddress': '00:50:56:c0:00:08', - 'hwtype': 'ether', - 'iface': 'ens33', - 'name': '?'}, - {'address': '192.168.71.254', - 'hwaddress': '00:50:56:fe:7a:b4', - 'hwtype': 'ether', - 'iface': 'ens33', - 'name': '?'}]) + self.assertEqual(jc.parsers.arp.parse(self.centos_7_7_arp_a, quiet=True), self.centos_7_7_arp_a_json) def test_arp_a_ubuntu_18_4(self): """ Test 'arp -a' on Ubuntu 18.4 """ - self.assertEqual(jc.parsers.arp.parse(self.ubuntu_18_4_arp_a), [{'address': '192.168.71.1', - 'hwaddress': '00:50:56:c0:00:08', - 'hwtype': 'ether', - 'iface': 'ens33', - 'name': '?'}, - {'address': '192.168.71.254', - 'hwaddress': '00:50:56:fe:7a:b4', - 'hwtype': 'ether', - 'iface': 'ens33', - 'name': '?'}, - {'address': '192.168.71.2', - 'hwaddress': '00:50:56:f7:4a:fc', - 'hwtype': 'ether', - 'iface': 'ens33', - 'name': '_gateway'}]) + self.assertEqual(jc.parsers.arp.parse(self.ubuntu_18_4_arp_a, quiet=True), self.ubuntu_18_4_arp_a_json) def test_arp_v_centos_7_7(self): """ Test 'arp -v' on Centos 7.7 """ - self.assertEqual(jc.parsers.arp.parse(self.centos_7_7_arp_v), [{'address': 'gateway', - 'flags_mask': 'C', - 'hwaddress': '00:50:56:f7:4a:fc', - 'hwtype': 'ether', - 'iface': 'ens33'}, - {'address': '192.168.71.254', - 'flags_mask': 'C', - 'hwaddress': '00:50:56:fe:7a:b4', - 'hwtype': 'ether', - 'iface': 'ens33'}]) + self.assertEqual(jc.parsers.arp.parse(self.centos_7_7_arp_v, quiet=True), self.centos_7_7_arp_v_json) def test_arp_v_ubuntu_18_4(self): """ Test 'arp -v' on Ubuntu 18.4 """ - self.assertEqual(jc.parsers.arp.parse(self.ubuntu_18_4_arp_v), [{'address': '192.168.71.254', - 'flags_mask': 'C', - 'hwaddress': '00:50:56:fe:7a:b4', - 'hwtype': 'ether', - 'iface': 'ens33'}, - {'address': '_gateway', - 'flags_mask': 'C', - 'hwaddress': '00:50:56:f7:4a:fc', - 'hwtype': 'ether', - 'iface': 'ens33'}]) + self.assertEqual(jc.parsers.arp.parse(self.ubuntu_18_4_arp_v, quiet=True), self.ubuntu_18_4_arp_v_json) if __name__ == '__main__':