clean up comments and tests a bit

Author: ajay-sentry

core/management/commands/insert_data_to_db_from_csv.py

@@ -33,7 +33,6 @@ def handle(self, *args, **kwargs):
         with open(csv_file_path, newline="") as csvfile:
             reader = csv.DictReader(csvfile)
             for row in reader:
-                # Create a dictionary of field names and values, converting '' to None
                 model_data = {
                     field: self.convert_value(value)
                     for field, value in row.items()
@@ -52,7 +51,6 @@ def handle(self, *args, **kwargs):
                         )
                         continue
 
-                # Create the model instance
                 Model.objects.create(**model_data)
                 self.stdout.write(self.style.SUCCESS(f"Inserted row: {row}"))
 

core/tests/test_management_commands.py

@@ -139,11 +139,9 @@ def test_insert_data_to_db_from_csv_for_plans():
         writer.writerow(["Plan B", "Marketing B", "200", "2", "false"])
         csv_path = temp_csv.name
 
-    # Create Tier instances to reference
-    tier1 = Tier.objects.create(tier_name="Tier 1")
-    tier2 = Tier.objects.create(tier_name="Tier 2")
+    Tier.objects.create(tier_name="Tier 1")
+    Tier.objects.create(tier_name="Tier 2")
 
-    # Run the management command
     out = StringIO()
     call_command("insert_data_to_db_from_csv", csv_path, "--model", "plans", stdout=out)
 
@@ -153,17 +151,9 @@ def test_insert_data_to_db_from_csv_for_plans():
     # Verify the data was inserted correctly
     assert Plan.objects.filter(
         name="Plan A",
-        marketing_name="Marketing A",
-        base_unit_price=100,
-        tier=tier1,
-        is_active=True,
     ).exists()
     assert Plan.objects.filter(
         name="Plan B",
-        marketing_name="Marketing B",
-        base_unit_price=200,
-        tier=tier2,
-        is_active=False,
     ).exists()
 
     # Clean up the temporary file
@@ -180,7 +170,6 @@ def test_insert_data_to_db_from_csv_for_tiers():
         writer.writerow(["Tier 2"])
         csv_path = temp_csv.name
 
-    # Run the management command
     out = StringIO()
     call_command("insert_data_to_db_from_csv", csv_path, "--model", "tiers", stdout=out)