def test_update(self, mock_write, mock_del): """ Unit test to update data to audisp conf file. """ params = {"priority_boost": "34", "q_depth": "13"} name = "audit" mock_del.return_value = {} mock_write.return_value = {} auditdisp = AuditdispModel() auditdisp.update(name, params)
def test_get_auditdisp_conf(self, mock_conf): """ Unittest to get audit dispatcher conf """ exp_out = {'overflow_action': 'SYSLOG', 'priority_boost': '4', 'q_depth': '150', 'max_restarts': '10', 'name_format': 'HOSTNAME' } open_mock = mock.mock_open() with mock.patch('wok.plugins.ginger.model.utils.open', open_mock, create=True): auditdisp = AuditdispModel() mock_conf.return_value = exp_out out = auditdisp.get_auditdisp_conf() self.assertEqual(out, exp_out)