37 lines
1.2 KiB
Python
Executable File
37 lines
1.2 KiB
Python
Executable File
import unittest
|
|
import process_factory
|
|
|
|
class Mock_Process_Prometheus() :
|
|
def __init__(self, *args, **kwargs) :
|
|
pass
|
|
|
|
class Mock_Process_Kairos() :
|
|
def __init__(self, *args, **kwargs) :
|
|
pass
|
|
|
|
class Mock_Alert_Config() :
|
|
def __init__(self, type) :
|
|
self.t = type
|
|
|
|
def type(self) :
|
|
return self.t
|
|
|
|
class Test_Process_Factory(unittest.TestCase):
|
|
def setUp(self) :
|
|
self.was_prom = process_factory.process_prometheus.Process_Prometheus
|
|
self.was_kai = process_factory.process_kairos.Process_Kairos
|
|
process_factory.process_prometheus.Process_Prometheus = Mock_Process_Prometheus
|
|
process_factory.process_kairos.Process_Kairos = Mock_Process_Kairos
|
|
|
|
def tearDown(self) :
|
|
process_factory.process_prometheus.Process_Prometheus = self.was_prom
|
|
process_factory.process_kairos.Process_Kairos = self.was_kai
|
|
|
|
def test(self) :
|
|
factory = process_factory.Process_Factory(None, None, None)
|
|
self.assertTrue(type(factory.build(Mock_Alert_Config("a"))) is Mock_Process_Kairos)
|
|
self.assertTrue(type(factory.build(Mock_Alert_Config("prometheus"))) is Mock_Process_Prometheus)
|
|
|
|
if __name__ == "__main__" :
|
|
unittest.main()
|