← Back to branch summary

~sseman/juju-chaos-monkey/add-shutdown

« back to all changes in this revision

Viewing changes to tests/test_net.py

  • Committer: seman.said at canonical
  • Date: 2015-05-06 05:55:47 UTC
  • Revision ID: seman.said@canonical.com-20150506055547-7dcwvx9pdw10o8g7
Added network block to Juju state server, API server and syslog

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/test_net.py
2
2
 
3
3
from mock import patch, call
4
4
 
 
5
from chaos_monkey_base import Chaos
5
6
from chaos.net import Net
6
7
 
7
8
 
8
9
class TestNet(TestCase):
9
10
 
10
 
    def _run_test(self, method_call, assert_args):
 
11
    def _run_test(self, method_call, assert_args, **kwargs):
11
12
        net = Net()
12
13
        with patch('utility.check_output', autospec=True) as mock:
13
 
            getattr(net, method_call)()
 
14
            getattr(net, method_call)(**kwargs)
14
15
        mock.assert_called_once_with(assert_args)
15
16
 
16
17
    def _run_mock_calls(self, method_call, call_list):
68
69
 
69
70
    def test_deny_port(self):
70
71
        self._run_test(
71
 
            'deny_port', ['ufw', 'deny', '8080'])
 
72
            'deny_port', ['ufw', 'deny', '8080'], port=8080)
 
73
 
 
74
    def test_allow_port(self):
 
75
        self._run_test(
 
76
            'allow_port', ['ufw', 'delete', 'deny', '8080'], port=8080)
 
77
 
 
78
    def test_deny_state_server(self):
 
79
        self._run_test('deny_state_server', ['ufw', 'deny', '37017'])
 
80
 
 
81
    def test_allow_state_server(self):
 
82
        self._run_test(
 
83
            'allow_state_server', ['ufw', 'delete', 'deny', '37017'])
 
84
 
 
85
    def test_deny_api_server(self):
 
86
        self._run_test('deny_api_server', ['ufw', 'deny', '17070'])
 
87
 
 
88
    def test_allow_api_server(self):
 
89
        self._run_test('allow_api_server', ['ufw', 'delete', 'deny', '17070'])
 
90
 
 
91
    def test_deny_sys_log(self):
 
92
        self._run_test('deny_sys_log', ['ufw', 'deny', '6514'])
 
93
 
 
94
    def test_allow_sys_log(self):
 
95
        self._run_test('allow_sys_log', ['ufw', 'delete', 'deny', '6514'])
72
96
 
73
97
    def test_get_chaos(self):
74
98
        net = Net()
75
99
        chaos = net.get_chaos()
76
 
        self.assertEqual(len(chaos),  5)
 
100
        self.assertEqual(len(chaos),  7)
77
101
        self.assertItemsEqual(
78
102
            self._get_all_command_str(chaos), self._command_strings())
79
103
        for c in chaos:
80
104
            self.assertEqual('net', c.group)
81
105
 
 
106
    def test_add_chaos(self):
 
107
        net = Net()
 
108
        chaos = net._add_chaos('enable', 'disable', 'command')
 
109
        self.assertIs(type(chaos), Chaos)
 
110
        self.assertEqual(chaos.enable, 'enable')
 
111
        self.assertEqual(chaos.disable, 'disable')
 
112
        self.assertEqual(chaos.group, 'net')
 
113
        self.assertEqual(chaos.command_str, 'command')
 
114
 
82
115
    def _get_all_command_str(self, chaos):
83
116
        return [c.command_str for c in chaos]
84
117
 
85
118
    def _command_strings(self):
86
119
        return ['deny-all', 'deny-incoming', 'deny-outgoing', 'allow-ssh',
87
 
                'deny-ssh']
 
120
                'deny-state-server', 'deny-api-server', 'deny-sys-log']
88
121
 
89
122
    def test_shutdown(self):
90
123
        self._run_test('reset', ['ufw', 'reset'])