Add support for model versions

Author: psousa50

integration-tests/models/model_references_model_with_version.sql

@@ -0,0 +1,3 @@
+{% if dbt_unit_testing.version_bigger_or_equal_to("1.5") %}
+  select * from {{ dbt_unit_testing.ref('model_with_version') }} where a > 0
+{% endif %}

integration-tests/models/model_references_model_with_version_1.sql

@@ -0,0 +1,3 @@
+{% if dbt_unit_testing.version_bigger_or_equal_to("1.5") %}
+  select * from {{ dbt_unit_testing.ref('model_with_version', v=1) }} where a > 1
+{% endif %} 

integration-tests/models/model_references_model_with_version_2.sql

@@ -0,0 +1,4 @@
+{% if dbt_unit_testing.version_bigger_or_equal_to("1.5") %}
+  select * from {{ dbt_unit_testing.ref('model_with_version', version=2) }} where a > 2
+{% endif %}
+

integration-tests/models/model_with_version_v1.sql

@@ -0,0 +1 @@
+select 1 as a

integration-tests/models/model_with_version_v2.sql

@@ -0,0 +1 @@
+select 2 as a

integration-tests/models/schema.yml

@@ -0,0 +1,8 @@
+version: 2
+
+models:
+  - name: model_with_version
+
+    versions:
+      - v: 1
+      - v: 2

integration-tests/tests/unit/model_references_model_with_version.sql

@@ -0,0 +1,43 @@
+{{
+    config(
+        tags=['unit-test', 'bigquery', 'snowflake', 'postgres', 'versioned', '1.5.4']
+    )
+}}
+
+{% call dbt_unit_testing.test('model_references_model_with_version', 'latest version') %}
+  {% call dbt_unit_testing.mock_ref ('model_with_version') %}
+    select 0 as a
+    UNION ALL
+    select 1234 as a
+  {% endcall %}
+  {% call dbt_unit_testing.expect() %}
+    select 1234 as a
+  {% endcall %}
+{% endcall %}
+
+UNION ALL
+
+{% call dbt_unit_testing.test('model_references_model_with_version_1', 'version 1') %}
+  {% call dbt_unit_testing.mock_ref ('model_with_version', v=1) %}
+    select 1 as a
+    UNION ALL
+    select 1234 as a
+  {% endcall %}
+  {% call dbt_unit_testing.expect() %}
+    select 1234 as a
+  {% endcall %}
+{% endcall %}
+
+UNION ALL
+
+{% call dbt_unit_testing.test('model_references_model_with_version_2', 'version 2') %}
+  {% call dbt_unit_testing.mock_ref ('model_with_version', version=2) %}
+    select 2 as a
+    UNION ALL
+    select 1234 as a
+  {% endcall %}
+  {% call dbt_unit_testing.expect() %}
+    select 1234 as a
+  {% endcall %}
+{% endcall %}
+ 

macros/mock_builders.sql

@@ -10,11 +10,13 @@
   {% if model_name is undefined %}
     {{ dbt_unit_testing.raise_error('model_name must be provided for mock_ref') }}
   {% endif %}
+  {% set node_version = kwargs["version"] | default(kwargs["v"]) | default(none) %}
   {% set mock = {
      "type": 'mock',
      "resource_type": 'model',
      "name": model_name,
      "package_name": project_or_package,
+     "version": node_version,
      "options": options,
      "input_values": caller(),
     }

macros/overrides.sql

@@ -1,10 +1,11 @@
 {% macro ref(project_or_package, model_name) %}
   {% set project_or_package, model_name = dbt_unit_testing.setup_project_and_model_name(project_or_package, model_name) %}
   {% if dbt_unit_testing.running_unit_test() %}
-      {% set node = {"package_name": project_or_package, "name": model_name} %}
-      {{ return (dbt_unit_testing.ref_cte_name(node)) }}
+    {% set node_version = kwargs["version"] | default (kwargs["v"]) %}
+    {% set node = {"package_name": project_or_package, "name": model_name, "version": node_version} %}
+    {{ return (dbt_unit_testing.ref_cte_name(node)) }}
   {% else %}
-      {{ return (builtins.ref(project_or_package, model_name)) }}
+    {{ return (builtins.ref(project_or_package, model_name, **kwargs)) }}
   {% endif %}
 {% endmacro %}
 

macros/sql_builders.sql

@@ -37,11 +37,15 @@
 {% endmacro %}
 
 {% macro ref_cte_name(node) %}
-  {% if node.project_or_package == model.package_name %}
-    {{ return (dbt_unit_testing.quote_identifier(node.name)) }}
-  {% else %}
-    {{ return (dbt_unit_testing.quote_identifier([node.package_name, node.name] | join("__"))) }}
+  {% set node = dbt_unit_testing.model_node(node) %}
+  {% set parts = [node.name] %}
+  {% if node.package_name != model.package_name %}
+    {% set parts = [node.package_name] + parts %}
+  {% endif %}
+  {% if node.version is not none %}
+    {% set parts = parts + [node.version] %}
   {% endif %}
+  {{ return (dbt_unit_testing.quote_identifier(parts | join("__"))) }}
 {% endmacro %}
 
 {% macro source_cte_name(node) %}