refactor: use reinitializable in spacev2

snapshot-labs · Aug 18, 2023 · 18b4b76 · 18b4b76
1 parent b2eea03
commit 18b4b76
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/starknet/src/tests/mocks/space_v2.cairo b/starknet/src/tests/mocks/space_v2.cairo
@@ -7,17 +7,20 @@ trait ISpaceV2<TContractState> {
 #[starknet::contract]
 mod SpaceV2 {
     use super::ISpaceV2;
+    use sx::utils::reinitializable::Reinitializable;
+    use sx::utils::ReinitializableImpl;
     #[storage]
     struct Storage {
-        _initialized: bool,
         _var: felt252
     }
 
     #[external(v0)]
     impl SpaceV2 of ISpaceV2<ContractState> {
         fn initialize(ref self: ContractState, var: felt252) {
-            assert(self._initialized.read() == false, 'Contract already initialized');
-            self._initialized.write(true);
+            // TODO: Temp component syntax
+            let mut state: Reinitializable::ContractState =
+                Reinitializable::unsafe_new_contract_state();
+            ReinitializableImpl::initialize(ref state);
             self._var.write(var);
         }
         fn get_var(self: @ContractState) -> felt252 {