def setUp(self): my_directory = os.path.dirname(os.path.abspath(__file__)) self.bridge = CarbonBlackThreatConnectBridge("threatconnect", os.path.join(my_directory, "mock_threatconnect.conf"), logfile='/tmp/threatconnect.log', pidfile='/tmp/threatconnect.pid', debug=True) self.bridge.validate_config() data_dir = os.path.abspath(os.path.join(my_directory, 'data')) self.mock_server = threatconnect_mock_server.get_mocked_server(data_dir) tc_thread = threading.Thread(target=self.mock_server.run, args=('127.0.0.1', 7982)) tc_thread.daemon = True tc_thread.start()
class ThreatConnectTest(unittest.TestCase): def setUp(self): my_directory = os.path.dirname(os.path.abspath(__file__)) self.bridge = CarbonBlackThreatConnectBridge("threatconnect", os.path.join(my_directory, "mock_threatconnect.conf"), logfile='/tmp/threatconnect.log', pidfile='/tmp/threatconnect.pid', debug=True) self.bridge.validate_config() data_dir = os.path.abspath(os.path.join(my_directory, 'data')) self.mock_server = threatconnect_mock_server.get_mocked_server(data_dir) tc_thread = threading.Thread(target=self.mock_server.run, args=('127.0.0.1', 7982)) tc_thread.daemon = True tc_thread.start() def testConnection(self): auth = self.bridge.bridge_auth tc = ThreatConnectFeedGenerator(auth["api_key"], auth['api_secret_key'], auth["url"], self.bridge.api_urns.items()) new_iocs = tc.get_threatconnect_iocs() print len(new_iocs)