From 9b164571c3ee8cdeebc8cb987b11cd8c37cc7278 Mon Sep 17 00:00:00 2001
From: Gavin D'souza <gavin18d@gmail.com>
Date: Thu, 3 Jun 2021 20:23:06 +0530
Subject: [PATCH] fix: Update coverage omit files

Apply 0d8aeba3ee66c34072e2deb135b85016aefaf0dc for parrallel test runner
---
 frappe/commands/utils.py       | 17 ++++++++++++++---
 frappe/parallel_test_runner.py | 23 ++++++++++++++++++++---
 2 files changed, 34 insertions(+), 6 deletions(-)

diff --git a/frappe/commands/utils.py b/frappe/commands/utils.py
index 306f6278cf..d6cf2309ba 100644
--- a/frappe/commands/utils.py
+++ b/frappe/commands/utils.py
@@ -572,19 +572,30 @@ def run_tests(context, app=None, module=None, doctype=None, test=(), profile=Fal
 
 		# Generate coverage report only for app that is being tested
 		source_path = os.path.join(get_bench_path(), 'apps', app or 'frappe')
-		omit=[
+		incl = [
+			'*.py',
+		]
+		omit = [
+			'*.js',
+			'*.xml',
+			'*.pyc',
+			'*.css',
+			'*.less',
+			'*.scss',
+			'*.vue',
 			'*.pyc',
+			'*.html',
 			'*/test_*',
 			'*/node_modules/*',
 			'*/doctype/*/*_dashboard.py',
-			'*/patches/*'
+			'*/patches/*',
 		]
 
 		if not app or app == 'frappe':
 			omit.append('*/tests/*')
 			omit.append('*/commands/*')
 
-		cov = Coverage(source=[source_path], omit=omit)
+		cov = Coverage(source=[source_path], omit=omit, include=incl)
 		cov.start()
 
 	ret = frappe.test_runner.main(app, module, doctype, context.verbose, tests=tests,
diff --git a/frappe/parallel_test_runner.py b/frappe/parallel_test_runner.py
index 1dbb24f191..2f83b88572 100644
--- a/frappe/parallel_test_runner.py
+++ b/frappe/parallel_test_runner.py
@@ -114,13 +114,30 @@ class ParallelTestRunner():
 
 			# Generate coverage report only for app that is being tested
 			source_path = os.path.join(get_bench_path(), 'apps', self.app)
-			omit=['*.html', '*.js', '*.xml', '*.css', '*.less', '*.scss',
-				'*.vue', '*/doctype/*/*_dashboard.py', '*/patches/*']
+			incl = [
+				'*.py',
+			]
+			omit = [
+				'*.js',
+				'*.xml',
+				'*.pyc',
+				'*.css',
+				'*.less',
+				'*.scss',
+				'*.vue',
+				'*.pyc',
+				'*.html',
+				'*/test_*',
+				'*/node_modules/*',
+				'*/doctype/*/*_dashboard.py',
+				'*/patches/*',
+			]
 
 			if self.app == 'frappe':
+				omit.append('*/tests/*')
 				omit.append('*/commands/*')
 
-			self.coverage = Coverage(source=[source_path], omit=omit)
+			self.coverage = Coverage(source=[source_path], omit=omit, include=incl)
 			self.coverage.start()
 
 	def save_coverage(self):