async def i_track_analysis_module_type(self, amt: AnalysisModuleType): assert isinstance(amt, AnalysisModuleType) db_amt = AnalysisModuleTracking(name=amt.name, json_data=amt.to_json()) async with self.get_db() as db: await db.merge(db_amt) await db.commit()
def test_analysis_module_type_serialization(): amt = AnalysisModuleType( name="test", description="test", observable_types=["test1", "test2"], directives=["test1", "test2"], dependencies=["test1", "test2"], tags=["test1", "test2"], modes=["test1", "test2"], cache_ttl=60, extended_version={"test1": "test2"}, types=["test1", "test2"], ) assert amt == AnalysisModuleType.from_dict(amt.to_dict()) assert amt == AnalysisModuleType.from_json(amt.to_json())