From c67dc14003e26c6428906b3bb7a3524983bfa0de Mon Sep 17 00:00:00 2001
From: KKlochko <KKlochko@protonmail.com>
Date: Fri, 10 Nov 2023 16:53:07 +0200
Subject: [PATCH] Update the test for the creating source feature to add
 destinations.

---
 features/creating_source.feature        |  9 +++++----
 features/steps/creating_source_steps.py | 17 ++++++++++++++++-
 2 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/features/creating_source.feature b/features/creating_source.feature
index 3ab2e7a..bd66b3c 100644
--- a/features/creating_source.feature
+++ b/features/creating_source.feature
@@ -3,13 +3,14 @@ Feature: Creating the source
   Scenario Outline: Adding an new unique source
     Given the label "<label>"
     And the path "<source_path>"
+    And the destinations <destinations>
     And the arguments "<arguments>"
     When I add the source
     Then the source should be added successfully
 
     Examples:
-      | label | source_path | arguments      |
-      | usb   | /mnt/usb    | <empty>        |
-      | db    | /db         | -avuP          |
-      | temp  | /tmp        | -avuP --delete |
+      | label | source_path | destinations                     | arguments      |
+      | usb   | /mnt/usb    | []                               | <empty>        |
+      | db    | /db         | ["/backup/db"]                   | -avuP          |
+      | temp  | /tmp        | ["/backup/tmp1", "/backup/tmp2"] | -avuP --delete |
 
diff --git a/features/steps/creating_source_steps.py b/features/steps/creating_source_steps.py
index 730fe1e..6fea459 100644
--- a/features/steps/creating_source_steps.py
+++ b/features/steps/creating_source_steps.py
@@ -2,6 +2,7 @@ from behave import given, when, then
 from peewee import IntegrityError
 from tui_rsync.models.models import create_tables
 from tui_rsync.models.models import Source
+import json
 
 @given('the label "{label}"')
 def given_source_label(context, label):
@@ -12,6 +13,10 @@ def given_source_label(context, label):
 def given_source_path(context, source_path):
     context.source_path = source_path
 
+@given('the destinations {destinations_json}')
+def given_source_destinations(context, destinations_json):
+    context.destinations = json.loads(destinations_json)
+
 @given('the arguments "{arguments}"')
 def given_source_arguments(context, arguments):
     context.args = arguments
@@ -25,19 +30,29 @@ def add_source(context):
         context.source = Source.create_save(
             context.label,
             context.source_path,
-            [],
+            context.destinations,
             context.args,
         )
+
+        context.source = Source.get_source(context.label)
     except IntegrityError:
         context.exception_raised = True
     else:
         context.exception_raised = False
 
+def compare_destinations(actual:list, expected: list[str]) -> bool:
+    actual_path_set = {destionation.path for destionation in actual}
+    return actual_path_set == set(expected)
+
 @then('the source should be added successfully')
 def path_has_added(context):
     assert context.exception_raised == False
 
     assert context.source.label  == context.label
     assert context.source.source.path == context.source_path
+    assert compare_destinations(
+        context.source.get_destinations(),
+        context.destinations
+    )
     assert context.source.args.command == context.args