new api service and logic implemented

Ztest/test_mongo.py

@@ -0,0 +1,104 @@
+"""Test MongoDB actions and models."""
+
+import pytest
+from pymongo import MongoClient
+
+from Services.MongoDb.Models.actions import MongoActions
+from Services.MongoDb.Models.action_models.domain import (
+    DomainData,
+    DomainDocumentCreate,
+    DomainDocumentUpdate,
+)
+from AllConfigs.NoSqlDatabase.configs import MongoConfig
+
+
+@pytest.fixture
+def mongo_client():
+    """Create MongoDB test client."""
+    # Connect using configured credentials
+    client = MongoClient(MongoConfig.URL)
+    client.admin.command("ping")  # Test connection
+    yield client
+    client.close()
+
+
+@pytest.fixture
+def mongo_actions(mongo_client):
+    """Create MongoActions instance for testing."""
+    if not mongo_client:
+        pytest.skip("MongoDB connection not available")
+
+    actions = MongoActions(
+        client=mongo_client,
+        database=MongoConfig.DATABASE_NAME,
+        company_uuid="test_company",
+        storage_reason="domains",
+    )
+    yield actions
+    try:
+        # Cleanup after tests
+        if actions.collection is not None:
+            actions.collection.drop()
+    except Exception as e:
+        print(f"Failed to cleanup test collection: {e}")
+
+
+def test_mongo_crud_operations(mongo_actions: MongoActions):
+    """Test CRUD operations with MongoActions."""
+
+    # Create test data
+    domain_data = DomainData(
+        user_uu_id="test_user",
+        main_domain="example.com",
+        other_domains_list=["old.com"],
+    )
+    create_doc = DomainDocumentCreate(data=domain_data)
+
+    # Test create
+    result = mongo_actions.insert_one(create_doc.model_dump())
+    assert result.inserted_id is not None
+
+    # Test read
+    doc = mongo_actions.find_one({"data.main_domain": "example.com"})
+    assert doc is not None
+    assert doc["data"]["main_domain"] == "example.com"
+
+    # Test update
+    update_data = DomainData(
+        user_uu_id="test_user",
+        main_domain="new.com",
+        other_domains_list=["example.com", "old.com"],
+    )
+    update_doc = DomainDocumentUpdate(data=update_data)
+    result = mongo_actions.update_one(
+        {"_id": doc["_id"]}, {"$set": update_doc.model_dump()}
+    )
+    assert result.modified_count == 1
+
+    # Test delete
+    result = mongo_actions.delete_one({"_id": doc["_id"]})
+    assert result.deleted_count == 1
+
+
+def test_mongo_aggregate(mongo_actions: MongoActions):
+    """Test aggregate operations with MongoActions."""
+
+    # Insert test documents
+    docs = [
+        DomainDocumentCreate(
+            data=DomainData(user_uu_id="user1", main_domain=f"domain{i}.com")
+        ).model_dump()
+        for i in range(3)
+    ]
+    mongo_actions.insert_many(docs)
+
+    # Test aggregation
+    pipeline = [{"$group": {"_id": "$data.user_uu_id", "count": {"$sum": 1}}}]
+    result = mongo_actions.aggregate(pipeline)
+    result_list = list(result)
+    assert len(result_list) == 1
+    assert result_list[0]["count"] == 3
+
+
+if __name__ == "__main__":
+    pytest.main([__file__, "-v"])