diff options
Diffstat (limited to 'src/leap/base/tests/test_checks.py')
| -rw-r--r-- | src/leap/base/tests/test_checks.py | 30 | 
1 files changed, 30 insertions, 0 deletions
diff --git a/src/leap/base/tests/test_checks.py b/src/leap/base/tests/test_checks.py index a3b3ea91..30746991 100644 --- a/src/leap/base/tests/test_checks.py +++ b/src/leap/base/tests/test_checks.py @@ -31,6 +31,8 @@ class LeapNetworkCheckTest(BaseLeapTest):          self.assertTrue(hasattr(checker, "check_internet_connection"),                          "missing meth") +        self.assertTrue(hasattr(checker, "check_tunnel_default_interface"), +                        "missing meth")          self.assertTrue(hasattr(checker, "is_internet_up"),                          "missing meth")          self.assertTrue(hasattr(checker, "ping_gateway"), @@ -42,6 +44,7 @@ class LeapNetworkCheckTest(BaseLeapTest):          mc = Mock()          checker.run_all(checker=mc)          self.assertTrue(mc.check_internet_connection.called, "not called") +        self.assertTrue(mc.check_tunnel_default_interface.called, "not called")          self.assertTrue(mc.ping_gateway.called, "not called")          self.assertTrue(mc.is_internet_up.called, "not called") @@ -55,6 +58,33 @@ class LeapNetworkCheckTest(BaseLeapTest):                      "Mask\tMTU\tWindow\tIRTT")                  checker.get_default_interface_gateway() +    def test_check_tunnel_default_interface(self): +        checker = checks.LeapNetworkChecker() +        with patch('leap.base.checks.open', create=True) as mock_open: +            with self.assertRaises(exceptions.TunnelNotDefaultRouteError): +                mock_open.return_value = StringIO( +                    "Iface\tDestination Gateway\t" +                    "Flags\tRefCntd\tUse\tMetric\t" +                    "Mask\tMTU\tWindow\tIRTT") +                checker.check_tunnel_default_interface() + +        with patch('leap.base.checks.open', create=True) as mock_open: +            with self.assertRaises(exceptions.TunnelNotDefaultRouteError): +                mock_open.return_value = StringIO( +                    "Iface\tDestination Gateway\t" +                    "Flags\tRefCntd\tUse\tMetric\t" +                    "Mask\tMTU\tWindow\tIRTT\n" +                    "wlan0\t00000000\t0102A8C0\t0003\t0\t0\t0\t00000000\t0\t0\t0") +                checker.check_tunnel_default_interface() + +        with patch('leap.base.checks.open', create=True) as mock_open: +            mock_open.return_value = StringIO( +                "Iface\tDestination Gateway\t" +                "Flags\tRefCntd\tUse\tMetric\t" +                "Mask\tMTU\tWindow\tIRTT\n" +                "tun0\t00000000\t01002A0A\t0003\t0\t0\t0\t00000080\t0\t0\t0") +            checker.check_tunnel_default_interface() +      def test_ping_gateway_fail(self):          checker = checks.LeapNetworkChecker()          with patch.object(ping, "quiet_ping") as mocked_ping:  | 
