updated tests.py again

This commit is contained in:
Raymond Jessop 2025-04-18 18:36:29 -05:00
parent 99667e7e42
commit a6016d7593
1 changed files with 11 additions and 7 deletions

View File

@ -27,15 +27,19 @@ User = get_user_model()
class DjangoAwsSesTests(TestCase): class DjangoAwsSesTests(TestCase):
def setUp(self): def setUp(self):
self.factory = RequestFactory() self.factory = RequestFactory()
# Create a Site instance # Use get_or_create to avoid duplicate Site creation
self.site = Site.objects.create(id=1, domain='example.com', name='example.com') self.site, _ = Site.objects.get_or_create(
id=1, defaults={'domain': 'example.com', 'name': 'example.com'}
)
# Create AwsSesSettings for the Site # Create AwsSesSettings for the Site
AwsSesSettings.objects.create( AwsSesSettings.objects.get_or_create(
site=self.site, site=self.site,
access_key='test-key', defaults={
secret_key='test-secret', 'access_key': 'test-key',
region_name='us-east-1', 'secret_key': 'test-secret',
region_endpoint='email.us-east-1.amazonaws.com' 'region_name': 'us-east-1',
'region_endpoint': 'email.us-east-1.amazonaws.com'
}
) )
# Create test user # Create test user
self.user = User.objects.create_user( self.user = User.objects.create_user(