Skip to content
Snippets Groups Projects
Commit 1ec527b3 authored by Jan David Mol's avatar Jan David Mol
Browse files

L2SS-577: Mock sys.argv instead of setting it

parent 99e2a85b
No related branches found
No related tags found
1 merge request!249L2SS-577: Support multiple XST streams
...@@ -12,22 +12,19 @@ from tangostationcontrol.statistics_writer import statistics_writer ...@@ -12,22 +12,19 @@ from tangostationcontrol.statistics_writer import statistics_writer
import sys import sys
from os.path import dirname from os.path import dirname
from tempfile import TemporaryDirectory from tempfile import TemporaryDirectory
from unittest import mock
class TestStatisticsWriter(base.TestCase): class TestStatisticsWriter(base.TestCase):
def setUp(self):
# strip all command-line arguments to allow us to redefine them in the test
sys.argv = [sys.argv[0]]
def test_sst(self): def test_sst(self):
with TemporaryDirectory() as tmpdir: with TemporaryDirectory() as tmpdir:
sys.argv += ["--mode", "SST", "--file", dirname(__file__) + "/SDP_SST_statistics_packets.bin", "--output_dir", tmpdir] new_sys_argv = [sys.argv[0], "--mode", "SST", "--file", dirname(__file__) + "/SDP_SST_statistics_packets.bin", "--output_dir", tmpdir]
with mock.patch.object(statistics_writer.sys, 'argv', new_sys_argv):
with self.assertRaises(SystemExit): with self.assertRaises(SystemExit):
statistics_writer.main() statistics_writer.main()
def test_xst(self): def test_xst(self):
with TemporaryDirectory() as tmpdir: with TemporaryDirectory() as tmpdir:
sys.argv += ["--mode", "XST", "--file", dirname(__file__) + "/SDP_XST_statistics_packets.bin", "--output_dir", tmpdir] new_sys_argv = [sys.argv[0], "--mode", "XST", "--file", dirname(__file__) + "/SDP_XST_statistics_packets.bin", "--output_dir", tmpdir]
with mock.patch.object(statistics_writer.sys, 'argv', new_sys_argv):
with self.assertRaises(SystemExit): with self.assertRaises(SystemExit):
statistics_writer.main() statistics_writer.main()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment