Merge pull request gocardless#40 from etoulas/dbt-deps-operator

Add dbt deps operator

README.md

@@ -55,6 +55,8 @@ There are five operators currently implemented:
 
 * `DbtDocsGenerateOperator`
   * Calls [`dbt docs generate`](https://docs.getdbt.com/reference/commands/cmd-docs)
+* `DbtDepsOperator`
+  * Calls [`dbt deps`](https://docs.getdbt.com/docs/deps)
 * `DbtSeedOperator`
   * Calls [`dbt seed`](https://docs.getdbt.com/docs/seed)
 * `DbtSnapshotOperator`

airflow_dbt/__init__.py

@@ -4,5 +4,6 @@
     DbtSnapshotOperator,
     DbtRunOperator,
     DbtTestOperator,
-    DbtDocsGenerateOperator
+    DbtDocsGenerateOperator,
+    DbtDepsOperator
 )

airflow_dbt/operators/__init__.py

@@ -3,5 +3,6 @@
     DbtSnapshotOperator,
     DbtRunOperator,
     DbtTestOperator,
-    DbtDocsGenerateOperator
+    DbtDocsGenerateOperator,
+    DbtDepsOperator
 )

airflow_dbt/operators/dbt_operator.py

@@ -133,3 +133,12 @@ def __init__(self, profiles_dir=None, target=None, *args, **kwargs):
 
     def execute(self, context):
         self.create_hook().run_cli('seed')
+
+
+class DbtDepsOperator(DbtBaseOperator):
+    @apply_defaults
+    def __init__(self, profiles_dir=None, target=None, *args, **kwargs):
+        super(DbtDepsOperator, self).__init__(profiles_dir=profiles_dir, target=target, *args, **kwargs)
+
+    def execute(self, context):
+        self.create_hook().run_cli('deps')

tests/operators/test_dbt_operator.py

@@ -6,7 +6,8 @@
     DbtSeedOperator,
     DbtSnapshotOperator,
     DbtRunOperator,
-    DbtTestOperator
+    DbtTestOperator,
+    DbtDepsOperator
 )
 
 
@@ -54,3 +55,12 @@ def test_dbt_seed(self, mock_run_cli):
         )
         operator.execute(None)
         mock_run_cli.assert_called_once_with('seed')
+
+    @mock.patch.object(DbtCliHook, 'run_cli')
+    def test_dbt_deps(self, mock_run_cli):
+        operator = DbtDepsOperator(
+            task_id='deps',
+            dag=self.dag
+        )
+        operator.execute(None)
+        mock_run_cli.assert_called_once_with('deps')