34 lines
817 B
Python
Executable File
34 lines
817 B
Python
Executable File
import unittest
|
|
import alert_factory
|
|
|
|
class Mock_Alert() :
|
|
def __init__(self, *args) :
|
|
self.args = args
|
|
|
|
class Mock_Logger() :
|
|
def __init__(self) :
|
|
self.info = self.log
|
|
self.warn = self.log
|
|
self.warning = self.log
|
|
self.error = self.log
|
|
self.debug = self.log
|
|
|
|
def log(self, *args, **kwargs) :
|
|
print(args, kwargs)
|
|
|
|
class Test_Alert_Factory(unittest.TestCase) :
|
|
def setUp(self) :
|
|
self.was = alert_factory.Alert
|
|
alert_factory.Alert = Mock_Alert
|
|
|
|
def tearDown(self) :
|
|
alert_factory.Alert = self.was
|
|
|
|
def test(self) :
|
|
af = alert_factory.Alert_Factory(None, Mock_Logger())
|
|
alert = af.build(0, 5, None, "tagA, tagB", False, "tagC, tagD")
|
|
self.assertTrue(type(alert) == Mock_Alert)
|
|
|
|
if __name__ == "__main__" :
|
|
unittest.main()
|