feat(ICRC_Ledger): FI-1604: Set index in existing SNS and ck ledgers

rs/ledger_suite/icrc1/ledger/BUILD.bazel

@@ -235,17 +235,36 @@ package(default_visibility = ["//visibility:public"])
     ]
 ]
 
-rust_test(
-    name = "ledger_canister_test",
-    crate = ":_wasm_ledger_canister",
-    data = [
-        ":ledger.did",
-    ],
-    env = {
-        "CARGO_MANIFEST_DIR": "rs/ledger_suite/icrc1/ledger",
-    },
-    deps = ["@crate_index//:candid_parser"],
-)
+[
+    rust_test(
+        name = "ledger_canister_test" + name_suffix,
+        crate = ":_wasm_ledger_canister" + name_suffix,
+        crate_features = features,
+        data = [
+            ":ledger.did",
+        ],
+        env = {
+            "CARGO_MANIFEST_DIR": "rs/ledger_suite/icrc1/ledger",
+        },
+        deps = ["@crate_index//:candid_parser"] + extra_deps,
+    )
+    for (name_suffix, features, extra_deps) in [
+        (
+            "",
+            [],
+            [
+                "//rs/ledger_suite/icrc1/tokens_u64",
+            ],
+        ),
+        (
+            "_u256",
+            ["u256-tokens"],
+            [
+                "//rs/ledger_suite/icrc1/tokens_u256",
+            ],
+        ),
+    ]
+]
 
 [
     rust_ic_test(

rs/ledger_suite/icrc1/ledger/src/lib.rs

@@ -784,6 +784,10 @@ impl Ledger {
     pub fn copy_token_pool(&mut self) {
         self.stable_balances.token_pool = self.balances.token_pool;
     }
+
+    pub fn set_index_principal(&mut self, index_principal: Principal) {
+        self.index_principal = Some(index_principal);
+    }
 }
 
 impl LedgerContext for Ledger {