From 99864c3f95deb2193a9f6c1fcb4e49a2c3c58d7b Mon Sep 17 00:00:00 2001
From: Bryan Brancotte <bryan.brancotte@pasteur.fr>
Date: Mon, 11 Mar 2024 15:54:39 +0100
Subject: [PATCH] Ability to have empty file for in test_channel

---
 tests/test_channel.py | 26 +++++++++++++++++---------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/tests/test_channel.py b/tests/test_channel.py
index b5c5547..b823d57 100644
--- a/tests/test_channel.py
+++ b/tests/test_channel.py
@@ -3,26 +3,34 @@ from src.channel import *
 from src.nextflow_file import Nextflow_File
 
 
+class EmptyNextflowFile(Nextflow_File):
+    def __init__(self, address="tests/ressources/channel/empty_wf.nf", display_info=False, *args, **kwargs):
+        super().__init__(address=address, display_info=display_info, *args, **kwargs)
+
+    def check_file_correctness_after_DSL(self):
+        return
+
+
 class TestChannel(unittest.TestCase):
 
     def test_get_code(self):
-        wf1 = Nextflow_File("tests/ressources/channel/empty_wf.nf", display_info=False)
+        wf1 = EmptyNextflowFile()
         ch1 = Channel(name = "ch1", origin = wf1)
         self.assertIsInstance(ch1, Channel)
         self.assertEqual(ch1.get_code(), "ch1")
 
     def test_get_name(self):
-        wf1 = Nextflow_File("tests/ressources/channel/empty_wf.nf", display_info=False)
+        wf1 = EmptyNextflowFile()
         ch1 = Channel(name = "ch1", origin = wf1)
         self.assertEqual(ch1.get_name(), "ch1")
 
     def test_get_type(self):
-        wf1 = Nextflow_File("tests/ressources/channel/empty_wf.nf", display_info=False)
+        wf1 = EmptyNextflowFile()
         ch1 = Channel(name = "ch1", origin = wf1)
         self.assertEqual(ch1.get_type(), "Channel")
 
     def test_add_source(self):
-        wf1 = Nextflow_File("tests/ressources/channel/empty_wf.nf", display_info=False)
+        wf1 = EmptyNextflowFile()
         ch1 = Channel(name = "ch1", origin = wf1)
         self.assertEqual(ch1.get_source(), [])
         ele = "This is a test"
@@ -30,7 +38,7 @@ class TestChannel(unittest.TestCase):
         self.assertEqual(ch1.get_source(), [ele])
 
     def test_add_sink(self):
-        wf1 = Nextflow_File("tests/ressources/channel/empty_wf.nf", display_info=False)
+        wf1 = EmptyNextflowFile()
         ch1 = Channel(name = "ch1", origin = wf1)
         self.assertEqual(ch1.get_sink(), [])
         ele = "This is a test"
@@ -38,7 +46,7 @@ class TestChannel(unittest.TestCase):
         self.assertEqual(ch1.get_sink(), [ele])
 
     def test_set_sink_null(self):
-        wf1 = Nextflow_File("tests/ressources/channel/empty_wf.nf", display_info=False)
+        wf1 = EmptyNextflowFile()
         ch1 = Channel(name = "ch1", origin = wf1)
         ele = "This is a test"
         ch1.add_sink(ele)
@@ -47,7 +55,7 @@ class TestChannel(unittest.TestCase):
         self.assertEqual(ch1.get_sink(), [])
 
     def test_remove_element_from_sink(self):
-        wf1 = Nextflow_File("tests/ressources/channel/empty_wf.nf", display_info=False)
+        wf1 = EmptyNextflowFile()
         ch1 = Channel(name = "ch1", origin = wf1)
         ele = "This is a test"
         ch1.add_sink(ele)
@@ -56,7 +64,7 @@ class TestChannel(unittest.TestCase):
         self.assertEqual(ch1.get_sink(), [])
 
     def test_equal(self):
-        wf1 = Nextflow_File("tests/ressources/channel/empty_wf.nf", display_info=False)
+        wf1 = EmptyNextflowFile()
         ch1 = Channel(name = "ch1", origin = wf1)
         ch1_1 = Channel(name = "ch1", origin = wf1)
         ch2 = Channel(name = "ch2", origin = wf1)
@@ -64,7 +72,7 @@ class TestChannel(unittest.TestCase):
         self.assertFalse(ch1.equal(channel=ch2))
 
     def test_get_structure(self):
-        wf1 = Nextflow_File("tests/ressources/channel/empty_wf.nf", display_info=False)
+        wf1 = EmptyNextflowFile()
         ch1 = Channel(name = "ch1", origin = wf1)
         dico = {}
         dico['nodes'] = []
-- 
GitLab