Added test to run a hive task and mock out the hive command

spotify · Feb 6, 2015 · 4c13e6a · 4c13e6a
1 parent 7f4d78f
commit 4c13e6a
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/test/contrib/hive_test.py b/test/contrib/hive_test.py
@@ -286,7 +286,23 @@ def test_ok_cmd(self):
 
     def test_import_old(self):
         import luigi.hive
-        self.assertEquals(luigi.hive, luigi.contrib.hive)
+        self.assertEquals(luigi.hive.HiveQueryTask, luigi.contrib.hive.HiveQueryTask)
+
+
+class MyHiveTask(luigi.contrib.hive.HiveQueryTask):
+    param = luigi.Parameter()
+
+    def query(self):
+        return 'banana banana %s' % self.param
+
+
+class TestHiveTask(unittest.TestCase):
+
+    @mock.patch('luigi.hadoop.run_and_track_hadoop_job')
+    def test_run(self, run_and_track_hadoop_job):
+        success = luigi.run(['MyHiveTask', '--param', 'foo', '--local-scheduler', '--no-lock'])
+        self.assertTrue(success)
+        self.assertEquals('hive', run_and_track_hadoop_job.call_args[0][0][0])
 
 
 if __name__ == '__main__':