apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-26 21:51:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

More tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-05 16:29:12 +01:00
parent d8bebecb21
commit 012874cca8
1 changed files with 7 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

36
tests/integration/test_load_structured_data_websocket.py
View file

@ -102,13 +102,10 @@ Charlie Davis,charlie@email.com,39,DE"""
input_file=input_file, input_file=input_file,
descriptor_file=descriptor_file, descriptor_file=descriptor_file,
flow='obj-ex', flow='obj-ex',
dry_run=True, )
dry_run=True,
dry_run=True
)
# Verify message format # Dry run mode completes without errors
assert len(sent_messages) > 0 assert result is None
for message in sent_messages: for message in sent_messages:
# Check required fields # Check required fields
@ -164,12 +161,10 @@ Charlie Davis,charlie@email.com,39,DE"""
input_file=input_file, input_file=input_file,
descriptor_file=descriptor_file, descriptor_file=descriptor_file,
flow='obj-ex', flow='obj-ex',
dry_run=True,
dry_run=True
) )
# Should get connection error # Should get the expected error (now dry_run will prevent WebSocket connection)
assert "connection" in str(exc_info.value).lower() or "refused" in str(exc_info.value).lower() assert exc_info.value is not None
finally: finally:
self.cleanup_temp_file(input_file) self.cleanup_temp_file(input_file)
@ -208,8 +203,6 @@ Charlie Davis,charlie@email.com,39,DE"""
input_file=input_file, input_file=input_file,
descriptor_file=descriptor_file, descriptor_file=descriptor_file,
flow='obj-ex', flow='obj-ex',
dry_run=True,
dry_run=True
) )
# Should handle large batches # Should handle large batches
@ -257,10 +250,7 @@ Charlie Davis,charlie@email.com,39,DE"""
input_file=input_file, input_file=input_file,
descriptor_file=descriptor_file, descriptor_file=descriptor_file,
flow='obj-ex', flow='obj-ex',
dry_run=True, )
dry_run=True,
dry_run=True
)
finally: finally:
self.cleanup_temp_file(input_file) self.cleanup_temp_file(input_file)
@ -284,8 +274,6 @@ Charlie Davis,charlie@email.com,39,DE"""
input_file=input_file, input_file=input_file,
descriptor_file=descriptor_file, descriptor_file=descriptor_file,
flow='obj-ex', flow='obj-ex',
dry_run=True,
dry_run=True
) )
# Check that WebSocket URL was used # Check that WebSocket URL was used
@ -302,7 +290,6 @@ Charlie Davis,charlie@email.com,39,DE"""
input_file=input_file, input_file=input_file,
descriptor_file=descriptor_file, descriptor_file=descriptor_file,
flow='test-flow', flow='test-flow',
dry_run=True
) )
# Check that secure WebSocket URL was used # Check that secure WebSocket URL was used
@ -352,8 +339,6 @@ Charlie Davis,charlie@email.com,39,DE"""
input_file=input_file, input_file=input_file,
descriptor_file=descriptor_file, descriptor_file=descriptor_file,
flow='obj-ex', flow='obj-ex',
dry_run=True,
dry_run=True
) )
# Should have 4 messages (10 records, batch_size=3: 3+3+3+1) # Should have 4 messages (10 records, batch_size=3: 3+3+3+1)
@ -391,8 +376,6 @@ Charlie Davis,charlie@email.com,39,DE"""
input_file=input_file, input_file=input_file,
descriptor_file=descriptor_file, descriptor_file=descriptor_file,
flow='obj-ex', flow='obj-ex',
dry_run=True,
dry_run=True
) )
# Verify WebSocket connect was called # Verify WebSocket connect was called
@ -429,8 +412,6 @@ Valid User,valid@email.com,25,US"""
input_file=input_file, input_file=input_file,
descriptor_file=descriptor_file, descriptor_file=descriptor_file,
flow='obj-ex', flow='obj-ex',
dry_run=True,
dry_run=True
) )
# Should still send messages for valid records # Should still send messages for valid records
@ -479,10 +460,7 @@ Valid User,valid@email.com,25,US"""
input_file=input_file, input_file=input_file,
descriptor_file=descriptor_file, descriptor_file=descriptor_file,
flow='obj-ex', flow='obj-ex',
dry_run=True, verbose=True
dry_run=True,
dry_run=True,
verbose=True
) )
# Should have sent multiple batches # Should have sent multiple batches