Files
flowsint/flowsint-transforms/tests/insert_command.py

28 lines
755 B
Python

import sys
import os
import asyncio
if __name__ == "__main__":
sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
from flowsint_types.domain import Domain
from flowsint_types.ip import Ip
from flowsint_transforms.domains.resolve import ResolveScanner
async def main():
# Create test data
domains = [Domain(domain="adaltas.com")]
ips = [Ip(address='12.23.34.45'), Ip(address='56.67.78.89')]
# Test the scanner
scanner = ResolveScanner("sketch_123", "scan_123")
# Test the new KISS postprocess method
scanner.postprocess(ips[:1], domains) # Only use first IP to match domains length
print("Postprocess test completed successfully!")
if __name__ == "__main__":
asyncio.run(main())