From 769ca2d59bb5b91f9d89d308ec0938b6468cba7b Mon Sep 17 00:00:00 2001
From: Quentin Long <qlo@yelp.com>
Date: Thu, 25 Jun 2015 10:10:08 -0700
Subject: [PATCH] Fixed elastalert-test-rule entrypoint to work

---
 elastalert/test_rule.py | 6 +++++-
 requirements-dev.txt    | 1 -
 requirements.txt        | 1 +
 setup.py                | 2 +-
 4 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/elastalert/test_rule.py b/elastalert/test_rule.py
index 0e08ded8b..5b902429b 100644
--- a/elastalert/test_rule.py
+++ b/elastalert/test_rule.py
@@ -282,6 +282,10 @@ def run_rule_test(self):
         if not args.schema_only and not args.count:
             self.run_elastalert(args)
 
-if __name__ == '__main__':
+
+def main():
     test_instance = MockElastAlerter()
     test_instance.run_rule_test()
+
+if __name__ == '__main__':
+    main()
diff --git a/requirements-dev.txt b/requirements-dev.txt
index 5b9f958a3..7417126e6 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -1,7 +1,6 @@
 -r requirements.txt
 coverage
 flake8
-mock
 pre-commit
 pylint<1.4
 pytest
diff --git a/requirements.txt b/requirements.txt
index 6e4d86c87..824f70875 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,6 +4,7 @@ argparse==1.3.0
 elasticsearch==1.3.0
 jira==0.32
 jsonschema==2.2.0
+mock==1.0.0
 oauthlib==0.7.2
 python-dateutil==2.4.0
 requests==2.5.1
diff --git a/setup.py b/setup.py
index 8fceb4917..ba3d33c98 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,7 @@
     license='Copyright 2014 Yelp',
     entry_points={
         'console_scripts': ['elastalert-create-index=elastalert.create_index:main',
-                            'elastalert-test-rule=elastalert.test_rule:run_rule_test',
+                            'elastalert-test-rule=elastalert.test_rule:main',
                             'elastalert-rule-from-kibana=elastalert.rule_from_kibana:main']},
     packages=find_packages(),
     package_data={'elastalert': ['schema.yaml']},