feat: add tests

Author: arnav-on-code

config/settings.py

@@ -20,6 +20,8 @@ class Config:
     GITHUB_USERNAME = os.getenv("GITHUB_USERNAME", "")
     GITHUB_EMAIL = os.getenv("GITHUB_EMAIL", "")
 
+    DEBUG = False
+
     STORAGE_DIR = BASE_DIR / os.getenv("STORAGE_DIR", "storage")
     LOGS_DIR = BASE_DIR / os.getenv("LOGS_DIR", "logs")
 

leetcode/client.py

@@ -17,10 +17,13 @@ def post(self, query: str, variables: dict = None):
             json=payload,
             timeout=Config.REQUEST_TIMEOUT
         )
+        if Config.DEBUG:
 
-        print("=" * 80)
-        print("Status Code:", response.status_code)
-        print(response.text)
-        print("=" * 80)
+            print("=" * 80)
+            print("Status Code:", response.status_code)
+            print(response.text)
+            print("=" * 80)
+
+        
 
         return response.json()

sync/manager.py

@@ -34,28 +34,45 @@ def run(self):
         Run one synchronization cycle.
         """
 
+        # Fetch recent accepted submissions
         submissions = self.api.get_recent_submissions()
 
+        # Detect which ones haven't been synced yet
         new_submissions = self.detector.find_new(submissions)
 
         if not new_submissions:
             print("No new submissions found.")
             return
 
-        latest = new_submissions[0]
+        print(f"Found {len(new_submissions)} new submission(s).\n")
 
-        detail = self.api.get_submission_detail(latest.id)
+        for submission in new_submissions:
+            try:
+                # Fetch complete submission details
+                detail = self.api.get_submission_detail(
+                    submission.id
+                )
 
-        self.downloader.download(detail)
+                # Download solution and metadata
+                self.downloader.download(detail)
 
-        self.git.add()
+                
+                self.git.add()
 
-        self.git.commit(
-            generate_commit_message(detail)
-        )
+                
+                self.git.commit(
+                    generate_commit_message(detail)
+                )
+
+                
+                self.git.push()
+
+                self.detector.update(submission)
 
-        self.git.push()
+                print(f"✓ Synced: {submission.title}")
 
-        self.detector.update(latest)
+            except Exception as e:
+                print(f"✗ Failed: {submission.title}")
+                print(e)
 
-        print(f"Successfully synced: {latest.title}")
+        print("\nSynchronization complete.")

tests/test_readme.py tests/__init__.pytests/test_readme.py renamed to tests/__init__.py

@@ -0,0 +1,57 @@
+from datetime import datetime
+
+import pytest
+
+from leetcode.models import Submission
+from leetcode.models import SubmissionDetail
+
+
+@pytest.fixture
+def sample_submission():
+
+    return Submission(
+        id="2051842379",
+        title="Find Median from Data Stream",
+        slug="find-median-from-data-stream",
+        timestamp=1782879665,
+    )
+
+
+@pytest.fixture
+def sample_submission_detail():
+    """
+    Sample submission detail.
+    """
+
+    return SubmissionDetail(
+        submission_id="2051842379",
+        question_id="295",
+        title_slug="find-median-from-data-stream",
+
+        language="python3",
+        language_verbose="Python3",
+
+        runtime=177,
+        runtime_display="177 ms",
+
+        memory=41264000,
+        memory_display="41.3 MB",
+
+        status_code=10,
+
+        code=(
+            "class MedianFinder:\n"
+            "    pass\n"
+        ),
+
+        timestamp=1782879665,
+    )
+
+
+@pytest.fixture
+def temp_solution_dir(tmp_path):
+    """
+    Temporary directory for downloader tests.
+    """
+
+    return tmp_path / "solutions"

tests/conftest.py

@@ -0,0 +1,93 @@
+from unittest.mock import MagicMock
+
+from leetcode.api import LeetCodeAPI
+
+
+def test_get_profile():
+
+    client = MagicMock()
+
+    client.post.return_value = {
+        "data": {
+            "matchedUser": {
+                "username": "arnav-on-code"
+            }
+        }
+    }
+
+    api = LeetCodeAPI(client)
+
+    profile = api.get_profile()
+
+    assert profile["username"] == "arnav-on-code"
+
+
+def test_get_recent_submissions():
+
+    client = MagicMock()
+
+    client.post.return_value = {
+        "data": {
+            "recentAcSubmissionList": [
+                {
+                    "id": "1",
+                    "title": "Two Sum",
+                    "titleSlug": "two-sum",
+                    "timestamp": "1780000000",
+                }
+            ]
+        }
+    }
+
+    api = LeetCodeAPI(client)
+
+    submissions = api.get_recent_submissions()
+
+    assert len(submissions) == 1
+
+    assert submissions[0].id == "1"
+
+    assert submissions[0].title == "Two Sum"
+
+    assert submissions[0].slug == "two-sum"
+
+
+def test_get_submission_detail():
+
+    client = MagicMock()
+
+    client.post.return_value = {
+        "data": {
+            "submissionDetails": {
+                "question": {
+                    "questionId": "1",
+                    "titleSlug": "two-sum",
+                },
+                "lang": {
+                    "name": "python3",
+                    "verboseName": "Python3",
+                },
+                "runtime": 45,
+                "runtimeDisplay": "45 ms",
+                "memory": 17000000,
+                "memoryDisplay": "17 MB",
+                "statusCode": 10,
+                "code": "print('hello')",
+                "timestamp": "1780000000",
+            }
+        }
+    }
+
+    api = LeetCodeAPI(client)
+
+    detail = api.get_submission_detail("123")
+
+    assert detail.submission_id == "123"
+
+    assert detail.question_id == "1"
+
+    assert detail.title_slug == "two-sum"
+
+    assert detail.language == "python3"
+
+    assert detail.status_code == 10

tests/test_api.py

@@ -0,0 +1,77 @@
+from leetcode.parser import (
+    parse_submissions,
+    parse_submission_detail,
+)
+
+
+def test_parse_submissions():
+
+    data = [
+        {
+            "id": "123",
+            "title": "Two Sum",
+            "titleSlug": "two-sum",
+            "timestamp": "1780000000",
+        }
+    ]
+
+    submissions = parse_submissions(data)
+
+    assert len(submissions) == 1
+
+    submission = submissions[0]
+
+    assert submission.id == "123"
+    assert submission.title == "Two Sum"
+    assert submission.slug == "two-sum"
+    assert submission.timestamp == 1780000000
+
+
+def test_parse_submission_detail():
+
+    data = {
+        "question": {
+            "questionId": "1",
+            "titleSlug": "two-sum",
+        },
+        "lang": {
+            "name": "python3",
+            "verboseName": "Python3",
+        },
+        "runtime": 45,
+        "runtimeDisplay": "45 ms",
+        "memory": 17000000,
+        "memoryDisplay": "17 MB",
+        "statusCode": 10,
+        "code": "print('hello')",
+        "timestamp": "1780000000",
+    }
+
+    detail = parse_submission_detail(
+        "999",
+        data,
+    )
+
+    assert detail.submission_id == "999"
+
+    assert detail.question_id == "1"
+
+    assert detail.title_slug == "two-sum"
+
+    assert detail.language == "python3"
+
+    assert detail.language_verbose == "Python3"
+
+    assert detail.runtime == 45
+
+    assert detail.runtime_display == "45 ms"
+
+    assert detail.memory == 17000000
+
+    assert detail.memory_display == "17 MB"
+
+    assert detail.status_code == 10
+
+    assert detail.code == "print('hello')"
+
+    assert detail.timestamp == 1780000000

tests/test_parser.py

@@ -0,0 +1,55 @@
+import logging
+from pathlib import Path
+
+from config.settings import Config
+
+
+class Logger:
+    """
+    Application logger.
+    """
+
+    _logger = None
+
+    @classmethod
+    def get_logger(cls) -> logging.Logger:
+
+        if cls._logger is not None:
+            return cls._logger
+
+        log_file = Config.LOGS_DIR / "sync.log"
+
+        logger = logging.getLogger("leetcode_sync")
+
+        logger.setLevel(logging.INFO)
+
+        logger.propagate = False
+
+        if logger.handlers:
+            return logger
+
+        formatter = logging.Formatter(
+            "%(asctime)s | %(levelname)-8s | %(message)s",
+            "%Y-%m-%d %H:%M:%S",
+        )
+
+        # Log to file
+        file_handler = logging.FileHandler(
+            log_file,
+            encoding="utf-8",
+        )
+
+        file_handler.setFormatter(formatter)
+
+        logger.addHandler(file_handler)
+
+        # Log to console
+        console_handler = logging.StreamHandler()
+
+        console_handler.setFormatter(formatter)
+
+        logger.addHandler(console_handler)
+
+        cls._logger = logger
+
+        return logger