recce-nightly 1.2.0.20250506__py3-none-any.whl → 1.26.0.20251124__py3-none-any.whl

tests/test_server.py

@@ -5,17 +5,20 @@ from fastapi.testclient import TestClient
 
 from recce.core import default_context
 from recce.server import app
+
 # noinspection PyUnresolvedReferences
-from tests.adapter.dbt_adapter.conftest import dbt_test_helper
+from tests.adapter.dbt_adapter.conftest import dbt_test_helper  # noqa: F401
 
 
 @pytest.fixture
 def temp_folder():
     import tempfile
+
     temp_dir = tempfile.mkdtemp()
     yield temp_dir
 
     import shutil
+
     shutil.rmtree(temp_dir)
 
 
@@ -28,38 +31,41 @@ def test_health():
 
 def test_stateless(dbt_test_helper):
     context = default_context()
-    from recce.state import RecceStateLoader
-    context.state_loader = RecceStateLoader()
+    from recce.state import FileStateLoader
+
+    context.state_loader = FileStateLoader()
     client = TestClient(app)
     response = client.get("/api/info")
     assert response.status_code == 200
     info = response.json()
-    assert info['file_mode'] is False
-    assert info['cloud_mode'] is False
+    assert info["file_mode"] is False
+    assert info["cloud_mode"] is False
 
 
 def test_file_mode(dbt_test_helper):
     context = default_context()
-    from recce.state import RecceStateLoader
-    context.state_loader = RecceStateLoader(state_file='/tmp/recce_state.json')
+    from recce.state import FileStateLoader
+
+    context.state_loader = FileStateLoader(state_file="/tmp/recce_state.json")
     client = TestClient(app)
     response = client.get("/api/info")
     assert response.status_code == 200
     info = response.json()
-    assert info['file_mode'] is True
-    assert info['filename'] == 'recce_state.json'
-    assert info['cloud_mode'] is False
+    assert info["file_mode"] is True
+    assert info["filename"] == "recce_state.json"
+    assert info["cloud_mode"] is False
 
 
 def test_saveas_and_rename(dbt_test_helper, temp_folder):
     context = default_context()
-    state_file = os.path.join(temp_folder, 'recce_state.json')
-    state_file2 = os.path.join(temp_folder, 'recce_state2.json')
-    state_file3 = os.path.join(temp_folder, 'recce_state3.json')
-    os.makedirs(os.path.join(temp_folder, 'dir.json'))
+    state_file = os.path.join(temp_folder, "recce_state.json")
+    state_file2 = os.path.join(temp_folder, "recce_state2.json")
+    state_file3 = os.path.join(temp_folder, "recce_state3.json")
+    os.makedirs(os.path.join(temp_folder, "dir.json"))
+
+    from recce.state import FileStateLoader
 
-    from recce.state import RecceStateLoader
-    context.state_loader = RecceStateLoader(state_file=state_file)
+    context.state_loader = FileStateLoader(state_file=state_file)
     client = TestClient(app)
 
     response = client.post("/api/save", json={"filename": "recce_state2.json"})
@@ -69,7 +75,7 @@ def test_saveas_and_rename(dbt_test_helper, temp_folder):
     response = client.post("/api/save-as", json={"filename": "recce_state2.json"})
     assert response.status_code == 200
     assert os.path.exists(state_file2)
-    assert context.state_loader.state_file == os.path.join(temp_folder, 'recce_state2.json')
+    assert context.state_loader.state_file == os.path.join(temp_folder, "recce_state2.json")
 
     # Same file
     response = client.post("/api/save-as", json={"filename": "recce_state2.json"})
@@ -84,7 +90,7 @@ def test_saveas_and_rename(dbt_test_helper, temp_folder):
     assert response.status_code == 200
     assert not os.path.exists(state_file2)
     assert os.path.exists(state_file3)
-    assert context.state_loader.state_file == os.path.join(temp_folder, 'recce_state3.json')
+    assert context.state_loader.state_file == os.path.join(temp_folder, "recce_state3.json")
 
     # Conflict
     response = client.post("/api/save-as", json={"filename": "recce_state.json"})
@@ -95,4 +101,4 @@ def test_saveas_and_rename(dbt_test_helper, temp_folder):
     # Overwrite
     response = client.post("/api/save-as", json={"filename": "recce_state.json", "overwrite": True})
     assert response.status_code == 200
-    assert context.state_loader.state_file == os.path.join(temp_folder, 'recce_state.json')
+    assert context.state_loader.state_file == os.path.join(temp_folder, "recce_state.json")

tests/test_summary.py

@@ -1,20 +1,26 @@
 import os
-import unittest
 
-from recce.adapter.dbt_adapter import load_manifest, DbtAdapter, DbtVersion
+import pytest
+
+from recce.adapter.dbt_adapter import DbtAdapter, DbtVersion, load_manifest
 from recce.core import RecceContext, set_default_context
-from recce.summary import generate_summary_metadata, _build_lineage_graph, generate_mermaid_lineage_graph
+from recce.summary import (
+    _build_lineage_graph,
+    generate_mermaid_lineage_graph,
+    generate_summary_metadata,
+)
 
 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()
+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
 
 
-@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'))
+    dbt_version = DbtVersion()
+    if dbt_version < "1.8.1":
+        pytest.skip("Dbt version is less than 1.8.1")
+
+    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()
@@ -23,18 +29,21 @@ def test_generate_summary_metadata():
     # 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'))
+    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_version = DbtVersion()
+    if dbt_version < "1.8.1":
+        pytest.skip("Dbt version is less than 1.8.1")
+
+    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)
@@ -47,11 +56,14 @@ def test_build_lineage_graph():
     assert len(lineage_graph.modified_set) == 3
 
 
-@unittest.skipIf(dbt_version < 'v1.8.1', "Dbt version is less than 1.8.1")
 def test_generate_mermaid_lineage_graph():
+    dbt_version = DbtVersion()
+    if dbt_version < "1.8.1":
+        pytest.skip("Dbt version is less than 1.8.1")
+
     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'))
+    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)

recce/data/_next/static/Kcbs3GEIyH2LxgLYat0es/_buildManifest.js

@@ -1 +0,0 @@
-self.__BUILD_MANIFEST={__rewrites:{afterFiles:[],beforeFiles:[],fallback:[]},"/_error":["static/chunks/pages/_error-ed75be3f25588548.js"],sortedPages:["/_app","/_error"]},self.__BUILD_MANIFEST_CB&&self.__BUILD_MANIFEST_CB();