test_macro_resolver.py

import itertools
import unittest
import os
from typing import Set, Dict, Any
from unittest import mock

import pytest

# make sure 'redshift' is available
from dbt.adapters import postgres, redshift
from dbt.adapters import factory
from dbt.adapters.base import AdapterConfig
from dbt.contracts.graph.parsed import (
    ParsedModelNode, NodeConfig, DependsOn, ParsedMacro
)
from dbt.context import base, target, configured, providers, docs, manifest, macros
from dbt.contracts.files import FileHash
from dbt.node_types import NodeType
import dbt.exceptions
from .utils import profile_from_dict, config_from_parts_or_dicts, inject_adapter, clear_plugin
from .mock_adapter import adapter_factory

from dbt.context.macro_resolver import MacroResolver


def mock_macro(name, package_name):
    macro = mock.MagicMock(
        __class__=ParsedMacro,
        package_name=package_name,
        resource_type='macro',
        unique_id=f'macro.{package_name}.{name}',
    )
    # Mock(name=...) does not set the `name` attribute, this does.
    macro.name = name
    return macro

class TestMacroResolver(unittest.TestCase):

    def test_resolver(self):
        data = [
            {'package_name': 'my_test', 'name': 'unique'},
            {'package_name': 'my_test', 'name': 'macro_xx'},
            {'package_name': 'one', 'name': 'unique'},
            {'package_name': 'one', 'name': 'not_null'},
            {'package_name': 'two', 'name': 'macro_a'},
            {'package_name': 'two', 'name': 'macro_b'},
        ]
        macros = {}
        for mdata in data:
            macro = mock_macro(mdata['name'], mdata['package_name'])
            macros[macro.unique_id] = macro
        resolver = MacroResolver(macros, 'my_test', ['one'])
        assert(resolver)
        self.assertEqual(resolver.get_macro_id('one', 'not_null'), 'macro.one.not_null')