def test_faulty_queue_ID(self):
def add_callback(result):
self.assertEqual(result[0], aliceDQ.DQ_REJECT)
callback_called[0] = True
sim_reset()
callback_called = [False]
alice = QuantumNode("alice", nodeID=0)
bob = QuantumNode("bob", nodeID=1)
conn = ClassicalFibreConnection(alice, bob, length=.0001)
aliceDQ = EGPDistributedQueue(alice,
conn,
accept_all=True,
numQueues=1)
bobDQ = EGPDistributedQueue(bob, conn, accept_all=True, numQueues=1)
aliceDQ.add_callback = add_callback
nodes = [
(alice, [aliceDQ]),
(bob, [bobDQ]),
]
conns = [(conn, "dqp_conn", [aliceDQ, bobDQ])]
network = EasyNetwork(name="DistQueueNetwork",
nodes=nodes,
connections=conns)
network.start()
request = SchedulerRequest()
aliceDQ.add(request, qid=1)
sim_run(10)
self.assertTrue(callback_called[0])
def test_multiple_queues(self):
sim_reset()
alice = QuantumNode("alice", nodeID=0)
bob = QuantumNode("bob", nodeID=1)
conn = ClassicalFibreConnection(alice, bob, length=.0001)
aliceDQ = EGPDistributedQueue(alice,
conn,
accept_all=True,
numQueues=2)
bobDQ = EGPDistributedQueue(bob, conn, accept_all=True, numQueues=2)
nodes = [
(alice, [aliceDQ]),
(bob, [bobDQ]),
]
conns = [(conn, "dqp_conn", [aliceDQ, bobDQ])]
network = EasyNetwork(name="DistQueueNetwork",
nodes=nodes,
connections=conns)
network.start()
alice_requests = [SchedulerRequest(), SchedulerRequest()]
bob_requests = [SchedulerRequest(), SchedulerRequest()]
aliceDQ.add(alice_requests[0], qid=0)
aliceDQ.add(alice_requests[1], qid=1)
bobDQ.add(bob_requests[0], qid=0)
bobDQ.add(bob_requests[1], qid=1)
sim_run(10)
self.assertEqual(len(aliceDQ.queueList[0].queue), 2)
self.assertEqual(len(aliceDQ.queueList[1].queue), 2)
self.assertEqual(len(bobDQ.queueList[0].queue), 2)
self.assertEqual(len(bobDQ.queueList[1].queue), 2)
def test_update_mhp_cycle_number(self):
def callback_alice(queue_item):
callback_called[0] = True
def callback_bob(queue_item):
callback_called[1] = True
sim_reset()
callback_called = [False, False]
alice = QuantumNode("alice", nodeID=0)
bob = QuantumNode("bob", nodeID=1)
conn = ClassicalFibreConnection(alice, bob, length=.0001)
aliceDQ = EGPDistributedQueue(alice,
conn,
timeout_callback=callback_alice,
accept_all=True)
bobDQ = EGPDistributedQueue(bob,
conn,
timeout_callback=callback_bob,
accept_all=True)
nodes = [
(alice, [aliceDQ]),
(bob, [bobDQ]),
]
conns = [(conn, "dqp_conn", [aliceDQ, bobDQ])]
network = EasyNetwork(name="DistQueueNetwork",
nodes=nodes,
connections=conns)
network.start()
request = SchedulerRequest(sched_cycle=1, timeout_cycle=2)
aliceDQ.add(request, 0)
sim_run(10)
queue_item_alice = aliceDQ.local_peek(0)
queue_item_bob = bobDQ.local_peek(0)
self.assertFalse(queue_item_alice.ready)
aliceDQ.update_mhp_cycle_number(1, 10)
self.assertTrue(queue_item_alice.ready)
self.assertFalse(queue_item_bob.ready)
self.assertFalse(callback_called[0])
self.assertFalse(callback_called[1])
aliceDQ.update_mhp_cycle_number(2, 10)
self.assertTrue(callback_called[0])
self.assertFalse(callback_called[1])
bobDQ.update_mhp_cycle_number(1, 10)
self.assertTrue(queue_item_bob.ready)
self.assertFalse(callback_called[1])
bobDQ.update_mhp_cycle_number(2, 10)
self.assertTrue(queue_item_bob.ready)
self.assertTrue(callback_called[1])