Fix: Indexing integration test is complicated and inefficient (#6676,…

… PR #6677)
DataBiosphere · Nov 7, 2024 · cfe7acf · cfe7acf
2 parents bee4a60 + 3998aba
commit cfe7acf
Show file tree

Hide file tree

Showing 3 changed files with 105 additions and 166 deletions.
diff --git a/src/azul/indexer/__init__.py b/src/azul/indexer/__init__.py
@@ -552,6 +552,9 @@ def spec_cls(cls) -> type[SourceSpec]:
         spec_cls, ref_cls = get_generic_type_params(cls, SourceSpec, SourceRef)
         return spec_cls
 
+    def with_prefix(self, prefix: Prefix) -> Self:
+        return attrs.evolve(self, spec=attrs.evolve(self.spec, prefix=prefix))
+
 
 class SourcedBundleFQIDJSON(BundleFQIDJSON):
     source: SourceJSON

diff --git a/src/azul/plugins/__init__.py b/src/azul/plugins/__init__.py
@@ -640,7 +640,7 @@ def partition_source(self,
                 prefix = Prefix.for_main_deployment(count)
             else:
                 prefix = Prefix.for_lesser_deployment(count)
-            source = attr.evolve(source, spec=attr.evolve(source.spec, prefix=prefix))
+            source = source.with_prefix(prefix)
         return source
 
     @abstractmethod