From bb833799dcbc398e69c215a7ff865e1a0a23a46b Mon Sep 17 00:00:00 2001
From: Suraj Shetty <surajshetty3416@gmail.com>
Date: Wed, 5 May 2021 14:47:41 +0530
Subject: [PATCH] fix: Click ctx error

---
 frappe/test_runner.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/frappe/test_runner.py b/frappe/test_runner.py
index 2452742f6a..8d6fe48270 100644
--- a/frappe/test_runner.py
+++ b/frappe/test_runner.py
@@ -19,11 +19,13 @@ import click
 import requests
 import unittest.util
 
-click.get_current_context().color = True
-
 unittest_runner = unittest.TextTestRunner
 SLOW_TEST_THRESHOLD = 2
 
+click_ctx = click.get_current_context(True)
+if click_ctx:
+	click_ctx.color = True
+
 def xmlrunner_wrapper(output):
 	"""Convenience wrapper to keep method signature unchanged for XMLTestRunner and TextTestRunner"""
 	def _runner(*args, **kwargs):