1 /** 2 * Licensed to the Apache Software Foundation (ASF) under one 3 * or more contributor license agreements. See the NOTICE file 4 * distributed with this work for additional information 5 * regarding copyright ownership. The ASF licenses this file 6 * to you under the Apache License, Version 2.0 (the 7 * "License"); you may not use this file except in compliance 8 * with the License. You may obtain a copy of the License at 9 * 10 * http://www.apache.org/licenses/LICENSE-2.0 11 * 12 * Unless required by applicable law or agreed to in writing, software 13 * distributed under the License is distributed on an "AS IS" BASIS, 14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 * See the License for the specific language governing permissions and 16 * limitations under the License. 17 */ 18 package org.apache.hadoop.hbase.master; 19 20 import org.apache.hadoop.hbase.master.AssignmentManager.RegionState; 21 22 /** 23 * Package scoped mocking utility. 24 */ 25 public class Mocking { 26 27 static void waitForRegionPendingOpenInRIT(AssignmentManager am, String encodedName) 28 throws InterruptedException { 29 // We used to do a check like this: 30 //!Mocking.verifyRegionState(this.watcher, REGIONINFO, EventType.M_ZK_REGION_OFFLINE)) { 31 // There is a race condition with this: because we may do the transition to 32 // RS_ZK_REGION_OPENING before the RIT is internally updated. We need to wait for the 33 // RIT to be as we need it to be instead. This cannot happen in a real cluster as we 34 // update the RIT before sending the openRegion request. 35 36 boolean wait = true; 37 while (wait) { 38 RegionState state = am.getRegionsInTransition().get(encodedName); 39 if (state != null && state.isPendingOpen()){ 40 wait = false; 41 } else { 42 Thread.sleep(1); 43 } 44 } 45 } 46 47 static void waitForRegionOfflineInRIT(AssignmentManager am, String encodedName) 48 throws InterruptedException { 49 boolean wait = true; 50 while (wait) { 51 RegionState state = am.getRegionsInTransition().get(encodedName); 52 if (state != null && state.isOffline()) { 53 wait = false; 54 } else { 55 Thread.sleep(1); 56 } 57 } 58 } 59 }