class MNodeTests(unittest.TestCase):
def setUp(self):
self.node = MNode('TestNode')
def tearDown(self):
self.node = None
def testAddOptionPositive(self):
self.node = MNode('TestNode')
self.node.addoption(MNode('AnotherTest'))
self.assertEqual(self.node.totalcount, 1, 'Legal node was not added.')
def testAddOptionNegative(self):
self.node = MNode('TestNode')
self.node.addoption(None)
self.assertEqual(self.node.totalcount, 0, 'Illegal node was added.')
def testNextPositive(self):
self.node = MNode('TestNode')
self.assertEqual(str(self.node.next()), '',
'Node with no options returned a node with a non-null state.')
self.node.addoption(MNode('ChildNode'))
self.assertEqual(str(self.node.next()), 'ChildNode',
'Node with only one option did not return expected option.')
def getGeoAndTopo():
mc = memcache.Client([MEMCACHED_ADDR + ':' + MEMCACHED_PORT], debug = 0)
nodes = pickle.loads(mc.get('nodes'))
r = {}
for node_key in nodes:
n = {}
m_node = MNode(mc.get(node_key))
n['speed'] = m_node.speed
n['x'] = m_node.x
n['y'] = m_node.y
n['direction'] = m_node.get_direction_in_str();
n['neighbors'] = pickle.loads(mc.get('neighbors_' + node_key))
r[node_key] = n
return r
def testNextPositive(self):
self.node = MNode('TestNode')
self.assertEqual(str(self.node.next()), '',
'Node with no options returned a node with a non-null state.')
self.node.addoption(MNode('ChildNode'))
self.assertEqual(str(self.node.next()), 'ChildNode',
'Node with only one option did not return expected option.')
PRIORITY_CHUNK = 2
PRIORITY_MASTER = 1
PRIORITY_CLIENT = 3
NUM_CLIENT = 10
NUM_CHUNK = 10
NUM_UPLOAD = 20
NUM_UPDATE = 10
NUM_REMOVE = 2
NUM_DOWNLOAD = 2
next = 0
if __name__ == "__main__":
master = MNode()
clist = []
cclist = []
for w in range(NUM_CLIENT):
client = Client(master,NUM_CLIENT,"abcd")
clist.append(client)
for w in range(next,next + NUM_CHUNK):
strid = str(w)
c = ChunkInfo(strid,master)
master.add(c)
cclist.append(c)
next = next + 1
for w in range(NUM_CHUNK):
def setUp(self):
self.node = MNode('TestNode')
def testAddOptionNegative(self):
self.node = MNode('TestNode')
self.node.addoption(None)
self.assertEqual(self.node.totalcount, 0, 'Illegal node was added.')
def testAddOptionPositive(self):
self.node = MNode('TestNode')
self.node.addoption(MNode('AnotherTest'))
self.assertEqual(self.node.totalcount, 1, 'Legal node was not added.')
