Created hashed customers. migrated to service instead of using db logic directly
This commit is contained in:
216
tests/test_customer_service.py
Normal file
216
tests/test_customer_service.py
Normal file
@@ -0,0 +1,216 @@
|
||||
"""Tests for CustomerService functionality."""
|
||||
|
||||
import pytest
|
||||
import pytest_asyncio
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
|
||||
|
||||
from alpine_bits_python.customer_service import CustomerService
|
||||
from alpine_bits_python.db import Base, Customer, HashedCustomer
|
||||
|
||||
|
||||
@pytest_asyncio.fixture
|
||||
async def async_session():
|
||||
"""Create an async session for testing."""
|
||||
engine = create_async_engine("sqlite+aiosqlite:///:memory:", echo=False)
|
||||
async with engine.begin() as conn:
|
||||
await conn.run_sync(Base.metadata.create_all)
|
||||
|
||||
async_session_maker = async_sessionmaker(engine, expire_on_commit=False)
|
||||
async with async_session_maker() as session:
|
||||
yield session
|
||||
|
||||
await engine.dispose()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_create_customer_creates_hashed_version(async_session: AsyncSession):
|
||||
"""Test that creating a customer automatically creates hashed version."""
|
||||
service = CustomerService(async_session)
|
||||
|
||||
customer_data = {
|
||||
"given_name": "John",
|
||||
"surname": "Doe",
|
||||
"contact_id": "test123",
|
||||
"email_address": "john@example.com",
|
||||
"phone": "+1234567890",
|
||||
}
|
||||
|
||||
customer = await service.create_customer(customer_data)
|
||||
|
||||
assert customer.id is not None
|
||||
assert customer.given_name == "John"
|
||||
|
||||
# Check that hashed version was created
|
||||
hashed = await service.get_hashed_customer(customer.id)
|
||||
assert hashed is not None
|
||||
assert hashed.customer_id == customer.id
|
||||
assert hashed.hashed_email is not None
|
||||
assert hashed.hashed_phone is not None
|
||||
assert hashed.hashed_given_name is not None
|
||||
assert hashed.hashed_surname is not None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_update_customer_updates_hashed_version(async_session: AsyncSession):
|
||||
"""Test that updating a customer updates the hashed version."""
|
||||
service = CustomerService(async_session)
|
||||
|
||||
# Create initial customer
|
||||
customer_data = {
|
||||
"given_name": "John",
|
||||
"surname": "Doe",
|
||||
"contact_id": "test123",
|
||||
"email_address": "john@example.com",
|
||||
}
|
||||
customer = await service.create_customer(customer_data)
|
||||
|
||||
# Get initial hashed email
|
||||
hashed = await service.get_hashed_customer(customer.id)
|
||||
original_hashed_email = hashed.hashed_email
|
||||
|
||||
# Update customer email
|
||||
update_data = {"email_address": "newemail@example.com"}
|
||||
updated_customer = await service.update_customer(customer, update_data)
|
||||
|
||||
# Check that hashed version was updated
|
||||
updated_hashed = await service.get_hashed_customer(updated_customer.id)
|
||||
assert updated_hashed.hashed_email != original_hashed_email
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_or_create_customer_creates_new(async_session: AsyncSession):
|
||||
"""Test get_or_create creates new customer when not found."""
|
||||
service = CustomerService(async_session)
|
||||
|
||||
customer_data = {
|
||||
"given_name": "Jane",
|
||||
"surname": "Smith",
|
||||
"contact_id": "new123",
|
||||
"email_address": "jane@example.com",
|
||||
}
|
||||
|
||||
customer = await service.get_or_create_customer(customer_data)
|
||||
assert customer.id is not None
|
||||
assert customer.contact_id == "new123"
|
||||
|
||||
# Verify hashed version exists
|
||||
hashed = await service.get_hashed_customer(customer.id)
|
||||
assert hashed is not None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_or_create_customer_updates_existing(async_session: AsyncSession):
|
||||
"""Test get_or_create updates existing customer when found."""
|
||||
service = CustomerService(async_session)
|
||||
|
||||
# Create initial customer
|
||||
customer_data = {
|
||||
"given_name": "Jane",
|
||||
"surname": "Smith",
|
||||
"contact_id": "existing123",
|
||||
"email_address": "jane@example.com",
|
||||
}
|
||||
original_customer = await service.create_customer(customer_data)
|
||||
|
||||
# Try to create again with same contact_id but different data
|
||||
updated_data = {
|
||||
"given_name": "Janet",
|
||||
"surname": "Smith",
|
||||
"contact_id": "existing123",
|
||||
"email_address": "janet@example.com",
|
||||
}
|
||||
customer = await service.get_or_create_customer(updated_data)
|
||||
|
||||
# Should be same customer ID but updated data
|
||||
assert customer.id == original_customer.id
|
||||
assert customer.given_name == "Janet"
|
||||
assert customer.email_address == "janet@example.com"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_hash_existing_customers_backfills(async_session: AsyncSession):
|
||||
"""Test that hash_existing_customers backfills missing hashed data."""
|
||||
# Create a customer directly in DB without using service
|
||||
customer = Customer(
|
||||
given_name="Bob",
|
||||
surname="Builder",
|
||||
contact_id="bob123",
|
||||
email_address="bob@example.com",
|
||||
phone="+9876543210",
|
||||
)
|
||||
async_session.add(customer)
|
||||
await async_session.commit()
|
||||
await async_session.refresh(customer)
|
||||
|
||||
# Verify no hashed version exists
|
||||
result = await async_session.execute(
|
||||
select(HashedCustomer).where(HashedCustomer.customer_id == customer.id)
|
||||
)
|
||||
hashed = result.scalar_one_or_none()
|
||||
assert hashed is None
|
||||
|
||||
# Run backfill
|
||||
service = CustomerService(async_session)
|
||||
count = await service.hash_existing_customers()
|
||||
|
||||
assert count == 1
|
||||
|
||||
# Verify hashed version now exists
|
||||
result = await async_session.execute(
|
||||
select(HashedCustomer).where(HashedCustomer.customer_id == customer.id)
|
||||
)
|
||||
hashed = result.scalar_one_or_none()
|
||||
assert hashed is not None
|
||||
assert hashed.hashed_email is not None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_hash_existing_customers_skips_already_hashed(
|
||||
async_session: AsyncSession,
|
||||
):
|
||||
"""Test that hash_existing_customers skips customers already hashed."""
|
||||
service = CustomerService(async_session)
|
||||
|
||||
# Create customer with service (creates hashed version)
|
||||
customer_data = {
|
||||
"given_name": "Alice",
|
||||
"surname": "Wonder",
|
||||
"contact_id": "alice123",
|
||||
"email_address": "alice@example.com",
|
||||
}
|
||||
await service.create_customer(customer_data)
|
||||
|
||||
# Run backfill - should return 0 since customer is already hashed
|
||||
count = await service.hash_existing_customers()
|
||||
assert count == 0
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_hashing_normalization(async_session: AsyncSession):
|
||||
"""Test that hashing properly normalizes data."""
|
||||
service = CustomerService(async_session)
|
||||
|
||||
# Create customer with mixed case and spaces
|
||||
customer_data = {
|
||||
"given_name": " John ",
|
||||
"surname": "DOE",
|
||||
"contact_id": "test123",
|
||||
"email_address": " John.Doe@Example.COM ",
|
||||
"phone": "+1 (234) 567-8900",
|
||||
}
|
||||
customer = await service.create_customer(customer_data)
|
||||
|
||||
hashed = await service.get_hashed_customer(customer.id)
|
||||
|
||||
# Verify hashes exist (normalization should have occurred)
|
||||
assert hashed.hashed_email is not None
|
||||
assert hashed.hashed_phone is not None
|
||||
|
||||
# Hash should be consistent for same normalized value
|
||||
from alpine_bits_python.db import Customer as CustomerModel
|
||||
|
||||
normalized_email_hash = CustomerModel._normalize_and_hash(
|
||||
"john.doe@example.com"
|
||||
)
|
||||
assert hashed.hashed_email == normalized_email_hash
|
||||
Reference in New Issue
Block a user