Check use_imaginary_chunk_locations in test

test/e2e/ytsaurus_controller_test.go

@@ -287,6 +287,7 @@ var _ = Describe("Basic e2e test for Ytsaurus controller", Label("e2e"), func()
 					By("Waiting cluster update completes")
 					EventuallyYtsaurus(ctx, ytsaurus, upgradeTimeout).Should(HaveClusterStateRunning())
 					checkClusterBaseViability(ytClient)
+					checkChunkLocations(ytClient)
 
 					podsAfterFullUpdate := getComponentPods(ctx, namespace)
 					pods := getChangedPods(podsBeforeUpdate, podsAfterFullUpdate)
@@ -1022,8 +1023,6 @@ func checkClusterBaseViability(ytClient yt.Client) {
 	res := make([]string, 0)
 	Expect(ytClient.ListNode(ctx, ypath.Path("/"), &res, nil)).Should(Succeed())
 
-	checkChunkLocations(ytClient)
-
 	By("Check that tablet cell bundles are in `good` health")
 	Eventually(func() bool {
 		notGoodBundles, err := components.GetNotGoodTabletCellBundles(ctx, ytClient)