Parse arguments to get address and port

Signed-off-by: Yohann D'ANELLO <ynerant@crans.org>
This commit is contained in:
2020-12-21 16:03:54 +01:00
parent 272d0c25d6
commit d994ac9d67
4 changed files with 51 additions and 3 deletions

View File

@@ -0,0 +1,25 @@
import sys
import unittest
from squinnondation.squinnondation import Squinnondation
class TestSquinnondation(unittest.TestCase):
def test_arguments(self):
"""
Parse arguments and check errors.
"""
squinnondation = Squinnondation()
# No parameter was given
self.assertRaises(SystemExit, squinnondation.parse_arguments)
# Wrong port
sys.argv = ["squinnondation", "localhost", "42"]
self.assertRaises(ValueError, squinnondation.parse_arguments)
# Esnure that the parsing is correct
sys.argv = ["squinnondation", "localhost", "4242"]
squinnondation.parse_arguments()
self.assertEqual(squinnondation.client_address, "localhost")
self.assertEqual(squinnondation.client_port, 4242)