add TestRuntimeRefResolver and TestRuntimeSourceResolver unit test

tests/unit/test_providers.py

@@ -1,7 +1,9 @@
 import pytest
 from unittest import mock
 
-from dbt.context.providers import BaseResolver
+from dbt.adapters.base import BaseRelation
+from dbt.context.providers import BaseResolver, RuntimeRefResolver, RuntimeSourceResolver
+from dbt.contracts.graph.unparsed import Quoting
 
 
 class TestBaseResolver:
@@ -26,3 +28,77 @@ def test_resolve_limit(self, resolver, empty, expected_resolve_limit):
  resolver.config.args.EMPTY = empty
 
  assert resolver.resolve_limit == expected_resolve_limit
+
+
+class TestRuntimeRefResolver:
+ @pytest.fixture
+ def resolver(self):
+ mock_db_wrapper = mock.Mock()
+ mock_db_wrapper.Relation = BaseRelation
+
+ return RuntimeRefResolver(
+ db_wrapper=mock_db_wrapper,
+ model=mock.Mock(),
+ config=mock.Mock(),
+ manifest=mock.Mock(),
+ )
+
+ @pytest.mark.parametrize(
+ "empty,is_ephemeral_model,expected_limit",
+ [
+ (False, False, None),
+ (True, False, 0),
+ (False, True, None),
+ (True, True, 0),
+ ],
+ )
+ def test_create_relation_with_empty(self, resolver, empty, is_ephemeral_model, expected_limit):
+ # setup resolver and input node
+ resolver.config.args.EMPTY = empty
+ mock_node = mock.Mock()
+ mock_node.database = "test"
+ mock_node.schema = "test"
+ mock_node.identifier = "test"
+ mock_node.alias = "test"
+ mock_node.is_ephemeral_model = is_ephemeral_model
+
+ # create limited relation
+ with mock.patch("dbt.adapters.base.relation.ParsedNode", new=mock.Mock):
+ relation = resolver.create_relation(mock_node)
+ assert relation.limit == expected_limit
+
+
+class TestRuntimeSourceResolver:
+ @pytest.fixture
+ def resolver(self):
+ mock_db_wrapper = mock.Mock()
+ mock_db_wrapper.Relation = BaseRelation
+
+ return RuntimeSourceResolver(
+ db_wrapper=mock_db_wrapper,
+ model=mock.Mock(),
+ config=mock.Mock(),
+ manifest=mock.Mock(),
+ )
+
+ @pytest.mark.parametrize(
+ "empty,expected_limit",
+ [
+ (False, None),
+ (True, 0),
+ ],
+ )
+ def test_create_relation_with_empty(self, resolver, empty, expected_limit):
+ # setup resolver and input source
+ resolver.config.args.EMPTY = empty
+
+ mock_source = mock.Mock()
+ mock_source.database = "test"
+ mock_source.schema = "test"
+ mock_source.identifier = "test"
+ mock_source.quoting = Quoting()
+ resolver.manifest.resolve_source.return_value = mock_source
+
+ # create limited relation
+ relation = resolver.resolve("test", "test")
+ assert relation.limit == expected_limit