class MockMonitor(Monitor):
def __init__(self):
self.logger = logger
self.fa_rabbit_routing_key = 'FAUCET.Event'
self.faucet_event = None
self.controller = get_test_controller()
self.s = SDNConnect(self.controller)
self.s.controller['TYPE'] = 'None'
self.s.get_sdn_context()
self.s.controller['TYPE'] = 'faucet'
self.s.get_sdn_context()
self.job_queue = queue.Queue()
self.m_queue = queue.Queue()
endpoint = endpoint_factory('foo')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
endpoint.mirror()
endpoint.p_prev_states.append((endpoint.state, int(time.time())))
self.s.endpoints[endpoint.name] = endpoint
endpoint = endpoint_factory('foo2')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
endpoint.p_next_state = 'mirror'
endpoint.queue()
endpoint.p_prev_states.append((endpoint.state, int(time.time())))
self.s.endpoints[endpoint.name] = endpoint
endpoint = endpoint_factory('foo3')
endpoint.endpoint_data = {
'tenant': 'foo',
'mac': '00:00:00:00:00:00',
'segment': 'foo',
'port': '1'
}
self.s.endpoints[endpoint.name] = endpoint
self.s.store_endpoints()
self.s.get_stored_endpoints()
self.results = 0
def get_q_item(self, q, timeout=1):
if not self.results:
self.results += 1
return (True, ('foo', {'data': {}}))
return (False, None)
def bad_get_q_item(self, q, timeout=1):
return (False, ('bar', {'data': {}}))
def format_rabbit_message(self, item):
return ({'data': {}}, False)
class MockMonitor(Monitor):
def __init__(self):
self.logger = logger
self.fa_rabbit_routing_key = 'FAUCET.Event'
self.faucet_event = None
self.controller = Config().get_config()
self.s = SDNConnect()
self.s.controller['TYPE'] = 'bcf'
self.s.get_sdn_context()
self.s.controller['TYPE'] = 'faucet'
self.s.get_sdn_context()
if 'POSEIDON_TRAVIS' in os.environ:
self.s.r = redis.StrictRedis(host='localhost',
port=6379,
db=0,
decode_responses=True)
endpoint = Endpoint('foo')
endpoint.endpoint_data = {
'tenant': 'foo', 'mac': '00:00:00:00:00:00', 'segment': 'foo', 'port': '1'}
endpoint.mirror()
endpoint.p_prev_states.append(
(endpoint.state, int(time.time())))
self.s.endpoints.append(endpoint)
endpoint = Endpoint('foo2')
endpoint.endpoint_data = {
'tenant': 'foo', 'mac': '00:00:00:00:00:00', 'segment': 'foo', 'port': '1'}
endpoint.p_next_state = 'mirror'
endpoint.queue()
endpoint.p_prev_states.append(
(endpoint.state, int(time.time())))
self.s.endpoints.append(endpoint)
endpoint = Endpoint('foo3')
endpoint.endpoint_data = {
'tenant': 'foo', 'mac': '00:00:00:00:00:00', 'segment': 'foo', 'port': '1'}
self.s.endpoints.append(endpoint)
self.s.store_endpoints()
self.s.get_stored_endpoints()
def get_q_item(self):
return (True, ('foo', {}))
def bad_get_q_item(self):
return (False, ('bar', {}))
def format_rabbit_message(self, item):
return {}
