def setUp(self, mock_load_config, mock_delete_all, mock_open,
mock_multidigraph):
mock_open.return_value = "config.yaml"
self.base_graph = dict(graph_db='graph_db', refresh=True)
self.statuses = dict(status=['Status1', 'Status2'],
motd='I am the MOTD')
def load_config(filename):
if "config.yaml" in filename:
return self.base_graph
elif "statuses.yaml" in filename:
return self.statuses
mock_load_config.side_effect = load_config
self.popittoneo = PopItToNeo()
self.graph_patcher = patch("popit_to_networkx.nx.MultiDiGraph")
self.add_node_patcher = patch(
"popit_to_networkx.nx.MultiDiGraph.PopItRelationship.add_node")
self.graph = self.graph_patcher.start()
self.add_node = self.add_node_patcher.start()
self.popit2networkx = PopItRelationship()
