net-test: fix gpylint warnings in kernel_feature_test.py

Test: TreeHugger, manually in UML on 6.1-V
Signed-off-by: Maciej Żenczykowski <maze@google.com>
Change-Id: If98f6016a357aa8bbfe0eaf198f8806b6e523bb5

1 files changed, 9 insertions, 8 deletions

diff --git a/net/test/kernel_feature_test.py b/net/test/kernel_feature_test.py
index ea093ec..56f7fa0 100755
--- a/net/test/kernel_feature_test.py
+++ b/net/test/kernel_feature_test.py
@@ -15,15 +15,16 @@
 # limitations under the License.
 
 import errno
-from socket import *  # pylint: disable=wildcard-import
+import gzip
+from socket import *  # pylint: disable=wildcard-import,g-importing-member
 import unittest
 
-import gzip
 import net_test
 
 
 class RemovedFeatureTest(net_test.NetworkTest):
   KCONFIG = None
+  AID_NET_RAW = 3004
 
   @classmethod
   def loadKernelConfig(cls):
@@ -39,13 +40,14 @@ class RemovedFeatureTest(net_test.NetworkTest):
 
   @classmethod
   def setUpClass(cls):
+    super(net_test.NetworkTest, cls).setUpClass()
     cls.loadKernelConfig()
 
-  def assertFeatureEnabled(self, featureName):
-    return self.assertEqual("y", self.KCONFIG[featureName])
+  def assertFeatureEnabled(self, feature_name):
+    return self.assertEqual("y", self.KCONFIG[feature_name])
 
-  def assertFeatureAbsent(self, featureName):
-    return self.assertTrue(featureName not in self.KCONFIG)
+  def assertFeatureAbsent(self, feature_name):
+    return self.assertNotIn(feature_name, self.KCONFIG)
 
   def testNetfilterRejectEnabled(self):
     """Verify that CONFIG_IP{,6}_NF_{FILTER,TARGET_REJECT} is enabled."""
@@ -68,8 +70,7 @@ class RemovedFeatureTest(net_test.NetworkTest):
        -       default y
        +       default n
     """
-    AID_NET_RAW = 3004
-    with net_test.RunAsUidGid(12345, AID_NET_RAW):
+    with net_test.RunAsUidGid(12345, self.AID_NET_RAW):
       self.assertRaisesErrno(errno.EPERM, socket, AF_PACKET, SOCK_RAW, 0)