From d3bc4bd5a20ac543ddc835c68001fb209533584c Mon Sep 17 00:00:00 2001
From: KKlochko <KKlochko@protonmail.com>
Date: Thu, 9 Nov 2023 22:06:17 +0200
Subject: [PATCH] Update the test for the creating source feature to add
 arguments.

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

diff --git a/features/creating_source.feature b/features/creating_source.feature
index 4383e4a..3ab2e7a 100644
--- a/features/creating_source.feature
+++ b/features/creating_source.feature
@@ -3,12 +3,13 @@ Feature: Creating the source
   Scenario Outline: Adding an new unique source
     Given the label "<label>"
     And the path "<source_path>"
+    And the arguments "<arguments>"
     When I add the source
     Then the source should be added successfully
 
     Examples:
-      | label | source_path |
-      | usb   | /mnt/usb    |
-      | db    | /db         |
-      | temp  | /tmp        |
+      | label | source_path | arguments      |
+      | usb   | /mnt/usb    | <empty>        |
+      | db    | /db         | -avuP          |
+      | temp  | /tmp        | -avuP --delete |
 
diff --git a/features/steps/creating_source_steps.py b/features/steps/creating_source_steps.py
index 955a514..730fe1e 100644
--- a/features/steps/creating_source_steps.py
+++ b/features/steps/creating_source_steps.py
@@ -12,6 +12,13 @@ def given_source_label(context, label):
 def given_source_path(context, source_path):
     context.source_path = source_path
 
+@given('the arguments "{arguments}"')
+def given_source_arguments(context, arguments):
+    context.args = arguments
+
+    if arguments == "<empty>":
+        context.args = ""
+
 @when('I add the source')
 def add_source(context):
     try:
@@ -19,7 +26,7 @@ def add_source(context):
             context.label,
             context.source_path,
             [],
-            "",
+            context.args,
         )
     except IntegrityError:
         context.exception_raised = True
@@ -32,4 +39,5 @@ def path_has_added(context):
 
     assert context.source.label  == context.label
     assert context.source.source.path == context.source_path
+    assert context.source.args.command == context.args