#!/usr/bin/python3

import asyncio
from python.BIRD.Test import Test, BIRDInstance

class ThisTest(Test):
    async def run(self):
        # Prepare machines and links
        self.src, self.dest = await self.machines(
                "src", "dest",
                t=BIRDInstance,
                )
        self.links = {
                "L": await self.link("L", "src", "dest")
                }

        # Start machines and links
        await self.start()

        # Startup check
        await self.route_dump(10, "startup")

        wtb = ("p170", "p180", "p190", "p200")
        btw = reversed(wtb)

        # Enable worst to best
        for p in wtb:
            await self.src.enable(p)
            await self.route_dump(1, f"enable-{p}")

        # Disable worst to best
        for p in wtb:
            await self.src.disable(p)
            await self.route_dump(1, f"disable-{p}")

        # Enable best to worst
        for p in btw:
            await self.src.enable(p)
            await self.route_dump(1, f"enable-{p}")

        # Disable worst to best
        for p in btw:
            await self.src.disable(p)
            await self.route_dump(1, f"disable-{p}")

        # Re-enable all at once
        await asyncio.gather(*[ self.src.enable(p) for p in wtb ])
        await self.route_dump(5, f"add-all")

        await self.cleanup()