Add NetlinkIPaddress class to ethtool module and fix test

frenzymadness · frenzymadness · commit 82857af5b020 · 2017-04-03T21:42:58.000+02:00
diff --git a/python-ethtool/ethtool.c b/python-ethtool/ethtool.c
@@ -955,6 +955,9 @@ MODULE_INIT_FUNC(ethtool)
 	Py_INCREF(&PyEtherInfo_Type);
 	PyModule_AddObject(m, "etherinfo", (PyObject *)&PyEtherInfo_Type);
 
+	Py_INCREF(&ethtool_netlink_ip_address_Type);
+	PyModule_AddObject(m, "NetlinkIPaddress", (PyObject *)&ethtool_netlink_ip_address_Type);
+
 	return m;
 }
 
diff --git a/tests/test_ethtool.py b/tests/test_ethtool.py
@@ -174,7 +174,7 @@ def _verify_etherinfo_object(self, ei):
 
         i6s = ei.get_ipv6_addresses()
         for i6 in i6s:
-            self.assert_(isinstance(i6, ethtool.etherinfo_ipv6addr))
+            self.assert_(isinstance(i6, ethtool.NetlinkIPaddress))
             self.assertIsString(i6.address)
             self.assertIsInt(i6.netmask)
             self.assertIsString(i6.scope)

Original file line number	Diff line number	Diff line change
`@@ -955,6 +955,9 @@ MODULE_INIT_FUNC(ethtool)`
`955`	`955`	`Py_INCREF(&PyEtherInfo_Type);`
`956`	`956`	`PyModule_AddObject(m, "etherinfo", (PyObject *)&PyEtherInfo_Type);`
`957`	`957`
	`958`	`+ Py_INCREF(&ethtool_netlink_ip_address_Type);`
	`959`	`+ PyModule_AddObject(m, "NetlinkIPaddress", (PyObject *)&ethtool_netlink_ip_address_Type);`
	`960`	`+`
`958`	`961`	`return m;`
`959`	`962`	`}`
`960`	`963`