Feat/get stock list adapter (#9)

* refactor: file structure

* feat: add list stock adapter

* test: add testing

---------

Co-authored-by: Eyo Chen <eyo.chen@amazingtalker.com>

Author: eyo-chen

src/adapters/repositories/stock/base.py src/adapters/base.pysrc/adapters/repositories/stock/base.py renamed to src/adapters/base.py

@@ -1,8 +1,13 @@
+from typing import List
 from abc import ABC, abstractmethod
-from domain.stock import CreateStock
+
+from domain.stock import CreateStock, Stock
 
 
 class AbstractStockRepository(ABC):
     @abstractmethod
     def create(self, stock: CreateStock) -> str:
         """Create a new stock entry in the repository."""
+
+    def list(self, user_id: int) -> List[Stock]:
+        """List all stock by user id"""

src/adapters/repositories/stock/stock.py src/adapters/stock.pysrc/adapters/repositories/stock/stock.py renamed to src/adapters/stock.py

@@ -1,7 +1,10 @@
+from typing import List
+
 from pymongo import MongoClient
 from pymongo.database import Database
+
 from .base import AbstractStockRepository
-from domain.stock import CreateStock
+from domain.stock import CreateStock, Stock, ActionType
 
 
 class StockRepository(AbstractStockRepository):
@@ -24,5 +27,24 @@ def create(self, stock: CreateStock) -> str:
         result = self.collection.insert_one(stock_dict)
         return str(result.inserted_id)
 
+    def list(self, user_id: int) -> List[Stock]:
+        stock_docs = self.collection.find({"user_id": user_id})
+
+        stocks = [
+            Stock(
+                id=str(doc["_id"]),
+                user_id=doc["user_id"],
+                symbol=doc["symbol"],
+                price=doc["price"],
+                quantity=doc["quantity"],
+                action_type=ActionType(doc["action_type"]),
+                created_at=doc["created_at"],
+                updated_at=doc["updated_at"],
+            )
+            for doc in stock_docs
+        ]
+
+        return stocks
+
     def __del__(self):
         self.client.close()

src/domain/stock.py

@@ -20,3 +20,15 @@ class CreateStock:
     quantity: int
     action_type: ActionType
     created_at: datetime
+
+
+@dataclass
+class Stock:
+    id: str
+    user_id: int
+    symbol: str
+    price: float
+    quantity: int
+    action_type: ActionType
+    created_at: datetime
+    updated_at: datetime

src/index.py

@@ -4,7 +4,7 @@
 from pymongo import MongoClient
 from dotenv import load_dotenv
 from handler.stock import StockService
-from adapters.repositories.stock.stock import StockRepository
+from adapters.stock import StockRepository
 from usecase.stock.stock import StockUsecase
 
 

src/tests/test_stock_adapters.py

@@ -1,9 +1,11 @@
 import pytest
-from datetime import datetime
+from datetime import datetime, timezone
+
 from pymongo import MongoClient
 from bson.objectid import ObjectId
-from domain.stock import CreateStock, ActionType
-from adapters.repositories.stock.stock import StockRepository
+
+from domain.stock import CreateStock, ActionType, Stock
+from adapters.stock import StockRepository
 
 
 @pytest.fixture(scope="module")
@@ -115,3 +117,133 @@ def test_create_multiple_stocks(self, stock_repository):
 
         # Compare the sets
         assert expected_set == actual_set, f"Expected {expected_set}, but got {actual_set}"
+
+    def test_list_stocks(self, stock_repository):
+        # Create mock stock data
+        created_at = datetime.now(timezone.utc)
+        mock_stock1 = {
+            "user_id": 1,
+            "symbol": "AAPL",
+            "price": 150.25,
+            "quantity": 100,
+            "action_type": ActionType.BUY.value,
+            "created_at": created_at,
+            "updated_at": created_at,
+        }
+        mock_stock2 = {
+            "user_id": 1,
+            "symbol": "TSLA",
+            "price": 110.25,
+            "quantity": 50,
+            "action_type": ActionType.SELL.value,
+            "created_at": created_at,
+            "updated_at": created_at,
+        }
+        mock_stock3 = {
+            "user_id": 2,  # Different user_id
+            "symbol": "GOOGL",
+            "price": 2500.50,
+            "quantity": 10,
+            "action_type": ActionType.BUY.value,
+            "created_at": created_at,
+            "updated_at": created_at,
+        }
+
+        # Insert mock stocks directly into the MongoDB collection
+        result1 = stock_repository.collection.insert_one(mock_stock1)
+        result2 = stock_repository.collection.insert_one(mock_stock2)
+        stock_repository.collection.insert_one(mock_stock3)
+
+        # Get the inserted IDs
+        stock_id1 = str(result1.inserted_id)
+        stock_id2 = str(result2.inserted_id)
+
+        # Expected Stock objects for user_id=1
+        expected_stocks = [
+            Stock(
+                id=stock_id1,
+                user_id=mock_stock1["user_id"],
+                symbol=mock_stock1["symbol"],
+                price=mock_stock1["price"],
+                quantity=mock_stock1["quantity"],
+                action_type=ActionType(mock_stock1["action_type"]),
+                created_at=mock_stock1["created_at"],
+                updated_at=mock_stock1["updated_at"],
+            ),
+            Stock(
+                id=stock_id2,
+                user_id=mock_stock2["user_id"],
+                symbol=mock_stock2["symbol"],
+                price=mock_stock2["price"],
+                quantity=mock_stock2["quantity"],
+                action_type=ActionType(mock_stock2["action_type"]),
+                created_at=mock_stock2["created_at"],
+                updated_at=mock_stock2["updated_at"],
+            ),
+        ]
+
+        # Action
+        result = stock_repository.list(1)
+
+        # Assertions
+        assert len(result) == 2, f"Expected 2 stocks, but got {len(result)}"
+        assert all(isinstance(stock, Stock) for stock in result), "All results should be Stock objects"
+
+        # Convert to sets for comparison (to ignore order)
+        result_set = {
+            (
+                stock.id,
+                stock.user_id,
+                stock.symbol,
+                stock.price,
+                stock.quantity,
+                stock.action_type,
+            )
+            for stock in result
+        }
+        expected_set = {
+            (
+                stock.id,
+                stock.user_id,
+                stock.symbol,
+                stock.price,
+                stock.quantity,
+                stock.action_type,
+            )
+            for stock in expected_stocks
+        }
+
+        assert result_set == expected_set, f"Expected {expected_set}, but got {result_set}"
+
+    def test_list_stocks_no_data(self, stock_repository):
+        # Create mock stock data for user_id=1
+        created_at = datetime.now(timezone.utc)
+        mock_stock1 = {
+            "user_id": 1,
+            "symbol": "AAPL",
+            "price": 150.25,
+            "quantity": 100,
+            "action_type": ActionType.BUY.value,
+            "created_at": created_at,
+            "updated_at": created_at,
+        }
+        mock_stock2 = {
+            "user_id": 1,
+            "symbol": "TSLA",
+            "price": 110.25,
+            "quantity": 50,
+            "action_type": ActionType.SELL.value,
+            "created_at": created_at,
+            "updated_at": created_at,
+        }
+
+        # Insert mock stocks directly into the MongoDB collection
+        stock_repository.collection.insert_many([mock_stock1, mock_stock2])
+
+        # Query for a user_id with no stock data
+        result = stock_repository.list(999)  # Non-existent user_id
+
+        # Assertions
+        assert len(result) == 0, f"Expected empty list, but got {len(result)} stocks"
+        assert isinstance(result, list), "Result should be a list"
+        assert all(isinstance(stock, Stock) for stock in result), "All results should be Stock objects (if any)"

src/usecase/stock/stock.py

@@ -1,5 +1,5 @@
 from domain.stock import CreateStock
-from adapters.repositories.stock.base import AbstractStockRepository
+from adapters.base import AbstractStockRepository
 from .base import AbstractStockUsecase