recce-nightly 1.15.0.20250812__py3-none-any.whl → 1.16.0.20250814__py3-none-any.whl

recce/state/state_loader.py

@@ -0,0 +1,179 @@
+import logging
+import threading
+import time
+from abc import ABC, abstractmethod
+from typing import Dict, Literal, Optional, Tuple, Union, final
+
+from recce.exceptions import RecceException
+from recce.pull_request import fetch_pr_metadata
+
+from ..util.io import SupportedFileTypes, file_io_factory
+from .const import (
+    RECCE_CLOUD_TOKEN_MISSING,
+)
+from .state import RecceState
+
+logger = logging.getLogger("uvicorn")
+
+
+class RecceStateLoader(ABC):
+    def __init__(
+        self,
+        review_mode: bool = False,
+        cloud_mode: bool = False,
+        state_file: Optional[str] = None,
+        cloud_options: Optional[Dict[str, str]] = None,
+        initial_state: Optional[RecceState] = None,
+    ):
+        self.review_mode = review_mode
+        self.cloud_mode = cloud_mode
+        self.state_file = state_file
+        self.cloud_options = cloud_options or {}
+        self.error_message = None
+        self.hint_message = None
+        self.state: RecceState | None = initial_state
+        self.state_lock = threading.Lock()
+        self.state_etag = None
+        self.pr_info = None
+        self.catalog: Literal["github", "preview"] = "github"
+        self.share_id = None
+
+        if self.cloud_mode:
+            if self.cloud_options.get("github_token"):
+                self.catalog = "github"
+                self.pr_info = fetch_pr_metadata(
+                    cloud=self.cloud_mode, github_token=self.cloud_options.get("github_token")
+                )
+                if self.pr_info.id is None:
+                    raise RecceException("Cannot get the pull request information from GitHub.")
+            elif self.cloud_options.get("api_token"):
+                self.catalog = "preview"
+                self.share_id = self.cloud_options.get("share_id")
+            else:
+                raise RecceException(RECCE_CLOUD_TOKEN_MISSING.error_message)
+
+    @abstractmethod
+    def verify(self) -> bool:
+        """
+        Verify the state loader configuration.
+        Returns:
+            bool: True if the configuration is valid, False otherwise.
+        """
+        raise NotImplementedError("Subclasses must implement this method.")
+
+    @property
+    def token(self):
+        return self.cloud_options.get("github_token") or self.cloud_options.get("api_token")
+
+    @property
+    def error_and_hint(self) -> (Union[str, None], Union[str, None]):
+        return self.error_message, self.hint_message
+
+    def update(self, state: RecceState):
+        self.state = state
+
+    @final
+    def load(self, refresh=False) -> RecceState:
+        if self.state is not None and refresh is False:
+            return self.state
+        self.state_lock.acquire()
+        try:
+            self.state, self.state_etag = self._load_state()
+        finally:
+            self.state_lock.release()
+        return self.state
+
+    @abstractmethod
+    def _load_state(self) -> Tuple[RecceState, str]:
+        """
+        Load the state from the specified source (file or cloud).
+        Returns:
+            RecceState: The loaded state object.
+            str: The etag of the state file (if applicable).
+        """
+        raise NotImplementedError("Subclasses must implement this method.")
+
+    def save_as(self, state_file: str, state: RecceState = None):
+        if self.cloud_mode:
+            raise Exception("Cannot save the state to Recce Cloud.")
+
+        self.state_file = state_file
+        self.export(state)
+
+    @final
+    def export(self, state: RecceState = None) -> Union[str, None]:
+        if state is not None:
+            self.update(state)
+
+        start_time = time.time()
+        self.state_lock.acquire()
+        try:
+            message, state_etag = self._export_state()
+            self.state_etag = state_etag
+            end_time = time.time()
+            elapsed_time = end_time - start_time
+        finally:
+            self.state_lock.release()
+        logger.info(f"Store state completed in {elapsed_time:.2f} seconds")
+        return message
+
+    @abstractmethod
+    def _export_state(self, state: RecceState = None) -> Tuple[Union[str, None], str]:
+        """
+        Export the current Recce state to a file or cloud storage.
+        Returns:
+            str: A message indicating the result of the export operation.
+            str: The etag of the exported state file (if applicable).
+        """
+        raise NotImplementedError("Subclasses must implement this method.")
+
+    def _export_state_to_file(self, file_path: str, file_type: SupportedFileTypes = SupportedFileTypes.FILE) -> str:
+        """
+        Store the state to a file. Store happens when terminating the server or run instance.
+        """
+
+        json_data = self.state.to_json()
+        io = file_io_factory(file_type)
+
+        io.write(file_path, json_data)
+        return f"The state file is stored at '{file_path}'"
+
+    def refresh(self):
+        new_state = self.load(refresh=True)
+        return new_state
+
+    def check_conflict(self) -> bool:
+        if not self.cloud_mode:
+            return False
+
+        metadata = self._get_metadata_from_recce_cloud()
+        if not metadata:
+            return False
+
+        state_etag = metadata.get("etag")
+        return state_etag != self.state_etag
+
+    def info(self):
+        if self.state is None:
+            self.error_message = "No state is loaded."
+            return None
+
+        state_info = {
+            "mode": "cloud" if self.cloud_mode else "local",
+            "source": None,
+        }
+        if self.cloud_mode:
+            state_info["source"] = "Recce Cloud"
+            state_info["pull_request"] = self.pr_info
+        else:
+            state_info["source"] = self.state_file
+        return state_info
+
+    @abstractmethod
+    def purge(self) -> bool:
+        """
+        Purge the state file or cloud storage.
+        Returns:
+            bool: True if the purge was successful, False otherwise.
+        """
+        raise NotImplementedError("Subclasses must implement this method.")

