class MasterNotificationsHandlerTests(MasterHandlerTests): def setUp(self): super(MasterNotificationsHandlerTests, self).setUp() self.handler = PrimaryNotificationsHandler(self.app) def test_connectionClosed(self): conn = self.getFakeConnection() node = Mock() self.app.master_conn = conn self.app.primary_master_node = node self.handler.connectionClosed(conn) self.assertEqual(self.app.master_conn, None) self.assertEqual(self.app.primary_master_node, None) def test_invalidateObjects(self): conn = self.getFakeConnection() tid = self.getNextTID() oid1, oid2, oid3 = self.getOID(1), self.getOID(2), self.getOID(3) self.app._cache = Mock({ 'invalidate': None, }) self.handler.invalidateObjects(conn, tid, [oid1, oid3]) cache_calls = self.app._cache.mockGetNamedCalls('invalidate') self.assertEqual(len(cache_calls), 2) cache_calls[0].checkArgs(oid1, tid) cache_calls[1].checkArgs(oid3, tid) invalidation_calls = self.db.mockGetNamedCalls('invalidate') self.assertEqual(len(invalidation_calls), 1) invalidation_calls[0].checkArgs(tid, [oid1, oid3]) def test_notifyPartitionChanges(self): conn = self.getFakeConnection() self.app.pt = Mock({'filled': True}) ptid = 0 cell_list = (Mock(), Mock()) self.handler.notifyPartitionChanges(conn, ptid, cell_list) update_calls = self.app.pt.mockGetNamedCalls('update') self.assertEqual(len(update_calls), 1) update_calls[0].checkArgs(ptid, cell_list, self.app.nm)
def setUp(self): super(MasterNotificationsHandlerTests, self).setUp() self.handler = PrimaryNotificationsHandler(self.app)