From 7bb4b4e2512d4b937a35d2049036d5629739d143 Mon Sep 17 00:00:00 2001 From: Jesper Hodge Date: Mon, 24 Feb 2025 16:21:47 -0500 Subject: [PATCH] fix: lint --- .../commands/backfill_ecu_table_user_foreign_key.py | 6 ++---- .../test_backfill_ecu_table_user_foreign_key.py | 8 +++++--- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/enterprise/management/commands/backfill_ecu_table_user_foreign_key.py b/enterprise/management/commands/backfill_ecu_table_user_foreign_key.py index 007318d8f..cb7122bdb 100644 --- a/enterprise/management/commands/backfill_ecu_table_user_foreign_key.py +++ b/enterprise/management/commands/backfill_ecu_table_user_foreign_key.py @@ -8,11 +8,10 @@ from django.contrib import auth from django.core.management.base import BaseCommand -from django.db import transaction, DatabaseError +from django.db import DatabaseError, transaction from enterprise.models import EnterpriseCustomerUser - log = logging.getLogger(__name__) User = auth.get_user_model() @@ -104,8 +103,7 @@ def backfill_ecu_table_user_foreign_key(self, options): log.info(f"Backfill complete! Processed {total_processed}/{total_rows} records.") - - def handle(self, *args, **options): # pylint: disable=unused-argument + def handle(self, *_args, **options): """ Entry point for management command execution. """ diff --git a/tests/test_enterprise/management/test_backfill_ecu_table_user_foreign_key.py b/tests/test_enterprise/management/test_backfill_ecu_table_user_foreign_key.py index 155321716..74b6ee9f9 100644 --- a/tests/test_enterprise/management/test_backfill_ecu_table_user_foreign_key.py +++ b/tests/test_enterprise/management/test_backfill_ecu_table_user_foreign_key.py @@ -13,12 +13,14 @@ from django.db.models import signals from django.test import TestCase -from enterprise.models import EnterpriseCustomerUser, EnterpriseCustomer +from enterprise.models import EnterpriseCustomer, EnterpriseCustomerUser from test_utils import factories + EXCEPTION = "DUMMY_TRACE_BACK" User = auth.get_user_model() + @mark.django_db @ddt.ddt class CreateEnterpriseCourseEnrollmentCommandTests(TestCase): @@ -83,8 +85,8 @@ def test_skips_rows_that_already_have_user_fk(self): @patch('logging.Logger.warning') def test_retry_5_times_on_failure(self, mock_log): with patch( - ("enterprise.management.commands.backfill_ecu_table_user_foreign_key." - "EnterpriseCustomerUser.objects.bulk_update"), + ("enterprise.management.commands.backfill_ecu_table_user_foreign_key." + + "EnterpriseCustomerUser.objects.bulk_update"), side_effect=Exception(EXCEPTION) ): with self.assertRaises(Exception) as e: