recce-nightly 0.62.0.20250417__py3-none-any.whl → 1.30.0.20251221__py3-none-any.whl

tests/test_state.py

@@ -1,123 +0,0 @@
-# add the RecceState test case
-import os
-import unittest
-from datetime import datetime
-
-from recce.core import RecceContext
-from recce.models import Check, Run
-from recce.state import RecceState, ArtifactsRoot, RecceStateLoader
-
-current_dir = os.path.dirname(os.path.abspath(__file__))
-
-
-class TestRecceState(unittest.TestCase):
-    def test_merge_checks(self):
-        check1 = Check(name='test1', description='', type='query')
-        check2 = Check(name='test2', description='', type='query',
-                       updated_at=datetime(2000, 1, 1))
-        check2_2 = Check(name='test2_2', description='', type='query',
-                         updated_at=datetime(2020, 1, 1),
-                         check_id=check2.check_id)
-        check3 = Check(name='test3', description='', type='query')
-
-        context = RecceContext()
-        state = RecceState(checks=[check1], runs=[])
-        context.import_state(state)
-        self.assertEqual(1, len(context.checks))
-        self.assertEqual(check1.name, context.checks[0].name)
-
-        context = RecceContext(checks=[check1, check2])
-        state = RecceState(checks=[check1, check2_2, check3], runs=[])
-        context.import_state(state)
-        self.assertEqual(3, len(context.checks))
-        self.assertEqual(check2_2.name, context.checks[1].name)
-
-    def test_merge_preset_checks(self):
-        check1 = Check(name='test1', description='test1', type='query', params=dict(foo='bar'),
-                       updated_at=datetime(2000, 1, 1), is_preset=True)
-        check2 = Check(name='test2', description='test2', type='query', params=dict(foo='bar'),
-                       updated_at=datetime(2001, 1, 1), is_preset=True)
-
-        context = RecceContext(checks=[check1])
-        state = RecceState(checks=[check2], runs=[])
-        context.import_state(state)
-        self.assertEqual(1, len(context.checks))
-        self.assertEqual(check2.name, context.checks[0].name)
-
-        context = RecceContext(checks=[check2])
-        state = RecceState(checks=[check1], runs=[])
-        context.import_state(state)
-        self.assertEqual(1, len(context.checks))
-        self.assertEqual(check2.name, context.checks[0].name)
-
-    def test_revert_checks(self):
-        check1 = Check(name='test1', description='', type='query')
-        check2 = Check(name='test2', description='', type='query')
-        check2_2 = Check(name='test2_2', description='', type='query', check_id=check2.check_id)
-        check3 = Check(name='test3', description='', type='query')
-
-        context = RecceContext(checks=[check1, check2])
-        state = RecceState(checks=[check2_2, check3], runs=[])
-        context.import_state(state, merge=False)
-        self.assertEqual(2, len(context.checks))
-        self.assertEqual(check2_2.name, context.checks[0].name)
-
-    def test_merge_runs(self):
-        run1 = Run(type='query')
-        run2 = Run(type='query')
-        run3 = Run(type='query')
-
-        context = RecceContext(runs=[])
-        state = RecceState(runs=[run1])
-        context.import_state(state)
-        self.assertEqual(1, len(context.runs))
-
-        context = RecceContext(runs=[run1, run2])
-        state = RecceState(runs=[run2, run3])
-        context.import_state(state)
-        self.assertEqual(3, len(context.runs))
-
-    def test_merge_dbt_artifacts(self):
-        import json
-        import os
-        with open(os.path.join(current_dir, 'manifest.json'), 'r') as f:
-            manifest = json.load(f)
-        manifest['metadata']['generated_at'] = '2000-01-01T00:00:00Z'
-        artifacts = ArtifactsRoot(
-            base=dict(
-                manifest=manifest,
-            ),
-            current=dict(
-                manifest=manifest,
-            ),
-        )
-
-        from tests.adapter.dbt_adapter.dbt_test_helper import DbtTestHelper
-        adapter = DbtTestHelper().adapter
-        adapter.import_artifacts(artifacts)
-        self.assertNotEqual(adapter.base_manifest.metadata.invocation_id, manifest.get('metadata').get('invocation_id'))
-
-        manifest['metadata']['generated_at'] = '2099-01-01T00:00:00Z'
-        adapter.import_artifacts(artifacts)
-        self.assertEqual(adapter.base_manifest.metadata.invocation_id, manifest.get('metadata').get('invocation_id'))
-
-    def test_state_loader(self):
-        context = RecceContext()
-
-        # copy ./recce_state.json to temp and open
-
-        # use library to create a temp file in the context
-        import tempfile
-        import shutil
-        import os
-
-        with tempfile.NamedTemporaryFile() as f:
-            # copy ./recce_state.json to temp file
-            current_dir = os.path.dirname(os.path.abspath(__file__))
-            state_file = os.path.join(current_dir, 'recce_state.json')
-            shutil.copy(state_file, f.name)
-
-            # load the state file
-            state_loader = RecceStateLoader(state_file=f.name)
-            state = state_loader.load()
-            assert len(state.runs) == 17

