kernel/kselftest.py: fix cancellation check for pmu event_code_tests

Move the cancellation check for pmu/event_code_tests to the test
execution block to ensure it correctly verifies the existence of
the required directory and avoids early cancellation.

Signed-off-by: Disha Goel <disgoel@linux.ibm.com>

Author: disgoel

kernel/kselftest.py

@@ -129,8 +129,6 @@ def setUp(self):
                 process.system("make install -C %s" % self.sourcedir,
                                shell=True, sudo=True)
         else:
-            if self.subtest == 'pmu/event_code_tests':
-                self.cancel("selftest not supported on distro")
             # Make sure kernel source repo is configured
             if detected_distro.name in ['centos', 'fedora', 'rhel']:
                 src_name = 'kernel'
@@ -157,6 +155,11 @@ def setUp(self):
         self.sourcedir = os.path.join(self.buldir, self.testdir)
         # cmsg_* tests from net/ subdirectory takes a lot of time to complete.
         # Until they have been root caused skip them.
+        if self.subtest == 'pmu/event_code_tests':
+            pmu_test_dir = os.path.join(
+                self.sourcedir, 'powerpc/pmu/event_code_tests')
+            if not os.path.exists(pmu_test_dir):
+                self.cancel("selftest not supported on distro")
         if self.comp == 'net':
             make_path = self.sourcedir + "/net/Makefile"
             process.system("sed -i 's/^.*cmsg_so_mark.sh/#&/g' %s" % make_path,