HBASE-2578 Add ability for tests to override server-side timestamp setting (currentTimeMillis)
git-svn-id: https://svn.apache.org/repos/asf/hbase/trunk@952479 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
parent
304464ee06
commit
a01c22af11
|
@ -671,6 +671,8 @@ Release 0.21.0 - Unreleased
|
|||
failing hudson on occasion)
|
||||
HBASE-2651 Allow alternate column separators to be specified for ImportTsv
|
||||
HBASE-2661 Add test case for row atomicity guarantee
|
||||
HBASE-2578 Add ability for tests to override server-side timestamp
|
||||
setting (currentTimeMillis) (Daniel Ploeg via Ryan Rawson)
|
||||
|
||||
NEW FEATURES
|
||||
HBASE-1961 HBase EC2 scripts
|
||||
|
|
|
@ -52,6 +52,7 @@ import org.apache.hadoop.hbase.regionserver.wal.HLog;
|
|||
import org.apache.hadoop.hbase.regionserver.wal.WALEdit;
|
||||
import org.apache.hadoop.hbase.util.Bytes;
|
||||
import org.apache.hadoop.hbase.util.ClassSize;
|
||||
import org.apache.hadoop.hbase.util.EnvironmentEdgeManager;
|
||||
import org.apache.hadoop.hbase.util.FSUtils;
|
||||
import org.apache.hadoop.hbase.util.Writables;
|
||||
import org.apache.hadoop.io.Writable;
|
||||
|
@ -358,7 +359,7 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
|
||||
// HRegion is ready to go!
|
||||
this.writestate.compacting = false;
|
||||
this.lastFlushTime = System.currentTimeMillis();
|
||||
this.lastFlushTime = EnvironmentEdgeManager.currentTimeMillis();
|
||||
LOG.info("region " + this +
|
||||
" available; sequence id is " + this.minSequenceId);
|
||||
}
|
||||
|
@ -665,7 +666,7 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
}
|
||||
// Calculate regionid to use. Can't be less than that of parent else
|
||||
// it'll insert into wrong location over in .META. table: HBASE-710.
|
||||
long rid = System.currentTimeMillis();
|
||||
long rid = EnvironmentEdgeManager.currentTimeMillis();
|
||||
if (rid < this.regionInfo.getRegionId()) {
|
||||
LOG.warn("Clock skew; parent regions id is " +
|
||||
this.regionInfo.getRegionId() + " but current time here is " + rid);
|
||||
|
@ -830,7 +831,7 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
}
|
||||
LOG.info("Starting" + (majorCompaction? " major " : " ") +
|
||||
"compaction on region " + this);
|
||||
long startTime = System.currentTimeMillis();
|
||||
long startTime = EnvironmentEdgeManager.currentTimeMillis();
|
||||
doRegionCompactionPrep();
|
||||
long maxSize = -1;
|
||||
for (Store store: stores.values()) {
|
||||
|
@ -841,7 +842,7 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
}
|
||||
}
|
||||
doRegionCompactionCleanup();
|
||||
String timeTaken = StringUtils.formatTimeDiff(System.currentTimeMillis(),
|
||||
String timeTaken = StringUtils.formatTimeDiff(EnvironmentEdgeManager.currentTimeMillis(),
|
||||
startTime);
|
||||
LOG.info("compaction completed on region " + this + " in " + timeTaken);
|
||||
} finally {
|
||||
|
@ -943,7 +944,7 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
* because a Snapshot was not properly persisted.
|
||||
*/
|
||||
protected boolean internalFlushcache() throws IOException {
|
||||
final long startTime = System.currentTimeMillis();
|
||||
final long startTime = EnvironmentEdgeManager.currentTimeMillis();
|
||||
// Clear flush flag.
|
||||
// Record latest flush time
|
||||
this.lastFlushTime = startTime;
|
||||
|
@ -1082,7 +1083,7 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
}
|
||||
|
||||
if (LOG.isDebugEnabled()) {
|
||||
long now = System.currentTimeMillis();
|
||||
long now = EnvironmentEdgeManager.currentTimeMillis();
|
||||
LOG.debug("Finished memstore flush of ~" +
|
||||
StringUtils.humanReadableInt(currentMemStoreSize) + " for region " +
|
||||
this + " in " + (now - startTime) + "ms, sequence id=" + sequenceId +
|
||||
|
@ -1268,7 +1269,7 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
*/
|
||||
public void delete(Map<byte[], List<KeyValue>> familyMap, boolean writeToWAL)
|
||||
throws IOException {
|
||||
long now = System.currentTimeMillis();
|
||||
long now = EnvironmentEdgeManager.currentTimeMillis();
|
||||
byte [] byteNow = Bytes.toBytes(now);
|
||||
boolean flush = false;
|
||||
|
||||
|
@ -1429,7 +1430,6 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
// If we did not pass an existing row lock, obtain a new one
|
||||
Integer lid = getLock(lockid, row);
|
||||
|
||||
byte [] now = Bytes.toBytes(System.currentTimeMillis());
|
||||
try {
|
||||
// All edits for the given row (across all column families) must happen atomically.
|
||||
put(put.getFamilyMap(), writeToWAL);
|
||||
|
@ -1476,8 +1476,6 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
checkFamily(family);
|
||||
get.addColumn(family, qualifier);
|
||||
|
||||
byte [] now = Bytes.toBytes(System.currentTimeMillis());
|
||||
|
||||
// Lock row
|
||||
Integer lid = getLock(lockId, get.getRow());
|
||||
List<KeyValue> result = new ArrayList<KeyValue>();
|
||||
|
@ -1627,7 +1625,7 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
*/
|
||||
private void put(final Map<byte [], List<KeyValue>> familyMap,
|
||||
boolean writeToWAL) throws IOException {
|
||||
long now = System.currentTimeMillis();
|
||||
long now = EnvironmentEdgeManager.currentTimeMillis();
|
||||
byte[] byteNow = Bytes.toBytes(now);
|
||||
boolean flush = false;
|
||||
this.updatesLock.readLock().lock();
|
||||
|
@ -2256,7 +2254,7 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
try {
|
||||
List<KeyValue> edits = new ArrayList<KeyValue>();
|
||||
edits.add(new KeyValue(row, CATALOG_FAMILY, REGIONINFO_QUALIFIER,
|
||||
System.currentTimeMillis(), Writables.getBytes(r.getRegionInfo())));
|
||||
EnvironmentEdgeManager.currentTimeMillis(), Writables.getBytes(r.getRegionInfo())));
|
||||
meta.put(HConstants.CATALOG_FAMILY, edits);
|
||||
} finally {
|
||||
meta.releaseRowLock(lid);
|
||||
|
@ -2692,12 +2690,12 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
|
||||
// bulid the KeyValue now:
|
||||
KeyValue newKv = new KeyValue(row, family,
|
||||
qualifier, System.currentTimeMillis(),
|
||||
qualifier, EnvironmentEdgeManager.currentTimeMillis(),
|
||||
Bytes.toBytes(result));
|
||||
|
||||
// now log it:
|
||||
if (writeToWAL) {
|
||||
long now = System.currentTimeMillis();
|
||||
long now = EnvironmentEdgeManager.currentTimeMillis();
|
||||
WALEdit walEdit = new WALEdit();
|
||||
walEdit.add(newKv);
|
||||
this.log.append(regionInfo, regionInfo.getTableDesc().getName(),
|
||||
|
@ -2908,7 +2906,7 @@ public class HRegion implements HConstants, HeapSize { // , Writable{
|
|||
Configuration c = HBaseConfiguration.create();
|
||||
FileSystem fs = FileSystem.get(c);
|
||||
Path logdir = new Path(c.get("hbase.tmp.dir"),
|
||||
"hlog" + tableDir.getName() + System.currentTimeMillis());
|
||||
"hlog" + tableDir.getName() + EnvironmentEdgeManager.currentTimeMillis());
|
||||
Path oldLogDir = new Path(c.get("hbase.tmp.dir"), HREGION_OLDLOGDIR_NAME);
|
||||
HLog log = new HLog(fs, logdir, oldLogDir, c, null);
|
||||
try {
|
||||
|
|
|
@ -0,0 +1,37 @@
|
|||
/*
|
||||
* Copyright 2010 The Apache Software Foundation
|
||||
*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one
|
||||
* or more contributor license agreements. See the NOTICE file
|
||||
* distributed with this work for additional information
|
||||
* regarding copyright ownership. The ASF licenses this file
|
||||
* to you under the Apache License, Version 2.0 (the
|
||||
* "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.apache.hadoop.hbase.util;
|
||||
|
||||
/**
|
||||
* Default implementation of an environment edge.
|
||||
*/
|
||||
public class DefaultEnvironmentEdge implements EnvironmentEdge {
|
||||
|
||||
|
||||
/**
|
||||
* {@inheritDoc}
|
||||
* <p/>
|
||||
* This implementation returns {@link System#currentTimeMillis()}
|
||||
*/
|
||||
@Override
|
||||
public long currentTimeMillis() {
|
||||
return System.currentTimeMillis();
|
||||
}
|
||||
}
|
|
@ -0,0 +1,36 @@
|
|||
/*
|
||||
* Copyright 2010 The Apache Software Foundation
|
||||
*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one
|
||||
* or more contributor license agreements. See the NOTICE file
|
||||
* distributed with this work for additional information
|
||||
* regarding copyright ownership. The ASF licenses this file
|
||||
* to you under the Apache License, Version 2.0 (the
|
||||
* "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.apache.hadoop.hbase.util;
|
||||
|
||||
/**
|
||||
* Has some basic interaction with the environment. Alternate implementations
|
||||
* can be used where required (eg in tests).
|
||||
*
|
||||
* @see EnvironmentEdgeManager
|
||||
*/
|
||||
public interface EnvironmentEdge {
|
||||
|
||||
/**
|
||||
* Returns the currentTimeMillis.
|
||||
*
|
||||
* @return currentTimeMillis.
|
||||
*/
|
||||
long currentTimeMillis();
|
||||
}
|
|
@ -0,0 +1,75 @@
|
|||
/*
|
||||
* Copyright 2010 The Apache Software Foundation
|
||||
*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one
|
||||
* or more contributor license agreements. See the NOTICE file
|
||||
* distributed with this work for additional information
|
||||
* regarding copyright ownership. The ASF licenses this file
|
||||
* to you under the Apache License, Version 2.0 (the
|
||||
* "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.apache.hadoop.hbase.util;
|
||||
|
||||
/**
|
||||
* Manages a singleton instance of the environment edge. This class shall
|
||||
* implement static versions of the interface {@link EnvironmentEdge}, then
|
||||
* defer to the delegate on invocation.
|
||||
*/
|
||||
public class EnvironmentEdgeManager {
|
||||
private static volatile EnvironmentEdge delegate = new DefaultEnvironmentEdge();
|
||||
|
||||
private EnvironmentEdgeManager() {
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieves the singleton instance of the {@link EnvironmentEdge} that is
|
||||
* being managed.
|
||||
*
|
||||
* @return the edge.
|
||||
*/
|
||||
public static EnvironmentEdge getDelegate() {
|
||||
return delegate;
|
||||
}
|
||||
|
||||
/**
|
||||
* Resets the managed instance to the default instance: {@link
|
||||
* DefaultEnvironmentEdge}.
|
||||
*/
|
||||
static void reset() {
|
||||
injectEdge(new DefaultEnvironmentEdge());
|
||||
}
|
||||
|
||||
/**
|
||||
* Injects the given edge such that it becomes the managed entity. If null is
|
||||
* passed to this method, the default type is assigned to the delegate.
|
||||
*
|
||||
* @param edge the new edge.
|
||||
*/
|
||||
static void injectEdge(EnvironmentEdge edge) {
|
||||
if (edge == null) {
|
||||
reset();
|
||||
} else {
|
||||
delegate = edge;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Defers to the delegate and calls the
|
||||
* {@link EnvironmentEdge#currentTimeMillis()} method.
|
||||
*
|
||||
* @return current time in millis according to the delegate.
|
||||
*/
|
||||
public static long currentTimeMillis() {
|
||||
return getDelegate().currentTimeMillis();
|
||||
}
|
||||
}
|
|
@ -0,0 +1,39 @@
|
|||
/*
|
||||
* Copyright 2010 The Apache Software Foundation
|
||||
*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one
|
||||
* or more contributor license agreements. See the NOTICE file
|
||||
* distributed with this work for additional information
|
||||
* regarding copyright ownership. The ASF licenses this file
|
||||
* to you under the Apache License, Version 2.0 (the
|
||||
* "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.apache.hadoop.hbase.util;
|
||||
|
||||
/**
|
||||
* Uses an incrementing algorithm instead of the default.
|
||||
*/
|
||||
public class IncrementingEnvironmentEdge implements EnvironmentEdge {
|
||||
|
||||
private long timeIncrement = 1;
|
||||
|
||||
/**
|
||||
* {@inheritDoc}
|
||||
* <p/>
|
||||
* This method increments a known value for the current time each time this
|
||||
* method is called. The first value is 1.
|
||||
*/
|
||||
@Override
|
||||
public synchronized long currentTimeMillis() {
|
||||
return timeIncrement++;
|
||||
}
|
||||
}
|
|
@ -24,12 +24,12 @@ import org.apache.commons.logging.LogFactory;
|
|||
import org.apache.hadoop.fs.Path;
|
||||
import org.apache.hadoop.hbase.HBaseConfiguration;
|
||||
import org.apache.hadoop.hbase.HBaseTestCase;
|
||||
import org.apache.hadoop.hbase.HBaseTestingUtility;
|
||||
import org.apache.hadoop.hbase.HColumnDescriptor;
|
||||
import org.apache.hadoop.hbase.HConstants;
|
||||
import org.apache.hadoop.hbase.HRegionInfo;
|
||||
import org.apache.hadoop.hbase.HTableDescriptor;
|
||||
import org.apache.hadoop.hbase.KeyValue;
|
||||
import org.apache.hadoop.hbase.HBaseTestingUtility;
|
||||
import org.apache.hadoop.hbase.client.Delete;
|
||||
import org.apache.hadoop.hbase.client.Get;
|
||||
import org.apache.hadoop.hbase.client.Put;
|
||||
|
@ -45,6 +45,8 @@ import org.apache.hadoop.hbase.filter.PrefixFilter;
|
|||
import org.apache.hadoop.hbase.filter.SingleColumnValueFilter;
|
||||
import org.apache.hadoop.hbase.regionserver.HRegion.RegionScanner;
|
||||
import org.apache.hadoop.hbase.util.Bytes;
|
||||
import org.apache.hadoop.hbase.util.EnvironmentEdgeManagerTestHelper;
|
||||
import org.apache.hadoop.hbase.util.IncrementingEnvironmentEdge;
|
||||
import org.apache.hadoop.hbase.util.Threads;
|
||||
|
||||
import java.io.IOException;
|
||||
|
@ -90,6 +92,12 @@ public class TestHRegion extends HBaseTestCase {
|
|||
super.setUp();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void tearDown() throws Exception {
|
||||
super.tearDown();
|
||||
EnvironmentEdgeManagerTestHelper.reset();
|
||||
}
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////////
|
||||
// New tests that doesn't spin up a mini cluster but rather just test the
|
||||
// individual code pieces in the HRegion. Putting files locally in
|
||||
|
@ -629,6 +637,7 @@ public class TestHRegion extends HBaseTestCase {
|
|||
byte [][] families = {fam};
|
||||
String method = this.getName();
|
||||
initHRegion(tableName, method, families);
|
||||
EnvironmentEdgeManagerTestHelper.injectEdge(new IncrementingEnvironmentEdge());
|
||||
|
||||
byte [] row = Bytes.toBytes("table_name");
|
||||
// column names
|
||||
|
@ -667,9 +676,6 @@ public class TestHRegion extends HBaseTestCase {
|
|||
result = region.get(get, null);
|
||||
assertEquals(1, result.size());
|
||||
|
||||
// Sleep to ensure timestamp of next Put is bigger than previous delete
|
||||
Thread.sleep(10);
|
||||
|
||||
// Assert that after a delete, I can put.
|
||||
put = new Put(row);
|
||||
put.add(fam, splitA, Bytes.toBytes("reference_A"));
|
||||
|
@ -683,9 +689,6 @@ public class TestHRegion extends HBaseTestCase {
|
|||
region.delete(delete, null, false);
|
||||
assertEquals(0, region.get(get, null).size());
|
||||
|
||||
// Sleep to ensure timestamp of next Put is bigger than previous delete
|
||||
Thread.sleep(10);
|
||||
|
||||
region.put(new Put(row).add(fam, splitA, Bytes.toBytes("reference_A")));
|
||||
result = region.get(get, null);
|
||||
assertEquals(1, result.size());
|
||||
|
@ -781,16 +784,14 @@ public class TestHRegion extends HBaseTestCase {
|
|||
public void doTestDelete_AndPostInsert(Delete delete)
|
||||
throws IOException, InterruptedException {
|
||||
initHRegion(tableName, getName(), fam1);
|
||||
EnvironmentEdgeManagerTestHelper.injectEdge(new IncrementingEnvironmentEdge());
|
||||
Put put = new Put(row);
|
||||
put.add(fam1, qual1, value1);
|
||||
region.put(put);
|
||||
|
||||
Thread.sleep(10);
|
||||
|
||||
// now delete the value:
|
||||
region.delete(delete, null, true);
|
||||
|
||||
Thread.sleep(10);
|
||||
|
||||
// ok put data:
|
||||
put = new Put(row);
|
||||
|
|
|
@ -0,0 +1,36 @@
|
|||
/*
|
||||
* Copyright 2010 The Apache Software Foundation
|
||||
*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one
|
||||
* or more contributor license agreements. See the NOTICE file
|
||||
* distributed with this work for additional information
|
||||
* regarding copyright ownership. The ASF licenses this file
|
||||
* to you under the Apache License, Version 2.0 (the
|
||||
* "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.apache.hadoop.hbase.util;
|
||||
|
||||
/**
|
||||
* Used by tests to inject an edge into the manager. The intent is to minimise
|
||||
* the use of the injectEdge method giving it default permissions, but in
|
||||
* testing we may need to use this functionality elsewhere.
|
||||
*/
|
||||
public class EnvironmentEdgeManagerTestHelper {
|
||||
|
||||
public static void reset() {
|
||||
EnvironmentEdgeManager.reset();
|
||||
}
|
||||
|
||||
public static void injectEdge(EnvironmentEdge edge) {
|
||||
EnvironmentEdgeManager.injectEdge(edge);
|
||||
}
|
||||
}
|
|
@ -0,0 +1,50 @@
|
|||
/*
|
||||
* Copyright 2010 The Apache Software Foundation
|
||||
*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one
|
||||
* or more contributor license agreements. See the NOTICE file
|
||||
* distributed with this work for additional information
|
||||
* regarding copyright ownership. The ASF licenses this file
|
||||
* to you under the Apache License, Version 2.0 (the
|
||||
* "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.apache.hadoop.hbase.util;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
import static junit.framework.Assert.assertTrue;
|
||||
import static junit.framework.Assert.fail;
|
||||
|
||||
/**
|
||||
* Tests to make sure that the default environment edge conforms to appropriate
|
||||
* behaviour.
|
||||
*/
|
||||
public class TestDefaultEnvironmentEdge {
|
||||
|
||||
@Test
|
||||
public void testGetCurrentTimeUsesSystemClock() {
|
||||
DefaultEnvironmentEdge edge = new DefaultEnvironmentEdge();
|
||||
long systemTime = System.currentTimeMillis();
|
||||
long edgeTime = edge.currentTimeMillis();
|
||||
assertTrue("System time must be either the same or less than the edge time",
|
||||
systemTime < edgeTime || systemTime == edgeTime);
|
||||
try {
|
||||
Thread.sleep(1);
|
||||
} catch (InterruptedException e) {
|
||||
fail(e.getMessage());
|
||||
}
|
||||
long secondEdgeTime = edge.currentTimeMillis();
|
||||
assertTrue("Second time must be greater than the first",
|
||||
secondEdgeTime > edgeTime);
|
||||
}
|
||||
|
||||
}
|
|
@ -0,0 +1,63 @@
|
|||
/*
|
||||
* Copyright 2010 The Apache Software Foundation
|
||||
*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one
|
||||
* or more contributor license agreements. See the NOTICE file
|
||||
* distributed with this work for additional information
|
||||
* regarding copyright ownership. The ASF licenses this file
|
||||
* to you under the Apache License, Version 2.0 (the
|
||||
* "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.apache.hadoop.hbase.util;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
import static org.mockito.Mockito.mock;
|
||||
import static org.mockito.Mockito.when;
|
||||
import static org.mockito.Mockito.verify;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertFalse;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
public class TestEnvironmentEdgeManager {
|
||||
|
||||
@Test
|
||||
public void testManageSingleton() {
|
||||
EnvironmentEdge edge = EnvironmentEdgeManager.getDelegate();
|
||||
assertNotNull(edge);
|
||||
assertTrue(edge instanceof DefaultEnvironmentEdge);
|
||||
EnvironmentEdgeManager.reset();
|
||||
EnvironmentEdge edge2 = EnvironmentEdgeManager.getDelegate();
|
||||
assertFalse(edge == edge2);
|
||||
IncrementingEnvironmentEdge newEdge = new IncrementingEnvironmentEdge();
|
||||
EnvironmentEdgeManager.injectEdge(newEdge);
|
||||
assertEquals(newEdge, EnvironmentEdgeManager.getDelegate());
|
||||
|
||||
//injecting null will result in default being assigned.
|
||||
EnvironmentEdgeManager.injectEdge(null);
|
||||
EnvironmentEdge nullResult = EnvironmentEdgeManager.getDelegate();
|
||||
assertTrue(nullResult instanceof DefaultEnvironmentEdge);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCurrentTimeInMillis() {
|
||||
EnvironmentEdge mock = mock(EnvironmentEdge.class);
|
||||
EnvironmentEdgeManager.injectEdge(mock);
|
||||
long expectation = 3456;
|
||||
when(mock.currentTimeMillis()).thenReturn(expectation);
|
||||
long result = EnvironmentEdgeManager.currentTimeMillis();
|
||||
verify(mock).currentTimeMillis();
|
||||
assertEquals(expectation, result);
|
||||
}
|
||||
}
|
|
@ -0,0 +1,40 @@
|
|||
/*
|
||||
* Copyright 2010 The Apache Software Foundation
|
||||
*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one
|
||||
* or more contributor license agreements. See the NOTICE file
|
||||
* distributed with this work for additional information
|
||||
* regarding copyright ownership. The ASF licenses this file
|
||||
* to you under the Apache License, Version 2.0 (the
|
||||
* "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.apache.hadoop.hbase.util;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
import static junit.framework.Assert.assertEquals;
|
||||
|
||||
/**
|
||||
* Tests that the incrementing environment edge increments time instead of using
|
||||
* the default.
|
||||
*/
|
||||
public class TestIncrementingEnvironmentEdge {
|
||||
|
||||
@Test
|
||||
public void testGetCurrentTimeUsesSystemClock() {
|
||||
IncrementingEnvironmentEdge edge = new IncrementingEnvironmentEdge();
|
||||
assertEquals(1, edge.currentTimeMillis());
|
||||
assertEquals(2, edge.currentTimeMillis());
|
||||
assertEquals(3, edge.currentTimeMillis());
|
||||
assertEquals(4, edge.currentTimeMillis());
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue