From 0496e5cd284594ba23d8ecdd714bf6938ed1eecf Mon Sep 17 00:00:00 2001
From: Rushabh Mehta <rmehta@gmail.com>
Date: Thu, 18 Apr 2019 12:07:29 +0530
Subject: [PATCH] fix(tests)

---
 .../automation/doctype/milestone/test_milestone.py   |  2 +-
 .../milestone_tracker/test_milestone_tracker.py      |  2 +-
 frappe/model/document.py                             | 12 ++++++------
 3 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/frappe/automation/doctype/milestone/test_milestone.py b/frappe/automation/doctype/milestone/test_milestone.py
index 34876faf27..75602d48db 100644
--- a/frappe/automation/doctype/milestone/test_milestone.py
+++ b/frappe/automation/doctype/milestone/test_milestone.py
@@ -3,7 +3,7 @@
 # See license.txt
 from __future__ import unicode_literals
 
-import frappe
+#import frappe
 import unittest
 
 class TestMilestone(unittest.TestCase):
diff --git a/frappe/automation/doctype/milestone_tracker/test_milestone_tracker.py b/frappe/automation/doctype/milestone_tracker/test_milestone_tracker.py
index e8544534e4..c9bb6b7d5f 100644
--- a/frappe/automation/doctype/milestone_tracker/test_milestone_tracker.py
+++ b/frappe/automation/doctype/milestone_tracker/test_milestone_tracker.py
@@ -9,7 +9,7 @@ import unittest
 class TestMilestoneTracker(unittest.TestCase):
 	def test_milestone(self):
 		frappe.db.sql('delete from `tabMilestone Tracker`')
-		tracker = frappe.get_doc(dict(
+		frappe.get_doc(dict(
 			doctype = 'Milestone Tracker',
 			document_type = 'ToDo',
 			track_field = 'status'
diff --git a/frappe/model/document.py b/frappe/model/document.py
index 051170a4eb..5fe1ba3bc6 100644
--- a/frappe/model/document.py
+++ b/frappe/model/document.py
@@ -117,6 +117,12 @@ class Document(BaseDocument):
 			# incorrect arguments. let's not proceed.
 			raise ValueError('Illegal arguments')
 
+	@staticmethod
+	def whitelist(f):
+		"""Decorator: Whitelist method to be called remotely via REST API."""
+		f.whitelisted = True
+		return f
+
 	def reload(self):
 		"""Reload document from database"""
 		self.load_from_db()
@@ -1008,12 +1014,6 @@ class Document(BaseDocument):
 			if not frappe.flags.in_migrate:
 				follow_document(self.doctype, self.name, frappe.session.user)
 
-	@staticmethod
-	def whitelist(f):
-		"""Decorator: Whitelist method to be called remotely via REST API."""
-		f.whitelisted = True
-		return f
-
 	@staticmethod
 	def hook(f):
 		"""Decorator: Make method `hookable` (i.e. extensible by another app).