Fix some issues 12
This commit is contained in:
137
services/notification/test_email_sending.py
Normal file
137
services/notification/test_email_sending.py
Normal file
@@ -0,0 +1,137 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Test email sending functionality
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
from email.mime.text import MIMEText
|
||||
from email.mime.multipart import MIMEMultipart
|
||||
|
||||
def test_email_message_construction():
|
||||
"""Test that email messages are constructed correctly"""
|
||||
print("Testing Email Message Construction...")
|
||||
|
||||
# Create a test message like the email service does
|
||||
message = MIMEMultipart('alternative')
|
||||
message['Subject'] = 'Test Subject'
|
||||
message['To'] = 'test@example.com'
|
||||
message['From'] = 'noreply@bakeryforecast.es'
|
||||
|
||||
# Add text content
|
||||
text_part = MIMEText('This is a test email', 'plain', 'utf-8')
|
||||
message.attach(text_part)
|
||||
|
||||
# Verify message structure
|
||||
assert message['Subject'] == 'Test Subject'
|
||||
assert message['To'] == 'test@example.com'
|
||||
assert message['From'] == 'noreply@bakeryforecast.es'
|
||||
# Check that the message has the expected structure
|
||||
assert len(message.get_payload()) == 1
|
||||
assert message.get_payload()[0].get_content_type() == 'text/plain'
|
||||
|
||||
print("✓ Email message constructed correctly")
|
||||
print(f" Subject: {message['Subject']}")
|
||||
print(f" To: {message['To']}")
|
||||
print(f" From: {message['From']}")
|
||||
|
||||
return True
|
||||
|
||||
async def test_aiosmtplib_compatibility():
|
||||
"""Test aiosmtplib send_message compatibility"""
|
||||
print("\nTesting aiosmtplib Compatibility...")
|
||||
|
||||
try:
|
||||
import aiosmtplib
|
||||
|
||||
# Create a test message
|
||||
message = MIMEMultipart('alternative')
|
||||
message['Subject'] = 'Test Subject'
|
||||
message['To'] = 'test@example.com'
|
||||
message['From'] = 'noreply@bakeryforecast.es'
|
||||
text_part = MIMEText('Test content', 'plain', 'utf-8')
|
||||
message.attach(text_part)
|
||||
|
||||
# Test that send_message method exists and can be called
|
||||
# (We won't actually send, just verify the method signature)
|
||||
server = aiosmtplib.SMTP(hostname='localhost', port=1025)
|
||||
|
||||
# Check if send_message method accepts the message
|
||||
import inspect
|
||||
sig = inspect.signature(server.send_message)
|
||||
params = list(sig.parameters.keys())
|
||||
|
||||
print(f" send_message parameters: {params}")
|
||||
|
||||
# The message should be the first parameter
|
||||
assert 'message' in params or len(params) > 0
|
||||
|
||||
print("✓ aiosmtplib send_message method is compatible")
|
||||
|
||||
return True
|
||||
|
||||
except ImportError:
|
||||
print("⚠ aiosmtplib not available for testing")
|
||||
return True
|
||||
except Exception as e:
|
||||
print(f"✗ aiosmtplib test failed: {e}")
|
||||
return False
|
||||
|
||||
def test_configuration_compatibility():
|
||||
"""Test that the configuration is compatible with both modes"""
|
||||
print("\nTesting Configuration Compatibility...")
|
||||
|
||||
# Test trusted relay configuration
|
||||
config1 = {
|
||||
'SMTP_HOST': 'mailu.local',
|
||||
'SMTP_PORT': 587,
|
||||
'SMTP_USER': '',
|
||||
'SMTP_PASSWORD': '',
|
||||
'SMTP_REQUIRE_AUTH': False
|
||||
}
|
||||
|
||||
# Test authenticated configuration
|
||||
config2 = {
|
||||
'SMTP_HOST': 'smtp.sendgrid.net',
|
||||
'SMTP_PORT': 587,
|
||||
'SMTP_USER': 'apikey',
|
||||
'SMTP_PASSWORD': 'test_key',
|
||||
'SMTP_REQUIRE_AUTH': True
|
||||
}
|
||||
|
||||
# Verify both configurations are valid
|
||||
assert isinstance(config1['SMTP_REQUIRE_AUTH'], bool)
|
||||
assert isinstance(config2['SMTP_REQUIRE_AUTH'], bool)
|
||||
|
||||
print("✓ Both configuration modes are valid")
|
||||
print(f" Trusted relay: SMTP_REQUIRE_AUTH={config1['SMTP_REQUIRE_AUTH']}")
|
||||
print(f" Authenticated: SMTP_REQUIRE_AUTH={config2['SMTP_REQUIRE_AUTH']}")
|
||||
|
||||
return True
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("Email Sending Functionality Test")
|
||||
print("=" * 40)
|
||||
|
||||
try:
|
||||
# Run tests
|
||||
results = []
|
||||
results.append(test_email_message_construction())
|
||||
results.append(asyncio.run(test_aiosmtplib_compatibility()))
|
||||
results.append(test_configuration_compatibility())
|
||||
|
||||
print("\n" + "=" * 40)
|
||||
if all(results):
|
||||
print("✓ All email sending tests passed!")
|
||||
print("\nThe email service should now work correctly with:")
|
||||
print("- Proper MIME message construction")
|
||||
print("- Compatible aiosmtplib usage")
|
||||
print("- Both trusted relay and authenticated modes")
|
||||
else:
|
||||
print("✗ Some tests failed!")
|
||||
exit(1)
|
||||
|
||||
except Exception as e:
|
||||
print(f"\n✗ Test failed with exception: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
exit(1)
|
||||
Reference in New Issue
Block a user