def setUp(self):
HTTPApiTestCases.setUp(self)
# Provision Router with additional Users and Routes
u2 = User(2, Group(2), 'user2', 'correct')
u3 = User(3, Group(2), 'user3', 'correct')
u3.mt_credential.setQuota('balance', 10)
self.RouterPB_f.users.append(u2)
self.RouterPB_f.users.append(u3)
filters = [GroupFilter(Group(2))]
route = StaticMTRoute(filters, SmppClientConnector('abc'), 1.5)
self.RouterPB_f.mt_routing_table.add(route, 2)
def test_add_list_and_remove_mt_route(self):
yield self.connect('127.0.0.1', self.pbPort)
yield self.mtroute_add(
StaticMTRoute([GroupFilter(Group(1))],
SmppClientConnector(id_generator()), 0.0), 2)
yield self.mtroute_add(
DefaultRoute(SmppClientConnector(id_generator())), 0)
listRet1 = yield self.mtroute_get_all()
listRet1 = pickle.loads(listRet1)
yield self.mtroute_remove(2)
listRet2 = yield self.mtroute_get_all()
listRet2 = pickle.loads(listRet2)
self.assertEqual(2, len(listRet1))
self.assertEqual(1, len(listRet2))
