Tests restructuring

tests/unit/models/test_database_models.py

@@ -0,0 +1,96 @@
+# tests/unit/models/test_database_models.py
+import pytest
+
+from models.database_models import Product, Shop, Stock, User
+
+
+@pytest.mark.unit
+@pytest.mark.database
+class TestDatabaseModels:
+    def test_user_model(self, db):
+        """Test User model creation and relationships"""
+        user = User(
+            email="db_test@example.com",
+            username="dbtest",
+            hashed_password="hashed_password_123",
+            role="user",
+            is_active=True,
+        )
+
+        db.add(user)
+        db.commit()
+        db.refresh(user)
+
+        assert user.id is not None
+        assert user.email == "db_test@example.com"
+        assert user.created_at is not None
+        assert user.updated_at is not None
+
+    def test_product_model(self, db):
+        """Test Product model creation"""
+        product = Product(
+            product_id="DB_TEST_001",
+            title="Database Test Product",
+            description="Testing product model",
+            price="25.99",
+            currency="USD",
+            brand="DBTest",
+            gtin="1234567890123",
+            availability="in stock",
+            marketplace="TestDB",
+            shop_name="DBTestShop",
+        )
+
+        db.add(product)
+        db.commit()
+        db.refresh(product)
+
+        assert product.id is not None
+        assert product.product_id == "DB_TEST_001"
+        assert product.created_at is not None
+
+    def test_stock_model(self, db):
+        """Test Stock model creation"""
+        stock = Stock(gtin="1234567890123", location="DB_WAREHOUSE", quantity=150)
+
+        db.add(stock)
+        db.commit()
+        db.refresh(stock)
+
+        assert stock.id is not None
+        assert stock.gtin == "1234567890123"
+        assert stock.location == "DB_WAREHOUSE"
+        assert stock.quantity == 150
+
+    def test_shop_model_with_owner(self, db, test_user):
+        """Test Shop model with owner relationship"""
+        shop = Shop(
+            shop_code="DBTEST",
+            shop_name="Database Test Shop",
+            description="Testing shop model",
+            owner_id=test_user.id,
+            is_active=True,
+            is_verified=False,
+        )
+
+        db.add(shop)
+        db.commit()
+        db.refresh(shop)
+
+        assert shop.id is not None
+        assert shop.shop_code == "DBTEST"
+        assert shop.owner_id == test_user.id
+        assert shop.owner.username == test_user.username
+
+    def test_database_constraints(self, db):
+        """Test database constraints and unique indexes"""
+        # Test unique product_id constraint
+        product1 = Product(product_id="UNIQUE_001", title="Product 1")
+        db.add(product1)
+        db.commit()
+
+        # This should raise an integrity error
+        with pytest.raises(Exception):  # Could be IntegrityError or similar
+            product2 = Product(product_id="UNIQUE_001", title="Product 2")
+            db.add(product2)
+            db.commit()