tests/test_summary.py

@@ -1,57 +0,0 @@
-import os
-import unittest
-
-from recce.adapter.dbt_adapter import load_manifest, DbtAdapter, DbtVersion
-from recce.core import RecceContext, set_default_context
-from recce.summary import generate_summary_metadata, _build_lineage_graph, generate_mermaid_lineage_graph
-
-current_dir = os.path.dirname(os.path.abspath(__file__))
-base_manifest_dir = os.path.join(current_dir, 'data', 'manifest', 'base')
-pr2_manifest_dir = os.path.join(current_dir, 'data', 'manifest', 'pr2')  # Pull Request 2l
-
-dbt_version = DbtVersion()
-
-
-@unittest.skipIf(dbt_version < '1.8.1', "Dbt version is less than 1.8.1")
-def test_generate_summary_metadata():
-    manifest = load_manifest(path=os.path.join(current_dir, 'manifest.json'))
-    assert manifest is not None
-    dbt_adapter = DbtAdapter(curr_manifest=manifest)
-    curr_lineage = dbt_adapter.get_lineage()
-    base_lineage = dbt_adapter.get_lineage()
-
-    # Summary with no changes
-    generate_summary_metadata(curr_lineage, base_lineage)
-
-    base_manifest = load_manifest(path=os.path.join(base_manifest_dir, 'manifest.json'))
-    curr_manifest = load_manifest(path=os.path.join(pr2_manifest_dir, 'manifest.json'))
-    dbt_adapter = DbtAdapter(curr_manifest=curr_manifest, base_manifest=base_manifest)
-    curr_lineage = dbt_adapter.get_lineage()
-    base_lineage = dbt_adapter.get_lineage(base=True)
-    generate_summary_metadata(curr_lineage, base_lineage)
-
-
-@unittest.skipIf(dbt_version < 'v1.8.1', "Dbt version is less than 1.8.1")
-def test_build_lineage_graph():
-    base_manifest = load_manifest(path=os.path.join(base_manifest_dir, 'manifest.json'))
-    curr_manifest = load_manifest(path=os.path.join(pr2_manifest_dir, 'manifest.json'))
-    dbt_adapter = DbtAdapter(curr_manifest=curr_manifest, base_manifest=base_manifest)
-    curr_lineage = dbt_adapter.get_lineage()
-    base_lineage = dbt_adapter.get_lineage(base=True)
-
-    lineage_graph = _build_lineage_graph(curr_lineage, base_lineage)
-    assert len(lineage_graph.modified_set) == 1
-
-
-@unittest.skipIf(dbt_version < 'v1.8.1', "Dbt version is less than 1.8.1")
-def test_generate_mermaid_lineage_graph():
-    set_default_context(RecceContext())
-    base_manifest = load_manifest(path=os.path.join(base_manifest_dir, 'manifest.json'))
-    curr_manifest = load_manifest(path=os.path.join(pr2_manifest_dir, 'manifest.json'))
-    dbt_adapter = DbtAdapter(curr_manifest=curr_manifest, base_manifest=base_manifest)
-    curr_lineage = dbt_adapter.get_lineage()
-    base_lineage = dbt_adapter.get_lineage(base=True)
-    graph = _build_lineage_graph(curr_lineage, base_lineage)
-    mermaid_content, is_empty_graph, is_partial_graph = generate_mermaid_lineage_graph(graph)
-    assert is_empty_graph is False
-    assert is_partial_graph is False