class GmpCreateAlertTestCase(unittest.TestCase): def setUp(self): self.connection = MockConnection() self.gmp = Gmp(self.connection) def test_missing_name(self): with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='', condition='foo', event='bar', method='lorem') with self.assertRaises(RequiredArgument): self.gmp.create_alert(name=None, condition='foo', event='bar', method='lorem') def test_missing_condition(self): with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition='', event='bar', method='lorem') with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition=None, event='bar', method='lorem') def test_missing_event(self): with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition='bar', event='', method='lorem') with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition='bar', event=None, method='lorem') def test_missing_method(self): with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition='bar', event='lorem', method='') with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition='bar', event='lorem', method=None) def test_invalid_condition(self): with self.assertRaises(InvalidArgument): self.gmp.create_alert(name='foo', condition='bar', event='Task run status changed', method='Email') def test_invalid_event(self): with self.assertRaises(InvalidArgument): self.gmp.create_alert(name='foo', condition='Always', event='lorem', method='Email') def test_invalid_method(self): with self.assertRaises(InvalidArgument): self.gmp.create_alert(name='foo', condition='Always', event='Task run status changed', method='ipsum') def test_invalid_condition_for_secinfo(self): with self.assertRaises(InvalidArgument): self.gmp.create_alert(name='foo', condition='Severity at least', event='Updated SecInfo arrived', method='Email') def test_invalid_method_for_secinfo(self): with self.assertRaises(InvalidArgument): self.gmp.create_alert(name='foo', condition='Always', event='Updated SecInfo arrived', method='HTTP Get') def test_create_alert(self): self.gmp.create_alert(name='foo', condition='Always', event='Task run status changed', method='Email') self.connection.send.has_been_called_with( '<create_alert>' '<name>foo</name>' '<condition>Always</condition>' '<event>Task run status changed</event>' '<method>Email</method>' '</create_alert>') def test_create_alert_with_filter_id(self): self.gmp.create_alert(name='foo', condition='Always', event='Task run status changed', method='Email', filter_id='f1') self.connection.send.has_been_called_with( '<create_alert>' '<name>foo</name>' '<condition>Always</condition>' '<event>Task run status changed</event>' '<method>Email</method>' '<filter id="f1"/>' '</create_alert>') def test_create_alert_with_comment(self): self.gmp.create_alert(name='foo', condition='Always', event='Task run status changed', method='Email', comment='hello') self.connection.send.has_been_called_with( '<create_alert>' '<name>foo</name>' '<condition>Always</condition>' '<event>Task run status changed</event>' '<method>Email</method>' '<comment>hello</comment>' '</create_alert>') def test_create_alert_with_condition_data(self): self.gmp.create_alert( name='foo', condition='Always', event='Task run status changed', method='Email', condition_data={'foo': 'bar'}, ) self.connection.send.has_been_called_with( '<create_alert>' '<name>foo</name>' '<condition>Always<data>bar<name>foo</name></data></condition>' '<event>Task run status changed</event>' '<method>Email</method>' '</create_alert>') def test_create_alert_with_event_data(self): self.gmp.create_alert( name='foo', condition='Always', event='Task run status changed', method='Email', event_data={'foo': 'bar'}, ) self.connection.send.has_been_called_with( '<create_alert>' '<name>foo</name>' '<condition>Always</condition>' '<event>Task run status changed' '<data>bar<name>foo</name></data>' '</event>' '<method>Email</method>' '</create_alert>') def test_create_alert_with_method_data(self): self.gmp.create_alert( name='foo', condition='Always', event='Task run status changed', method='Email', method_data={'foo': 'bar'}, ) self.connection.send.has_been_called_with( '<create_alert>' '<name>foo</name>' '<condition>Always</condition>' '<event>Task run status changed</event>' '<method>Email<data>bar<name>foo</name></data></method>' '</create_alert>')
class GmpCreateAlertTestCase(unittest.TestCase): def setUp(self): self.connection = MockConnection() self.gmp = Gmp(self.connection) def test_missing_name(self): with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='', condition='foo', event='bar', method='lorem') with self.assertRaises(RequiredArgument): self.gmp.create_alert(name=None, condition='foo', event='bar', method='lorem') def test_missing_condition(self): with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition='', event='bar', method='lorem') with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition=None, event='bar', method='lorem') def test_missing_event(self): with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition='bar', event='', method='lorem') with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition='bar', event=None, method='lorem') def test_missing_method(self): with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition='bar', event='lorem', method='') with self.assertRaises(RequiredArgument): self.gmp.create_alert(name='foo', condition='bar', event='lorem', method=None) def test_create_alert(self): self.gmp.create_alert(name='foo', condition='bar', event='lorem', method='ipsum') self.connection.send.has_been_called_with('<create_alert>' '<name>foo</name>' '<condition>bar</condition>' '<event>lorem</event>' '<method>ipsum</method>' '</create_alert>') def test_create_alert_with_filter_id(self): self.gmp.create_alert(name='foo', condition='bar', event='lorem', method='ipsum', filter_id='f1') self.connection.send.has_been_called_with('<create_alert>' '<name>foo</name>' '<condition>bar</condition>' '<event>lorem</event>' '<method>ipsum</method>' '<filter id="f1"/>' '</create_alert>') def test_create_alert_with_comment(self): self.gmp.create_alert(name='foo', condition='bar', event='lorem', method='ipsum', comment='hello') self.connection.send.has_been_called_with('<create_alert>' '<name>foo</name>' '<condition>bar</condition>' '<event>lorem</event>' '<method>ipsum</method>' '<comment>hello</comment>' '</create_alert>') def test_create_alert_with_condition_data(self): self.gmp.create_alert(name='foo', condition='bar', event='lorem', method='ipsum', condition_data={'foo': 'bar'}) self.connection.send.has_been_called_with( '<create_alert>' '<name>foo</name>' '<condition>bar<data>bar<name>foo</name></data></condition>' '<event>lorem</event>' '<method>ipsum</method>' '</create_alert>') def test_create_alert_with_event_data(self): self.gmp.create_alert(name='foo', condition='bar', event='lorem', method='ipsum', event_data={'foo': 'bar'}) self.connection.send.has_been_called_with( '<create_alert>' '<name>foo</name>' '<condition>bar</condition>' '<event>lorem<data>bar<name>foo</name></data></event>' '<method>ipsum</method>' '</create_alert>') def test_create_alert_with_method_data(self): self.gmp.create_alert(name='foo', condition='bar', event='lorem', method='ipsum', method_data={'foo': 'bar'}) self.connection.send.has_been_called_with( '<create_alert>' '<name>foo</name>' '<condition>bar</condition>' '<event>lorem</event>' '<method>ipsum<data>bar<name>foo</name></data></method>' '</create_alert>')