try fix postgres choices test

bckohan · bckohan · commit 55d12512eac0 · 2025-04-01T14:07:47.000-07:00
diff --git a/tests/test_parser_completers.py b/tests/test_parser_completers.py
@@ -2517,6 +2517,14 @@ def test_choices_completer(self):
                             f"choices --{field.replace('_', '-')}s {test_str}"
                         )
                     )
+                    # postgres normalizes the IPs - revert that
+                    if field == "ip_choice":
+                        for idx in range(0, len(completions)):
+                            completions[idx] = (
+                                completions[idx][0].split("/")[0],
+                                completions[idx][1],
+                            )
+
                     str_qry = self.MODEL_CLASS.objects.annotate(
                         field_as_str=Cast(field, output_field=CharField())
                     )

Original file line number	Diff line number	Diff line change
`@@ -2517,6 +2517,14 @@ def test_choices_completer(self):`
`2517`	`2517`	`f"choices --{field.replace('_', '-')}s {test_str}"`
`2518`	`2518`	`)`
`2519`	`2519`	`)`
	`2520`	`+ # postgres normalizes the IPs - revert that`
	`2521`	`+ if field == "ip_choice":`
	`2522`	`+ for idx in range(0, len(completions)):`
	`2523`	`+ completions[idx] = (`
	`2524`	`+ completions[idx][0].split("/")[0],`
	`2525`	`+ completions[idx][1],`
	`2526`	`+ )`
	`2527`	`+`
`2520`	`2528`	`str_qry = self.MODEL_CLASS.objects.annotate(`
`2521`	`2529`	`field_as_str=Cast(field, output_field=CharField())`
`2522`	`2530`	`)`