geutebruck/test_crud_operations.py

"""
Test all CRUD operations: CREATE, READ, UPDATE, DELETE
"""
import grpc
import sys
import io
sys.path.append(r'C:\DEV\COPILOT\geutebruck-api\src\api')

if sys.platform == 'win32':
    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
    sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding='utf-8')

from protos import configuration_pb2
from protos import configuration_pb2_grpc

print("="*70)
print("TESTING CRUD OPERATIONS FOR ACTION MAPPINGS")
print("="*70)

channel = grpc.insecure_channel('localhost:50051')
stub = configuration_pb2_grpc.ConfigurationServiceStub(channel)

# ============ READ - Get initial count ============
print("\n1. READ - Getting current mapping count...")
request = configuration_pb2.ReadActionMappingsRequest()
response = stub.ReadActionMappings(request)
initial_count = len(response.mappings)
print(f"   Initial count: {initial_count}")

# ============ CREATE - Add a new mapping ============
print("\n2. CREATE - Creating new mapping...")
output1 = configuration_pb2.ActionDefinition(action="GCoreDataBase")
output2 = configuration_pb2.ActionDefinition(action="GscMail")

mapping = configuration_pb2.ActionMappingInput(
    name="CRUD_TEST_MAPPING",
    video_input=101027,
    output_actions=[output1, output2]
)

create_request = configuration_pb2.CreateActionMappingRequest(mapping=mapping)
create_response = stub.CreateActionMapping(create_request)

if not create_response.success:
    print(f"   [ERROR] Create failed: {create_response.error_message}")
    sys.exit(1)

print(f"   ✓ Created successfully: {create_response.message}")

# Verify creation
verify_response = stub.ReadActionMappings(configuration_pb2.ReadActionMappingsRequest())
new_count = len(verify_response.mappings)
print(f"   Count after create: {new_count} (was {initial_count})")

# Find the new mapping ID (last one in the list)
mapping_id = new_count

# ============ UPDATE - Modify the mapping ============
print(f"\n3. UPDATE - Updating mapping ID {mapping_id}...")

updated_output = configuration_pb2.ActionDefinition(action="GscAction")
update_mapping = configuration_pb2.ActionMappingInput(
    name="CRUD_TEST_UPDATED",
    video_input=101028,  # Changed VideoInput
    output_actions=[updated_output]  # Changed to single action
)

update_request = configuration_pb2.UpdateActionMappingRequest(
    mapping_id=mapping_id,
    mapping=update_mapping
)

update_response = stub.UpdateActionMapping(update_request)

if not update_response.success:
    print(f"   [ERROR] Update failed: {update_response.error_message}")
else:
    print(f"   ✓ Updated successfully: {update_response.message}")
    print(f"   New name: {update_response.mapping.name}")

# ============ READ - Verify update ============
print(f"\n4. READ - Verifying update...")
verify_response = stub.ReadActionMappings(configuration_pb2.ReadActionMappingsRequest())
updated_mapping = verify_response.mappings[mapping_id - 1]
print(f"   Mapping {mapping_id}: {updated_mapping.name}")

# ============ DELETE - Remove the mapping ============
print(f"\n5. DELETE - Deleting mapping ID {mapping_id}...")

delete_request = configuration_pb2.DeleteActionMappingRequest(mapping_id=mapping_id)
delete_response = stub.DeleteActionMapping(delete_request)

if not delete_response.success:
    print(f"   [ERROR] Delete failed: {delete_response.error_message}")
else:
    print(f"   ✓ Deleted successfully: {delete_response.message}")

# ============ READ - Verify deletion ============
print(f"\n6. READ - Verifying deletion...")
final_response = stub.ReadActionMappings(configuration_pb2.ReadActionMappingsRequest())
final_count = len(final_response.mappings)
print(f"   Final count: {final_count} (should be {initial_count})")

if final_count == initial_count:
    print("\n" + "="*70)
    print("✓ ALL CRUD OPERATIONS SUCCESSFUL!")
    print("="*70)
else:
    print("\n" + "="*70)
    print(f"✗ Count mismatch: expected {initial_count}, got {final_count}")
    print("="*70)