def test_ti_config_and_load(self):
test_config1 = Path(_TEST_DATA).joinpath("msticpyconfig-askql.yaml")
with custom_mp_config(test_config1):
ti_settings = get_provider_settings()
self.assertIsInstance(ti_settings, dict)
self.assertGreaterEqual(1, len(ti_settings))
# Try to load TIProviders - should throw a warning on
# missing provider class
as_byoti_prov = AzSTI(query_provider=qry_prov)
ti_lookup = TILookup(primary_providers=[as_byoti_prov])
# should have 2 succesfully loaded providers
self.assertGreaterEqual(1, len(ti_lookup.loaded_providers))
self.assertGreaterEqual(1, len(ti_lookup.provider_status))
def test_ti_config_and_load(self):
config_path = Path(_TEST_DATA).parent.joinpath(
"msticpyconfig-test.yaml")
with custom_mp_config(self.config_path):
with warnings.catch_warnings():
# We want to ignore warnings from missing config
warnings.simplefilter("ignore", category=UserWarning)
ti_settings = get_provider_settings()
self.assertIsInstance(ti_settings, dict)
self.assertGreaterEqual(len(ti_settings), 4)
# Try to load TIProviders - should throw a warning on
# missing provider class
with custom_mp_config(self.config_path):
with self.assertWarns(UserWarning):
ti_lookup = TILookup()
# should have 2 succesfully loaded providers
self.assertGreaterEqual(len(ti_lookup.loaded_providers), 3)
self.assertGreaterEqual(len(ti_lookup.provider_status), 3)
def load_ti_lookup():
test_config1 = Path(_TEST_DATA).joinpath("msticpyconfig-askql.yaml").resolve()
with custom_mp_config(test_config1):
as_byoti_prov = AzSTI(query_provider=qry_prov)
return TILookup(primary_providers=[as_byoti_prov])
def result_to_df(cls, ioc_lookup):
"""Redirect to original method."""
return TILookup.result_to_df(ioc_lookup)
def load_ti_lookup(self):
with custom_mp_config(self.config_path):
with warnings.catch_warnings():
# We want to ignore warnings from missing config
warnings.simplefilter("ignore", category=UserWarning)
return TILookup()