{recce_nightly-1.15.0.20250812.dist-info → recce_nightly-1.16.0.20250814.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: recce-nightly
-Version: 1.15.0.20250812
+Version: 1.16.0.20250814
 Summary: Environment diff tool for dbt
 Home-page: https://github.com/InfuseAI/recce
 Author: InfuseAI Dev Team

{recce_nightly-1.15.0.20250812.dist-info → recce_nightly-1.16.0.20250814.dist-info}/RECORD

@@ -1,7 +1,7 @@
-recce/VERSION,sha256=PE02SiDrRHT6E1-Kn-oNB6j5okoMuiJVCpfMZOxki5U,16
+recce/VERSION,sha256=Q2iHeFOuCVI1u8mLRlEEA_5cR9HurbaSvimgqL_p-xY,16
 recce/__init__.py,sha256=yNb0QT-yoStex0VZALNJvUwtPLommoVCStcow31guqo,2392
 recce/artifact.py,sha256=ds0erp0-gGnH4Nx2vvapWR02pX8FmKQcqcd6QGqiVPw,6536
-recce/cli.py,sha256=KB9LEzeBasl2xJ4t5oBaNMIipOMOX0NprdyizszIU-Q,44564
+recce/cli.py,sha256=ECfV2IQg79_kOElCYBxd2ubBeG6KfoQqf6xcn4Tm9D4,44732
 recce/config.py,sha256=A4CbKcIdwQ4A9Q2ba4riHUshVQw1D-qDelMOdlt2khU,4661
 recce/connect_to_cloud.py,sha256=_SkX2pdyXa9FNyaHvItyYVPF2nZxy2JnCyd_o_psh2Y,4750
 recce/core.py,sha256=MtBWxemvCQDdUITwkU2JyuQYqcjlA0a76M8Kg53ECxQ,10754
@@ -12,7 +12,6 @@ recce/github.py,sha256=PEpM6ZRiinsPbXSWj4aJCKbZrN1jUXzpzAfJq_CGah4,7420
 recce/pull_request.py,sha256=aW0B1NE2LUKTam1S4TQ7smXB9KLE1DV8GnyBqNXA6j8,3832
 recce/run.py,sha256=PNafwUUJdIG8b1o0y1QuvjACpA3E-5a3keH45CrWHN0,14044
 recce/server.py,sha256=qW81QRZVYEfeDT0n6FwsN9TGhzk6-Zx2mPUjjy01DRM,22473
-recce/state.py,sha256=p4CBneaat-OFwhABfj_zG6n0mPGzrYE-F5xi28Cnzuc,34069
 recce/summary.py,sha256=Mbxvxr9KazR5o9icqhhjiGHsoAiWxQU4PdN7HytBJ1c,19154
 recce/adapter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 recce/adapter/base.py,sha256=T_JNeLHgiHSaegw-DbrvHOaYjMyZcjj2Qtg5cWh_fco,3548
@@ -24,13 +23,11 @@ recce/apis/check_api.py,sha256=KMCXSMl1qqzx2jQgRqCrD4j_cY3EHBbM3H2-t-6saAU,6227
 recce/apis/check_func.py,sha256=gktbCcyk3WGvWRJJ-wDnwv7NrIny2nTHWLl1-kdiVRo,4183
 recce/apis/run_api.py,sha256=eOaxOxXDkH59uqGCd4blld7edavUx7JU_DCd2WAYrL8,3416
 recce/apis/run_func.py,sha256=6wC8TDU-h7TLr2VZH7HNsWaUVlQ9HBN5N_dwqfi4lMY,7440
-recce/data/404.html,sha256=2J6JUm0Dtbqg3l7S0ziLIFgKO5ytt9tpTpfa7RHbxbA,59079
+recce/data/404.html,sha256=iw-kLe4eGZrqOQZUsZIZb80jBni44af496e5rAYJ2vY,59079
 recce/data/auth_callback.html,sha256=H-XfdlAFiw5VU2RpKBVQbwh1AIqJrPHrFA0S09nNJZA,94779
 recce/data/favicon.ico,sha256=B2mBumUOnzvUrXrqNkrc5QfdDXjzEXRcWkWur0fJ6sM,2565
-recce/data/index.html,sha256=jm6Ar9BP-X5RZ4tdKMTEvjrUDFsoCiz2GaL4ENy93SQ,77443
-recce/data/index.txt,sha256=nx3PpPmqerwDpCl79Jb6adxSr3TmX9Rnp5QMkjNqgbg,6403
-recce/data/_next/static/2Vkj92N9DmkZ6Tr4mOIoQ/_buildManifest.js,sha256=xih3ZGwUimzWECoZJsiH_MfTi3cNvK2DbJ_VXPRZHWY,544
-recce/data/_next/static/2Vkj92N9DmkZ6Tr4mOIoQ/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
+recce/data/index.html,sha256=wv4yxX1Z8oiPZjGTTqttscAdFXKC71Oh9hh7kB4U64w,77443
+recce/data/index.txt,sha256=5ala8bJ3iBuOFH-bLfoTqrAwzIBMSiIC9VLZK_Coxdo,6403
 recce/data/_next/static/chunks/0376eeba-3db2196398d62270.js,sha256=ef3PXUZocFm_JH0fYU_EfcaLKov9vMUhrat1YvV-IIU,80953
 recce/data/_next/static/chunks/068b80ea-833a129468ee1622.js,sha256=LrX07Wm6qBJ1uVFYOIkium2ZZ7TkhoCzaz5F_D5P-sM,4627
 recce/data/_next/static/chunks/0ddaf06c-c7961285f66460f6.js,sha256=rY2q_JCikDHZGB8z6nHMV7TON5MlsD4B2zxkj_9DFkE,697
@@ -41,7 +38,7 @@ recce/data/_next/static/chunks/2541941f-2cd3a7c2d629bd33.js,sha256=03PEb_eExziXt
 recce/data/_next/static/chunks/273-f3fa401bd2b6fc91.js,sha256=HopLAKoEGP0dFN57b_41PkPqWAdxDAGzXPrVB0tHSt4,1277071
 recce/data/_next/static/chunks/2fc37c1e-910deebeb3d77c90.js,sha256=npExGSkJthgMbnqcvsiEb9b8aCCf7EQLWu1Io-6By0U,3836
 recce/data/_next/static/chunks/338-2e7eed5135c64550.js,sha256=0MRikUq4wS_iboITYISGCEWffqgtIr95bvSA_gVx5KY,264552
-recce/data/_next/static/chunks/367-ab8b16dd5f8586ca.js,sha256=ij9VwVffJpkXU2-XwXW1SVA2LTLVoyoyAGyoVqUv0XU,84641
+recce/data/_next/static/chunks/367-ae35f5a152ee1ce5.js,sha256=PgRntGGaiUZc3lYFiGE1kddE12rkmi7Ao112KXIDzVo,84706
 recce/data/_next/static/chunks/3a92ee20-0400ffe460c7c803.js,sha256=wI3cuxjDqcE8hRKMg29Mt2dl16jcQ6lBr4JYcLk3_IU,177908
 recce/data/_next/static/chunks/62446465-423c03bb8c1f59b6.js,sha256=xiGo3FRqRpbbBqPwj2UV2zXSfCAPA35TRbyA77kiNMY,2622
 recce/data/_next/static/chunks/6af7f9e9-60aa8706f49dae45.js,sha256=bLrPO62eNT0ez4mooE8DDJITbbm_v6kFB_t8e5hlhx8,767
@@ -63,7 +60,7 @@ recce/data/_next/static/chunks/main-cd6c104af638214a.js,sha256=pYGvgDmnAThczzBHS
 recce/data/_next/static/chunks/polyfills-42372ed130431b0a.js,sha256=CXPB1kyIrcjjyVBBDLWLKI9yEY1ZZbeASUON648vloM,112594
 recce/data/_next/static/chunks/webpack-84df6dd5ae3cf908.js,sha256=_FsecrHvMVA-0EfxZv-v8svx7W50DkEH-jb1NbVSSlU,3580
 recce/data/_next/static/chunks/app/layout-744f0a78e9e50e60.js,sha256=C1sJ1DG8RYccuG-VVTUBO-mmSjiOvJPdxqS9z0maJLk,1410
-recce/data/_next/static/chunks/app/page-e8f798c2ae3f59c2.js,sha256=5zckfX5OoWAY3HAqSLKdawS1XgWiWkbFLvbwP02LDs4,166586
+recce/data/_next/static/chunks/app/page-501b751b630a2ef8.js,sha256=50pQu8x5XHrpV06Ft8mbRvk0PjbYimWV0lnIZTinbSo,166618
 recce/data/_next/static/chunks/app/_not-found/page-c7ef8ed6dc07aaeb.js,sha256=1FE2NEJaSY3ZE8PzMkICM5FcbAbGa8K5ee0ucm2gNfU,2673
 recce/data/_next/static/chunks/pages/_app-73008661edbd5e05.js,sha256=wV9XbJ2FbdpnEztEcX5YE7ecw4ec7LPp2fYhOnLbcoo,240
 recce/data/_next/static/chunks/pages/_error-cf8bbdc3cf76c83f.js,sha256=k0UKnF-ueCkgYTKsUejnYmf31Stt9q5mOjX66XWVqrY,217
@@ -71,6 +68,8 @@ recce/data/_next/static/css/188a3a1687e2a064.css,sha256=fKJMV0A2hM3cjuRmNhx2V_IL
 recce/data/_next/static/css/8edca58d4abcf908.css,sha256=Ab-wk7RhsXgiX5fkJsrCJ-EVvrS1hZtOFpI0uZQSgL0,8093
 recce/data/_next/static/css/abdb9814a3dd18bb.css,sha256=Do5x9FJaOIe-pSBlbk_PP7PQAYFbDXTTl_0PWY4cnBM,1327
 recce/data/_next/static/css/c21263c1520b615b.css,sha256=2TrfHvYUzCBJKLdPzYN6hOAg1n8CoOsJFo36LJPQbOI,13037
+recce/data/_next/static/ee0xgtj4tA9NSgi1ChDxe/_buildManifest.js,sha256=xih3ZGwUimzWECoZJsiH_MfTi3cNvK2DbJ_VXPRZHWY,544
+recce/data/_next/static/ee0xgtj4tA9NSgi1ChDxe/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
 recce/data/_next/static/media/montserrat-cyrillic-800-normal.22628180.woff2,sha256=P5Sx5PNkdTlDYzAdulW_OPRfMokDLi6XTpmS8KJSRoI,11148
 recce/data/_next/static/media/montserrat-cyrillic-800-normal.bd5c9f50.woff,sha256=PFm1Nwt11gmigNvVqPOoDbUFo70fZzUfyZ9S_5f8AnE,10920
 recce/data/_next/static/media/montserrat-cyrillic-ext-800-normal.94a63aea.woff2,sha256=qHQDcPj9VkK-6sE5nxIKvejtNLo5RwBULzyH9nZsEwo,12052
@@ -95,6 +94,12 @@ recce/models/__init__.py,sha256=F7cgALtdWnwv37R0eEgKZ_yBsMwxWnUfo3jAZ3u6qyU,209
 recce/models/check.py,sha256=jjR1SGyczjrqyK-0Zas6ikLIGSgVp54lvfcQA19AniI,1588
 recce/models/run.py,sha256=QK2gvOWvko9YYhd2NLs3BPt5l4MSCZGwpzTAiqx9zJw,1161
 recce/models/types.py,sha256=9ReOyIv3rUD3_cIQfB9Rplb0L2YQuZr4kS9Zai30nB8,5234
+recce/state/__init__.py,sha256=V1FRPTQJUz-uwI3Cn8wDa5Z9bueVs86MR_1Ti4RGfPc,650
+recce/state/cloud.py,sha256=FJS9ktORHT17nBeh1ZDq0PVC21W1Dj8Xz5u4u8X3FLM,15443
+recce/state/const.py,sha256=Me3uVQHi_uZysl3tfpmghPInuKF236X6LfMKVvJlCgA,827
+recce/state/local.py,sha256=bZIkl7cAfyYaGgYEr3uD_JLrtwlHBnu8_o1Qz2djQzw,1920
+recce/state/state.py,sha256=L1cEAl6S9rxTvSIc8ICNwOYoCYK5tR3M0kBo4NTcXK4,3617
+recce/state/state_loader.py,sha256=vGM8HCIv_3PGfcKPgq_GgG_3roloaDL_Am0oTreHmTk,5979
 recce/tasks/__init__.py,sha256=b553AtDHjYROgmMePv_Hv_X3fjh4iEn11gzzpUJz6_o,610
 recce/tasks/core.py,sha256=JFYa1CfgOiRPQ7KVTwMuxJjhMB-pvCwB-xezVt-h3RU,4080
 recce/tasks/dataframe.py,sha256=03UBWwt0DFTXlaEOtnV5i_mxdRKD7UbRayewEL2Ub48,3650
@@ -120,16 +125,15 @@ recce/util/pydantic_model.py,sha256=KumKuyCjbTzEMsKLE4-b-eZfp0gLhYDdmVtw1-hxiJw,
 recce/util/recce_cloud.py,sha256=mqZn63Fh1kZaT0L6AR9K4yqTnGpZBGEWGBHFapLmS_w,9882
 recce/util/singleton.py,sha256=1cU99I0f9tjuMQLMJyLsK1oK3fZJMsO5-TbRHAMXqds,627
 recce/yaml/__init__.py,sha256=EgXYlFeJZchatUClRDXbIC5Oqb2_nBvB2NqItYVihio,1292
-recce_nightly-1.15.0.20250812.dist-info/licenses/LICENSE,sha256=CQjjMy9aYPhfe8xG_bcpIfKtNkdxLZ5IOb8oPygtUhY,11343
+recce_nightly-1.16.0.20250814.dist-info/licenses/LICENSE,sha256=CQjjMy9aYPhfe8xG_bcpIfKtNkdxLZ5IOb8oPygtUhY,11343
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/test_cli.py,sha256=_ikuZW4bYEGyj5A_V7nDSyBa1tVZqc0TVm-TJ0rEYVo,5467
+tests/test_cli.py,sha256=4N_F_lg00fGBuK3I9EyaF2Doi7PpKCrjqbuJ5YI1KqE,5467
 tests/test_config.py,sha256=ODDFe_XF6gphmSmmc422dGLBaCCmG-IjDzTkD5SJsJE,1557
 tests/test_connect_to_cloud.py,sha256=b2fgV8L1iQBdEwh6RumMsIIyYg7GtMOMRz1dvE3WRPg,3059
-tests/test_core.py,sha256=7NKQc_8Krjsp3imMMmXHV6hQ7Kvw8QGE2nVtigebZYE,735
+tests/test_core.py,sha256=WgCFm8Au3YQI-V5UbZ6LA8irMNHRc7NWutIq2Mny0LE,6242
 tests/test_dbt.py,sha256=VzXvdoJNwwEmKNhJJDNB1N_qZYwp1aoJQ1sLcoyRBmk,1316
 tests/test_pull_request.py,sha256=HmZo5MoDaoKSgPwbLxJ3Ur3ajZ7IxhkzJxaOmhg6bwE,3562
-tests/test_server.py,sha256=l2EDIonpDU3CzuCpZSXTCyH_X32xdscSz-bdNSuTkjA,3397
-tests/test_state.py,sha256=phSYlyxTx2Sq0K8DXUrfaO5Jc7lgZeeWNNtc9JD3fYc,4988
+tests/test_server.py,sha256=sq03awzG-kvLNJwHaZGDvVjd2uAs502XUKZ2nLNwvRQ,3391
 tests/test_summary.py,sha256=D0WvAkdO-vzGcvholH2rfS1wTxUXjVHwWm59fWy45eA,2876
 tests/adapter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/adapter/dbt_adapter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -149,8 +153,8 @@ tests/tasks/test_row_count.py,sha256=21PaP2aq-x8-pqwzWHRT1sixhQ8g3CQNRWOZTTmbK0s
 tests/tasks/test_schema.py,sha256=7ds4Vx8ixaiIWDR49Lvjem4xlPkRP1cXazDRY3roUak,3121
 tests/tasks/test_top_k.py,sha256=YR_GS__DJsbDlQVaEEdJvNQ3fh1VmV5Nb3G7lb0r6YM,1779
 tests/tasks/test_valuediff.py,sha256=_xQJGgxsXoy2NYk_Z6Hsw2FlVh6zk2nN_iUueyRN1e8,2046
-recce_nightly-1.15.0.20250812.dist-info/METADATA,sha256=NsbJ0iWJbxRjdQTOyNZem6C5qigbSGflMe_RLPCwOYU,9366
-recce_nightly-1.15.0.20250812.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-recce_nightly-1.15.0.20250812.dist-info/entry_points.txt,sha256=oqoY_IiwIqXbgrIsPnlqUqao2eiIeP2dprowkOlmeyg,40
-recce_nightly-1.15.0.20250812.dist-info/top_level.txt,sha256=6PKGVpf75idP0C6KEaldDzzZUauIxNu1ZDstau1pI4I,12
-recce_nightly-1.15.0.20250812.dist-info/RECORD,,
+recce_nightly-1.16.0.20250814.dist-info/METADATA,sha256=apeErc90TLIbChqRgH2pgqMChGOzJBZO6qrPLUMqm_A,9366
+recce_nightly-1.16.0.20250814.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+recce_nightly-1.16.0.20250814.dist-info/entry_points.txt,sha256=oqoY_IiwIqXbgrIsPnlqUqao2eiIeP2dprowkOlmeyg,40
+recce_nightly-1.16.0.20250814.dist-info/top_level.txt,sha256=6PKGVpf75idP0C6KEaldDzzZUauIxNu1ZDstau1pI4I,12
+recce_nightly-1.16.0.20250814.dist-info/RECORD,,

tests/test_cli.py

@@ -6,7 +6,7 @@ from click.testing import CliRunner
 from recce.cli import run as cli_command_run
 from recce.cli import server as cli_command_server
 from recce.core import RecceContext
-from recce.state import RecceStateLoader
+from recce.state import CloudStateLoader
 
 
 def test_cmd_version():
@@ -48,11 +48,11 @@ class TestCommandServer(TestCase):
     @patch.object(RecceContext, "verify_required_artifacts")
     @patch("recce.util.recce_cloud.get_recce_cloud_onboarding_state")
     @patch("recce.cli.uvicorn.run")
-    @patch("recce.cli.RecceStateLoader")
+    @patch("recce.cli.CloudStateLoader")
     def test_cmd_server_with_cloud(
         self, mock_state_loader_class, mock_run, mock_get_recce_cloud_onboarding_state, mock_verify_required_artifacts
     ):
-        mock_state_loader = MagicMock(spec=RecceStateLoader)
+        mock_state_loader = MagicMock(spec=CloudStateLoader)
         mock_state_loader.verify.return_value = True
         mock_state_loader.review_mode = True
         mock_get_recce_cloud_onboarding_state.return_value = "completed"

tests/test_core.py

@@ -1,6 +1,153 @@
 # noinspection PyUnresolvedReferences
+import os
+import unittest
+from datetime import datetime
+
+from recce.core import RecceContext
+from recce.models import Check, Run, RunType
+from recce.state import ArtifactsRoot, FileStateLoader, RecceState
 from tests.adapter.dbt_adapter.conftest import dbt_test_helper  # noqa: F401
 
+current_dir = os.path.dirname(os.path.abspath(__file__))
+
+
+class TestRecceState(unittest.TestCase):
+    def test_load(self):
+        run = Run(type=RunType.QUERY, params=dict(sql_template="select * from users"))
+        check = Check(name="check 1", description="desc 1", type=run.type, params=run.params)
+
+        state = RecceState(runs=[run], checks=[check])
+        json_content = state.to_json()
+        new_state = RecceState.from_json(json_content)
+
+        run_loaded = new_state.runs[0]
+        check_loaded = new_state.checks[0]
+
+        assert run.run_id == run_loaded.run_id
+        assert check.check_id == check_loaded.check_id
+
+    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):
+        # copy ./recce_state.json to temp and open
+
+        # use library to create a temp file in the context
+        import os
+        import shutil
+        import tempfile
+
+        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 = FileStateLoader(state_file=f.name)
+            state = state_loader.load()
+            assert len(state.runs) == 17
+
 
 def test_lineage_diff(dbt_test_helper):
     sql_model1 = """

tests/test_server.py

@@ -31,9 +31,9 @@ def test_health():
 
 def test_stateless(dbt_test_helper):
     context = default_context()
-    from recce.state import RecceStateLoader
+    from recce.state import FileStateLoader
 
-    context.state_loader = RecceStateLoader()
+    context.state_loader = FileStateLoader()
     client = TestClient(app)
     response = client.get("/api/info")
     assert response.status_code == 200
@@ -44,9 +44,9 @@ def test_stateless(dbt_test_helper):
 
 def test_file_mode(dbt_test_helper):
     context = default_context()
-    from recce.state import RecceStateLoader
+    from recce.state import FileStateLoader
 
-    context.state_loader = RecceStateLoader(state_file="/tmp/recce_state.json")
+    context.state_loader = FileStateLoader(state_file="/tmp/recce_state.json")
     client = TestClient(app)
     response = client.get("/api/info")
     assert response.status_code == 200
@@ -63,9 +63,9 @@ def test_saveas_and_rename(dbt_test_helper, temp_folder):
     state_file3 = os.path.join(temp_folder, "recce_state3.json")
     os.makedirs(os.path.join(temp_folder, "dir.json"))
 
-    from recce.state import RecceStateLoader
+    from recce.state import FileStateLoader
 
-    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"})