@@ -517,14 +517,20 @@ def _validate_content_extension(self, model: DocumentReference):
517517 if not self ._has_valid_extensions (content .extension , i ):
518518 return
519519
520+ def _is_content_stability_extension (self , extension ):
521+ return "contentstability" in str (extension ).lower ()
522+
523+ def _is_retrieval_mechanism_extension (self , extension ):
524+ return "retrievalmechanism" in str (extension ).lower ()
525+
520526 def _has_valid_extensions (self , extensions , i ):
521527 content_stability_count = 0
522528 content_retrieval_count = 0
523529
524530 for extension in extensions :
525- if "contentstability" in str ( extension ). lower ( ):
531+ if self . _is_content_stability_extension ( extension ):
526532 content_stability_count += 1
527- elif "retrievalmechanism" in str ( extension ). lower ( ):
533+ elif self . _is_retrieval_mechanism_extension ( extension ):
528534 content_retrieval_count += 1
529535
530536 if content_stability_count != 1 :
@@ -545,14 +551,16 @@ def _has_valid_extensions(self, extensions, i):
545551 )
546552 return False
547553
554+ return self ._validate_content_extension_items (extensions , i )
555+
556+ def _validate_content_extension_items (self , extensions , i ):
548557 for j , extension in enumerate (extensions ):
549- if "contentstability" in str ( extension ). lower ( ):
558+ if self . _is_content_stability_extension ( extension ):
550559 if not self ._validate_content_stability_extension (extension , i , j ):
551560 return False
552- elif "retrievalmechanism" in str ( extension ). lower ( ):
561+ elif self . _is_retrieval_mechanism_extension ( extension ):
553562 if not self ._validate_retrieval_mechanism_extension (extension , i , j ):
554563 return False
555-
556564 return True
557565
558566 def _validate_content_stability_extension (self , extension , i , j ):
0 commit comments