209 lines
7.7 KiB
Python
209 lines
7.7 KiB
Python
# Copyright 2014 Modelling, Simulation and Design Lab (MSDL) at
|
|
# McGill University and the University of Antwerp (http://msdl.cs.mcgill.ca/)
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from testutils import *
|
|
from pypdevs.messageScheduler import MessageScheduler
|
|
from copy import deepcopy
|
|
|
|
class TestMessageScheduler(unittest.TestCase):
|
|
def setUp(self):
|
|
self.scheduler = MessageScheduler()
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def test_mscheduler_schedule(self):
|
|
self.assertTrue(len(self.scheduler.heap) == 0)
|
|
self.assertTrue(len(self.scheduler.processed) == 0)
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
# Only [0] and [3] are important, so don't even bother
|
|
# creating the rest
|
|
msg = NetworkMessage((1, 1), {}, 12345, False, 0)
|
|
|
|
self.scheduler.schedule(msg)
|
|
|
|
self.assertTrue(len(self.scheduler.heap) == 1)
|
|
self.assertTrue(len(self.scheduler.processed) == 0)
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
msg = deepcopy(msg)
|
|
msg.uuid = 444
|
|
self.scheduler.schedule(msg)
|
|
|
|
self.assertTrue(len(self.scheduler.heap) == 2)
|
|
self.assertTrue(len(self.scheduler.processed) == 0)
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
msg = deepcopy(msg)
|
|
msg.uuid = 456
|
|
self.scheduler.schedule(msg)
|
|
|
|
self.assertTrue(len(self.scheduler.heap) == 3)
|
|
self.assertTrue(len(self.scheduler.processed) == 0)
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
def test_mschedule_invalids(self):
|
|
self.assertTrue(len(self.scheduler.heap) == 0)
|
|
self.assertTrue(len(self.scheduler.processed) == 0)
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
# Only [0] and [3] are important, so don't even bother
|
|
# creating the rest
|
|
msg = NetworkMessage((1, 1), {}, 12345, False, 0)
|
|
|
|
self.scheduler.schedule(msg)
|
|
|
|
self.assertTrue(len(self.scheduler.heap) == 1)
|
|
self.assertTrue(len(self.scheduler.processed) == 0)
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
msg = deepcopy(msg)
|
|
msg.uuid = 1111
|
|
self.scheduler.invalids = set({1111})
|
|
self.scheduler.schedule(msg)
|
|
|
|
self.assertTrue(len(self.scheduler.heap) == 1)
|
|
self.assertTrue(len(self.scheduler.processed) == 0)
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
msg = deepcopy(msg)
|
|
msg.uuid = 1111
|
|
self.scheduler.schedule(msg)
|
|
|
|
self.assertTrue(len(self.scheduler.heap) == 2)
|
|
self.assertTrue(len(self.scheduler.processed) == 0)
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
msg = deepcopy(msg)
|
|
msg.uuid = 234
|
|
self.scheduler.invalids.add(234)
|
|
self.scheduler.schedule(msg)
|
|
|
|
self.assertTrue(len(self.scheduler.heap) == 2)
|
|
self.assertTrue(len(self.scheduler.processed) == 0)
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
def test_mscheduler_read_first(self):
|
|
msg1 = NetworkMessage((3, 1), {}, 1, False, 0)
|
|
msg2 = NetworkMessage((4, 1), {}, 2, False, 0)
|
|
msg3 = NetworkMessage((5, 1), {}, 3, False, 0)
|
|
self.scheduler.heap = [msg1, msg2, msg3]
|
|
self.scheduler.processed = []
|
|
|
|
self.assertTrue(self.scheduler.heap == [msg1, msg2, msg3])
|
|
self.assertTrue(self.scheduler.processed == [])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
first = self.scheduler.readFirst()
|
|
|
|
self.assertTrue(first == msg1)
|
|
self.assertTrue(self.scheduler.heap == [msg1, msg2, msg3])
|
|
self.assertTrue(self.scheduler.processed == [])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
first = self.scheduler.readFirst()
|
|
|
|
self.assertTrue(first == msg1)
|
|
self.assertTrue(self.scheduler.heap == [msg1, msg2, msg3])
|
|
self.assertTrue(self.scheduler.processed == [])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
self.scheduler.invalids = set({1})
|
|
first = self.scheduler.readFirst()
|
|
|
|
self.assertTrue(first == msg2)
|
|
self.assertTrue(self.scheduler.heap == [msg2, msg3])
|
|
self.assertTrue(self.scheduler.processed == [])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
def test_mscheduler_removeFirst(self):
|
|
msg1 = NetworkMessage((3, 1), {}, 1, False, 0)
|
|
msg2 = NetworkMessage((4, 1), {}, 2, False, 0)
|
|
msg3 = NetworkMessage((5, 1), {}, 3, False, 0)
|
|
self.scheduler.heap = [msg1, msg2, msg3]
|
|
self.scheduler.processed = []
|
|
|
|
self.assertTrue(self.scheduler.heap == [msg1, msg2, msg3])
|
|
self.assertTrue(self.scheduler.processed == [])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
self.scheduler.removeFirst()
|
|
|
|
self.assertTrue(self.scheduler.heap == [msg2, msg3])
|
|
self.assertTrue(self.scheduler.processed == [msg1])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
self.scheduler.invalids = set({2})
|
|
self.scheduler.removeFirst()
|
|
|
|
self.assertTrue(self.scheduler.heap == [])
|
|
self.assertTrue(self.scheduler.processed == [msg1, msg3])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
def test_mscheduler_revert(self):
|
|
msg1 = NetworkMessage((3, 1), {}, 1, False, 0)
|
|
msg2 = NetworkMessage((4, 1), {}, 2, False, 0)
|
|
msg3 = NetworkMessage((5, 1), {}, 3, False, 0)
|
|
self.scheduler.heap = [msg3]
|
|
self.scheduler.processed = [msg1, msg2]
|
|
|
|
self.scheduler.revert((4, 1))
|
|
|
|
self.assertTrue(self.scheduler.heap == [msg2, msg3])
|
|
self.assertTrue(self.scheduler.processed == [msg1])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
self.scheduler.revert((2, 1))
|
|
|
|
self.assertTrue(self.scheduler.heap == [msg1, msg3, msg2])
|
|
self.assertTrue(self.scheduler.processed == [])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
self.scheduler.heap = []
|
|
self.scheduler.processed = [msg1, msg2, msg3]
|
|
|
|
self.scheduler.revert((2, 1))
|
|
|
|
self.assertTrue(self.scheduler.heap == [msg1, msg2, msg3])
|
|
self.assertTrue(self.scheduler.processed == [])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
def test_mscheduler_cleanup(self):
|
|
msg1 = NetworkMessage((3, 1), {}, 1, False, 0)
|
|
msg2 = NetworkMessage((4, 1), {}, 2, False, 0)
|
|
msg3 = NetworkMessage((5, 1), {}, 3, False, 0)
|
|
self.scheduler.heap = [msg3]
|
|
self.scheduler.processed = [msg1, msg2]
|
|
|
|
self.scheduler.cleanup((4, 1))
|
|
|
|
self.assertTrue(self.scheduler.heap == [msg3])
|
|
self.assertTrue(self.scheduler.processed == [msg2])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
self.scheduler.cleanup((5, 1))
|
|
|
|
self.assertTrue(self.scheduler.heap == [msg3])
|
|
self.assertTrue(self.scheduler.processed == [])
|
|
self.assertTrue(self.scheduler.invalids == set({}))
|
|
|
|
self.scheduler.processed = [msg1, msg2]
|
|
self.scheduler.invalids = set({2})
|
|
self.scheduler.cleanup((4, 1))
|
|
|
|
self.assertTrue(self.scheduler.heap == [msg3])
|
|
self.assertTrue(self.scheduler.processed == [msg2])
|
|
self.assertTrue(self.scheduler.invalids == set({2}))
|