#!/usr/bin/env python2

from bithordetest import message, BithordeD, TestConnection
from random import randint

ASSET_IDS = [message.Identifier(type=message.TREE_TIGER, id='GIS3CRGMSBT7CKRBLQFXFAL3K4YIO5P5E3AMC2A')]
INITIAL_REQUEST = randint(0,(2**64)-1)

if __name__ == '__main__':
    bithorded = BithordeD(config={
        'friend.upstream1.addr': '',
        'friend.upstream2.addr': '',
    })
    upstream1 = TestConnection(bithorded, name='upstream1')
    upstream2 = TestConnection(bithorded, name='upstream2')
    downstream1 = TestConnection(bithorded, name='downstream1')
    downstream2 = TestConnection(bithorded, name='downstream2')

    # Request an asset-session
    downstream1.send(message.BindRead(handle=1, ids=ASSET_IDS, timeout=500, requesters=[INITIAL_REQUEST]))
    req1 = upstream1.expect(message.BindRead)
    assert len(req1.requesters) > 1
    assert INITIAL_REQUEST in req1.requesters
    assert INITIAL_REQUEST == req1.requesters[-1] # Try to keep backwards compatibility
    server_id = next(x for x in req1.requesters if x != INITIAL_REQUEST)

    # Let upstream1 join as server
    upstream1.send(message.AssetStatus(handle=req1.handle, status=message.SUCCESS, ids=ASSET_IDS, servers=[99998888], size=15))
    resp = downstream1.expect(message.AssetStatus)
    assert len(resp.servers) > 1
    assert 99998888 in resp.servers
    assert server_id in resp.servers

    # Let upstream2 join as server
    req2 = upstream2.expect(message.BindRead)
    upstream2.send(message.AssetStatus(handle=req2.handle, status=message.SUCCESS, ids=ASSET_IDS, servers=[12345], size=15))
    resp = downstream1.expect(message.AssetStatus)
    assert 99998888 in resp.servers
    assert server_id in resp.servers
    assert 12345 in resp.servers

    # Let upstream1 leave
    upstream1.send(message.AssetStatus(handle=req2.handle, status=message.NOTFOUND))
    resp = downstream1.expect(message.AssetStatus)
    assert 99998888 not in resp.servers
    assert server_id in resp.servers
    assert 12345 in resp.servers

    # Add another downstream
    downstream2.send(message.BindRead(handle=1, ids=ASSET_IDS, timeout=500, requesters=[9873474]))
    req2 = upstream2.expect(message.BindRead)
    assert len(req2.requesters) > 2
    assert INITIAL_REQUEST in req2.requesters
    assert 9873474 in req2.requesters

    resp2 = downstream2.expect(message.AssetStatus(status=message.SUCCESS))

    # Simulate loop on client1
    downstream1.send(message.BindRead(handle=1, ids=ASSET_IDS, timeout=500, requesters=[9873474, 12345]))
    resp1 = downstream1.expect(message.AssetStatus(status=message.WOULD_LOOP))

    # Simulate redundantly connected downstream on downstream1
    downstream1.send(message.BindRead(handle=1, ids=ASSET_IDS, timeout=500, requesters=[9873474]))
    resp1 = downstream1.expect(message.AssetStatus(status=message.SUCCESS))

    # Close both downstreams
    downstream1.send(message.BindRead(handle=1, ids=[]))
    downstream1.expect(message.AssetStatus(handle=1, status=message.NOTFOUND))
    downstream2.send(message.BindRead(handle=1, ids=[]))
    downstream2.expect(message.AssetStatus(handle=1, status=message.NOTFOUND))

    # Simulate late loop
    downstream1.send(message.BindRead(handle=1, ids=ASSET_IDS, timeout=500, requesters=req2.requesters))
    downstream1.expect(message.AssetStatus(status=message.WOULD_LOOP))

    # Reset
    for x in (upstream1, upstream2, downstream1, downstream2):
        x.close()
    upstream1 = TestConnection(bithorded, name='upstream1')
    downstream1 = TestConnection(bithorded, name='downstream1')

    # Test concurrent-request race-condition
    downstream1.send(message.BindRead(handle=1, ids=ASSET_IDS, timeout=500, requesters=[123124]))
    req = upstream1.expect(message.BindRead(handle=1, ids=ASSET_IDS))
    assert 123124 in req.requesters

    # Assume upstream1 had already sent this, unknowing that server would ask too.
    upstream1.send(message.BindRead(handle=1, ids=ASSET_IDS, timeout=500, requesters=[123124, 421321]))
    upstream1.send(message.AssetStatus(handle=req.handle, status=message.SUCCESS, ids=ASSET_IDS, size=15, servers=[1018,421321]))

    # Server is now assumed to reject upstream1, since the same node appear as requester and server
    upstream1.expect(message.BindRead(handle=1, ids=[]))
    downstream1.expect(message.AssetStatus(handle=1, status=message.NOTFOUND))