def test_get_broadcast_address(self): # FIXME: need some refactory here return model = self.make_one() # loading of b2 is less than b1 self.make_broadcast('b1', 1, 10, 1, 10) self.make_broadcast('b2', 1, 20, 1, 20) self.session.commit() address = model.get_broadcast_address() self.assertEqual(address, 'b2') # clean up self.session = create_session() # loading of b1 is less than b2 self.make_broadcast('b1', 1, 20, 1, 20) self.make_broadcast('b2', 1, 10, 1, 10) self.session.commit() address = model.get_broadcast_address() self.assertEqual(address, 'b1') # clean up self.session = create_session() # all are full self.make_broadcast('b1', 20, 20, 20, 20) self.make_broadcast('b2', 10, 10, 10, 10) self.session.commit() address = model.get_broadcast_address() self.assertEqual(address, None) # clean up self.session = create_session() # should compare res rate first self.make_broadcast('b1', 10, 20, 5, 20) self.make_broadcast('b2', 5, 20, 10, 20) self.session.commit() address = model.get_broadcast_address() self.assertEqual(address, 'b1') # clean up self.session = create_session() # should compare res rate first self.make_broadcast('b1', 5, 20, 10, 20) self.make_broadcast('b2', 10, 20, 5, 20) self.session.commit() address = model.get_broadcast_address() self.assertEqual(address, 'b2') # clean up self.session = create_session() # limit is 0, it means unlimited self.make_broadcast('b1') self.session.commit() address = model.get_broadcast_address() self.assertEqual(address, 'b1') # clean up self.session = create_session() # not alive server should not appear self.make_broadcast('b1', alive=False) self.session.commit() address = model.get_broadcast_address() self.assertEqual(address, None) # clean up self.session = create_session() # not active server should not appear self.make_broadcast('b1', active=False) self.session.commit() address = model.get_broadcast_address() self.assertEqual(address, None) # clean up self.session = create_session()
def setUp(self): import datetime from nowin_core.database import tables from nowin_core.tests.models._helper import create_session tables.set_now_func(datetime.datetime.utcnow) self.session = create_session(zope_transaction=True)
def setUp(self): import datetime tables.set_now_func(datetime.datetime.utcnow) self.session = create_session()
def setUp(self): from nowin_core.database import tables from nowin_core.tests.models._helper import create_session self.session = create_session() self.now_func = MockNow() self.oldNowFunc = tables.set_now_func(self.now_func)