Fix singular tests (#17)

* Add test paths

* Add singular test

* Fix refs for singular tests

* Bump version

dbt_project.yml

@@ -1,5 +1,5 @@
 name: 'upstream_prod'
-version: '0.6.1'
+version: '0.6.2'
 config-version: 2
 
 require-dbt-version: [">=1.5.0", "<2.0.0"]

integration_tests/_template/tests/singular_test.sql

@@ -0,0 +1,6 @@
+select 
+    *
+from 
+    {{ ref('defer_prod') }}
+where
+    this_model = 'This test should run but not fail'

integration_tests/create_symlinks.sh

@@ -7,4 +7,5 @@ do
     ln -f _template/macros/* $proj/macros
     ln -f _template/models/production/* $proj/models/production
     ln -f _template/models/staging/* $proj/models/staging
+    ln -f _template/tests/singular_test.sql $proj/tests
 done

integration_tests/dev_db/dbt_project.yml

@@ -5,6 +5,7 @@ config-version: 2
 profile: 'upstream_prod_db'
 
 model-paths: ["models"]
+test-paths: ["tests"]
 macro-paths: ["macros"]
 
 target-path: "target"  # directory which will store compiled SQL files

integration_tests/dev_db/tests/singular_test.sql

@@ -0,0 +1,6 @@
+select 
+    *
+from 
+    {{ ref('defer_prod') }}
+where
+    this_model = 'This test should run but not fail'

integration_tests/dev_db_dev_sch/dbt_project.yml

@@ -5,6 +5,7 @@ config-version: 2
 profile: 'upstream_prod_db_sch'
 
 model-paths: ["models"]
+test-paths: ["tests"]
 macro-paths: ["macros"]
 
 target-path: "target"  # directory which will store compiled SQL files

integration_tests/dev_db_dev_sch/tests/singular_test.sql

@@ -0,0 +1,6 @@
+select 
+    *
+from 
+    {{ ref('defer_prod') }}
+where
+    this_model = 'This test should run but not fail'

integration_tests/dev_db_env_sch/dbt_project.yml

@@ -5,6 +5,7 @@ config-version: 2
 profile: 'upstream_prod_db_sch'
 
 model-paths: ["models"]
+test-paths: ["tests"]
 macro-paths: ["macros"]
 
 target-path: "target"  # directory which will store compiled SQL files

integration_tests/dev_db_env_sch/tests/singular_test.sql

@@ -0,0 +1,6 @@
+select 
+    *
+from 
+    {{ ref('defer_prod') }}
+where
+    this_model = 'This test should run but not fail'

integration_tests/dev_sch/dbt_project.yml

@@ -5,6 +5,7 @@ config-version: 2
 profile: 'upstream_prod_sch'
 
 model-paths: ["models"]
+test-paths: ["tests"]
 macro-paths: ["macros"]
 
 target-path: "target"  # directory which will store compiled SQL files

integration_tests/dev_sch/tests/singular_test.sql

@@ -0,0 +1,6 @@
+select 
+    *
+from 
+    {{ ref('defer_prod') }}
+where
+    this_model = 'This test should run but not fail'

integration_tests/env_sch/dbt_project.yml

@@ -5,6 +5,7 @@ config-version: 2
 profile: 'upstream_prod_sch'
 
 model-paths: ["models"]
+test-paths: ["tests"]
 macro-paths: ["macros"]
 
 target-path: "target"  # directory which will store compiled SQL files

integration_tests/env_sch/tests/singular_test.sql

@@ -0,0 +1,6 @@
+select 
+    *
+from 
+    {{ ref('defer_prod') }}
+where
+    this_model = 'This test should run but not fail'

macros/ref.sql

@@ -77,11 +77,13 @@ The package can be disabled by setting the variable upstream_prod_enabled = Fals
 
     -- Find models being tested
     {% for test in selected_tests %}
-        {% set tested_model = graph.nodes[test].file_key_name.split(".")[1] %}
-        -- Return dev relation for explicitly selected models
-        {% if parent_model == tested_model %}
-            {{ return(parent_ref) }}
-        {% endif %}
+        {% set test_node = graph.nodes[test] %}
+        {% for test_ref in test_node.refs %}
+            -- Return dev relation for explicitly selected models
+            {% if parent_model == test_ref.name %}
+                {{ return(parent_ref) }}
+            {% endif %}
+        {% endfor %}
     {% endfor %}
 
     -- Use dev relations for models being built during the current run