From d4c7bb9cc4159e18ab935f8835866f6cd2babe7d Mon Sep 17 00:00:00 2001 From: La Min Ko Date: Thu, 22 Oct 2020 11:04:10 +0630 Subject: [PATCH] close #18 --- tests/test_madmac.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/test_madmac.py b/tests/test_madmac.py index 53e4031..f7d428a 100644 --- a/tests/test_madmac.py +++ b/tests/test_madmac.py @@ -97,3 +97,13 @@ def test_validate_MAC_TypeError(self): def test_validate_MAC_ValueError(self): self.assertFalse(madmac.validate_mac(None)) + + def test_handle_args(self): + fn = getattr(module, "handle_args") + self.assertIsNotNone(fn().parse_args(["-t", "1"])) + + def test_main(self): + handle_args = getattr(module, "handle_args") + main = getattr(module, "main") + args = vars(handle_args().parse_args(["-t", "1"])) + self.assertIs(str, type(main(args)))