fix: skip importing hardfork accounts for custom networks (#1128)

Author: jyeshe

lib/ae_mdw/db/hardfork_presets.ex

@@ -24,14 +24,19 @@ defmodule AeMdw.Db.HardforkPresets do
   end
 
   defp do_import_account_presets() do
-    State.commit(State.new(), [
-      hardfork_mutation(:genesis, &:aec_fork_block_settings.genesis_accounts/0),
-      hardfork_mutation(:minerva, &:aec_fork_block_settings.minerva_accounts/0),
-      hardfork_mutation(:fortuna, &:aec_fork_block_settings.fortuna_accounts/0),
-      hardfork_mutation(:lima, &:aec_fork_block_settings.lima_accounts/0),
-      lima_contracts_mutation(),
-      lima_extra_accounts_mutation()
-    ])
+    if :aec_governance.get_network_id() in ["ae_uat", "ae_mainnet"] do
+      State.commit(
+        State.new(),
+        [
+          hardfork_mutation(:genesis, &:aec_fork_block_settings.genesis_accounts/0),
+          hardfork_mutation(:minerva, &:aec_fork_block_settings.minerva_accounts/0),
+          hardfork_mutation(:fortuna, &:aec_fork_block_settings.fortuna_accounts/0),
+          hardfork_mutation(:lima, &:aec_fork_block_settings.lima_accounts/0),
+          lima_contracts_mutation(),
+          lima_extra_accounts_mutation()
+        ]
+      )
+    end
   end
 
   defp hardfork_mutation(hardfork, fork_settings_accounts_fn) do

mix.exs

@@ -92,6 +92,7 @@ defmodule AeMdw.MixProject do
           :aetx,
           :aetx_env,
           :aetx_sign,
+          :aeu_env,
           :aeu_info,
           :aeu_mtrees,
           :app_ctrl,

test/ae_mdw/db/hardfork_presets_test.exs

@@ -2,13 +2,20 @@ defmodule AeMdw.Db.HardforkPresetsTest do
   use ExUnit.Case, async: false
 
   alias AeMdw.Collection
+  alias AeMdw.Database
   alias AeMdw.Db.HardforkPresets
   alias AeMdw.Db.Model
   alias AeMdw.Db.State
 
+  import Mock
+
   describe "import_account_presets" do
-    test "saves genesis, minerva, fortuna and lima migrated accounts" do
-      HardforkPresets.import_account_presets()
+    test "on mainnet saves genesis, minerva, fortuna and lima migrated accounts" do
+      delete_previous_import()
+
+      with_mocks [{:aec_governance, [:passthrough], get_network_id: fn -> "ae_mainnet" end}] do
+        HardforkPresets.import_account_presets()
+      end
 
       assert 716 ==
                State.new()
@@ -49,5 +56,45 @@ defmodule AeMdw.Db.HardforkPresetsTest do
                |> Stream.take_while(&match?({"contracts_lima", _bi, _target, _txi}, &1))
                |> Enum.count()
     end
+
+    test "on testnet saves genesis, minerva, fortuna and lima migrated accounts" do
+      delete_previous_import()
+
+      with_mocks [{:aec_governance, [:passthrough], get_network_id: fn -> "ae_uat" end}] do
+        HardforkPresets.import_account_presets()
+      end
+
+      assert 1 ==
+               State.new()
+               |> Collection.stream(Model.KindIntTransferTx, {"accounts_genesis", nil, nil, nil})
+               |> Stream.take_while(&match?({"accounts_genesis", _bi, _target, _txi}, &1))
+               |> Enum.count()
+
+      assert 3 ==
+               State.new()
+               |> Collection.stream(Model.KindIntTransferTx, {"accounts_minerva", nil, nil, nil})
+               |> Stream.take_while(&match?({"accounts_minerva", _bi, _target, _txi}, &1))
+               |> Enum.count()
+    end
+
+    test "on custom network skips importing accounts" do
+      delete_previous_import()
+
+      with_mocks [{:aec_governance, [:passthrough], get_network_id: fn -> "ae_custom" end}] do
+        HardforkPresets.import_account_presets()
+      end
+
+      assert Database.count(Model.KindIntTransferTx) == 0
+    end
+  end
+
+  defp delete_previous_import do
+    State.new()
+    |> Collection.stream(
+      Model.KindIntTransferTx,
+      {"", nil, nil, nil}
+    )
+    |> Enum.to_list()
+    |> Enum.each(&State.delete(State.new(), Model.KindIntTransferTx, &1))
   end
 end