From a3f47a6418665a363d3fd765992c872b496b25c1 Mon Sep 17 00:00:00 2001
From: Dan Helfman <witten@torsion.org>
Date: Mon, 10 Jul 2023 09:38:28 -0700
Subject: [PATCH] Remove some sections from tests (#721).

---
 tests/end-to-end/test_database.py         |  6 +++---
 tests/integration/config/test_generate.py | 24 +++++++++++------------
 2 files changed, 14 insertions(+), 16 deletions(-)

diff --git a/tests/end-to-end/test_database.py b/tests/end-to-end/test_database.py
index 9209ec3..f9c3621 100644
--- a/tests/end-to-end/test_database.py
+++ b/tests/end-to-end/test_database.py
@@ -25,7 +25,7 @@ def write_configuration(
 source_directories:
     - {source_directory}
 repositories:
-    - {repository_path}
+    - path: {repository_path}
 borgmatic_source_directory: {borgmatic_source_directory}
 
 encryption_passphrase: "test"
@@ -96,7 +96,7 @@ def write_custom_restore_configuration(
 source_directories:
     - {source_directory}
 repositories:
-    - {repository_path}
+    - path: {repository_path}
 borgmatic_source_directory: {borgmatic_source_directory}
 
 encryption_passphrase: "test"
@@ -158,7 +158,7 @@ def write_simple_custom_restore_configuration(
 source_directories:
     - {source_directory}
 repositories:
-    - {repository_path}
+    - path: {repository_path}
 borgmatic_source_directory: {borgmatic_source_directory}
 
 encryption_passphrase: "test"
diff --git a/tests/integration/config/test_generate.py b/tests/integration/config/test_generate.py
index 5df1825..e332b89 100644
--- a/tests/integration/config/test_generate.py
+++ b/tests/integration/config/test_generate.py
@@ -58,14 +58,13 @@ foo:
         - baz
         - quux
 
-location:
-    repositories:
-        - one
-        - two
+repositories:
+    - one
+    - two
 
-    # This comment should be kept.
-    # COMMENT_OUT
-    other: thing
+# This comment should be kept.
+# COMMENT_OUT
+other: thing
     '''
 
     # flake8: noqa
@@ -75,13 +74,12 @@ location:
 #         - baz
 #         - quux
 
-location:
-    repositories:
-        - one
-        - two
+repositories:
+    - one
+    - two
 
-    # This comment should be kept.
-#     other: thing
+# This comment should be kept.
+# other: thing
     '''
 
     assert module.comment_out_optional_configuration(config.strip()) == expected_config.strip()