From 1072844bb29e282e411687af778e71d533366271 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=98yvind=20Eide?= <eide.oyvind87@gmail.com>
Date: Wed, 20 Sep 2023 15:05:17 +0200
Subject: [PATCH] Add test for migration from version 1

---
 src/ert/storage/migration/response_info.py    |  2 +-
 test-data/block_storage                       |  2 +-
 .../storage/migration/test_version_1.py       | 26 +++++++++++++++++++
 3 files changed, 28 insertions(+), 2 deletions(-)
 create mode 100644 tests/unit_tests/storage/migration/test_version_1.py

diff --git a/src/ert/storage/migration/response_info.py b/src/ert/storage/migration/response_info.py
index 327bf1c15a8..88f54c4bd4f 100644
--- a/src/ert/storage/migration/response_info.py
+++ b/src/ert/storage/migration/response_info.py
@@ -16,7 +16,7 @@ def migrate(path: Path) -> None:
         response_info = {}
         for response in ens_config.response_configuration:
             response_info[response.name] = response.to_dict()
-        with open(experiment / "response.json", "w", encoding="utf-8") as fout:
+        with open(experiment / "responses.json", "w", encoding="utf-8") as fout:
             fout.write(json.dumps(response_info))
     with open(path / "index.json", encoding="utf-8") as f:
         index_json = json.load(f)
diff --git a/test-data/block_storage b/test-data/block_storage
index aac08ccf968..84305c7fb68 160000
--- a/test-data/block_storage
+++ b/test-data/block_storage
@@ -1 +1 @@
-Subproject commit aac08ccf9687e507d38e3de6c386d3014b5e9865
+Subproject commit 84305c7fb683c8aa49d2e46723307721447658fa
diff --git a/tests/unit_tests/storage/migration/test_version_1.py b/tests/unit_tests/storage/migration/test_version_1.py
new file mode 100644
index 00000000000..71a79289fae
--- /dev/null
+++ b/tests/unit_tests/storage/migration/test_version_1.py
@@ -0,0 +1,26 @@
+import pytest
+import json
+
+from ert.config import ErtConfig
+from ert.storage.local_storage import local_storage_set_ert_config
+from ert.storage import open_storage
+
+
+@pytest.fixture(scope="module", autouse=True)
+def set_ert_config(block_storage_path):
+    ert_config = ErtConfig.from_file(
+        str(block_storage_path / "version-1/poly_example/poly.ert")
+    )
+    yield local_storage_set_ert_config(ert_config)
+    local_storage_set_ert_config(None)
+
+
+def test_migrate_gen_kw(setup_case, set_ert_config):
+    setup_case("block_storage/version-1/poly_example", "poly.ert")
+    with open_storage("storage", "w") as storage:
+        assert len(list(storage.experiments)) == 1
+        experiment = list(storage.experiments)[0]
+        param_info = json.loads(
+            (experiment._path / "parameter.json").read_text(encoding="utf-8")
+        )
+    assert "COEFFS" in param_info