From d50f617a85680d426e81a902969866aa258d12fa Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <pierre-antoine.rouby@inrae.fr>
Date: Mon, 26 Aug 2024 15:10:36 +0200
Subject: [PATCH] tests: Add AddFiles scenario tests.

---
 src/Model/test_Model.py | 49 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)

diff --git a/src/Model/test_Model.py b/src/Model/test_Model.py
index 5c6c118c..cc1dc589 100644
--- a/src/Model/test_Model.py
+++ b/src/Model/test_Model.py
@@ -107,6 +107,55 @@ class StudyScenarioTestCase(unittest.TestCase):
         self.assertEqual(study.description, "bar")
         self.assertEqual(study.status.scenario_id, nid)
 
+    def test_save_study_add_files(self):
+        study = Study.new("foo", "bar")
+        study.river.additional_files.new(0)
+
+        dir = tempfile.mkdtemp()
+        f = os.path.join(dir, "foo.pamhyr")
+        study.filename = f
+        study.save()
+        study.close()
+
+        study = Study.open(f)
+        self.assertNotEqual(study, None)
+        self.assertEqual(study.name, "foo")
+        self.assertEqual(study.description, "bar")
+        self.assertEqual(len(study.river.additional_files), 1)
+
+        new = study.new_scenario_from_current()
+        nid = new.id
+
+        study.river.additional_files.new(1)
+
+        study.save()
+        study.close()
+
+        study = Study.open(f)
+        self.assertEqual(len(study.river.additional_files), 2)
+
+    def test_save_study_add_files_reload(self):
+        study = Study.new("foo", "bar")
+        study.river.additional_files.new(0)
+
+        dir = tempfile.mkdtemp()
+        f = os.path.join(dir, "foo.pamhyr")
+        study.filename = f
+        study.save()
+
+        new = study.new_scenario_from_current()
+        nid = new.id
+
+        study.river.additional_files.new(1)
+        self.assertEqual(len(study.river.additional_files), 2)
+
+        study.save()
+        study.reload_from_scenario(study.scenarios[0])
+        self.assertEqual(len(study.river.additional_files), 1)
+
+        study.reload_from_scenario(study.scenarios[nid])
+        self.assertEqual(len(study.river.additional_files), 2)
+
 
 class RiverTestCase(unittest.TestCase):
     def test_create_river(self):