Parse arguments to get address and port
Signed-off-by: Yohann D'ANELLO <ynerant@crans.org>
This commit is contained in:
25
squinnondation/test/squinnondation_test.py
Normal file
25
squinnondation/test/squinnondation_test.py
Normal 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)
|
Reference in New Issue
Block a user