You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@accumulo.apache.org by ec...@apache.org on 2015/06/04 20:53:01 UTC
[20/43] accumulo git commit: ACCUMULO-3871 move ITs into distro jar,
stop building test jar
http://git-wip-us.apache.org/repos/asf/accumulo/blob/01ae5b85/test/src/test/java/org/apache/accumulo/test/ConditionalWriterIT.java
----------------------------------------------------------------------
diff --git a/test/src/test/java/org/apache/accumulo/test/ConditionalWriterIT.java b/test/src/test/java/org/apache/accumulo/test/ConditionalWriterIT.java
deleted file mode 100644
index 74d3593..0000000
--- a/test/src/test/java/org/apache/accumulo/test/ConditionalWriterIT.java
+++ /dev/null
@@ -1,1349 +0,0 @@
-/*
- * 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.accumulo.test;
-
-import static org.junit.Assert.assertTrue;
-
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Collections;
-import java.util.HashMap;
-import java.util.HashSet;
-import java.util.Iterator;
-import java.util.Map;
-import java.util.Map.Entry;
-import java.util.Random;
-import java.util.Set;
-import java.util.SortedSet;
-import java.util.TreeSet;
-import java.util.UUID;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.Executors;
-import java.util.concurrent.TimeUnit;
-import java.util.concurrent.atomic.AtomicBoolean;
-
-import org.apache.accumulo.cluster.AccumuloCluster;
-import org.apache.accumulo.cluster.ClusterUser;
-import org.apache.accumulo.core.client.AccumuloException;
-import org.apache.accumulo.core.client.AccumuloSecurityException;
-import org.apache.accumulo.core.client.BatchWriter;
-import org.apache.accumulo.core.client.BatchWriterConfig;
-import org.apache.accumulo.core.client.ClientConfiguration;
-import org.apache.accumulo.core.client.ClientConfiguration.ClientProperty;
-import org.apache.accumulo.core.client.ConditionalWriter;
-import org.apache.accumulo.core.client.ConditionalWriter.Result;
-import org.apache.accumulo.core.client.ConditionalWriter.Status;
-import org.apache.accumulo.core.client.ConditionalWriterConfig;
-import org.apache.accumulo.core.client.Connector;
-import org.apache.accumulo.core.client.IsolatedScanner;
-import org.apache.accumulo.core.client.IteratorSetting;
-import org.apache.accumulo.core.client.RowIterator;
-import org.apache.accumulo.core.client.Scanner;
-import org.apache.accumulo.core.client.TableDeletedException;
-import org.apache.accumulo.core.client.TableExistsException;
-import org.apache.accumulo.core.client.TableNotFoundException;
-import org.apache.accumulo.core.client.TableOfflineException;
-import org.apache.accumulo.core.client.admin.NewTableConfiguration;
-import org.apache.accumulo.core.client.security.tokens.PasswordToken;
-import org.apache.accumulo.core.data.ArrayByteSequence;
-import org.apache.accumulo.core.data.ByteSequence;
-import org.apache.accumulo.core.data.Condition;
-import org.apache.accumulo.core.data.ConditionalMutation;
-import org.apache.accumulo.core.data.Key;
-import org.apache.accumulo.core.data.Mutation;
-import org.apache.accumulo.core.data.Range;
-import org.apache.accumulo.core.data.Value;
-import org.apache.accumulo.core.iterators.LongCombiner.Type;
-import org.apache.accumulo.core.iterators.user.SummingCombiner;
-import org.apache.accumulo.core.iterators.user.VersioningIterator;
-import org.apache.accumulo.core.security.Authorizations;
-import org.apache.accumulo.core.security.ColumnVisibility;
-import org.apache.accumulo.core.security.SystemPermission;
-import org.apache.accumulo.core.security.TablePermission;
-import org.apache.accumulo.core.trace.DistributedTrace;
-import org.apache.accumulo.core.trace.Span;
-import org.apache.accumulo.core.trace.Trace;
-import org.apache.accumulo.core.util.FastFormat;
-import org.apache.accumulo.core.util.UtilWaitThread;
-import org.apache.accumulo.examples.simple.constraints.AlphaNumKeyConstraint;
-import org.apache.accumulo.harness.AccumuloClusterHarness;
-import org.apache.accumulo.minicluster.impl.MiniAccumuloClusterImpl;
-import org.apache.accumulo.test.functional.BadIterator;
-import org.apache.accumulo.test.functional.SlowIterator;
-import org.apache.accumulo.tracer.TraceDump;
-import org.apache.accumulo.tracer.TraceDump.Printer;
-import org.apache.accumulo.tracer.TraceServer;
-import org.apache.hadoop.io.Text;
-import org.junit.Assert;
-import org.junit.Assume;
-import org.junit.Before;
-import org.junit.Test;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-
-import com.google.common.collect.Iterables;
-
-/**
- *
- */
-public class ConditionalWriterIT extends AccumuloClusterHarness {
- private static final Logger log = LoggerFactory.getLogger(ConditionalWriterIT.class);
-
- @Override
- protected int defaultTimeoutSeconds() {
- return 60;
- }
-
- public static long abs(long l) {
- l = Math.abs(l); // abs(Long.MIN_VALUE) == Long.MIN_VALUE...
- if (l < 0)
- return 0;
- return l;
- }
-
- @Before
- public void deleteUsers() throws Exception {
- Connector conn = getConnector();
- Set<String> users = conn.securityOperations().listLocalUsers();
- ClusterUser user = getUser(0);
- if (users.contains(user.getPrincipal())) {
- conn.securityOperations().dropLocalUser(user.getPrincipal());
- }
- }
-
- @Test
- public void testBasic() throws Exception {
-
- Connector conn = getConnector();
- String tableName = getUniqueNames(1)[0];
-
- conn.tableOperations().create(tableName);
-
- ConditionalWriter cw = conn.createConditionalWriter(tableName, new ConditionalWriterConfig());
-
- // mutation conditional on column tx:seq not existing
- ConditionalMutation cm0 = new ConditionalMutation("99006", new Condition("tx", "seq"));
- cm0.put("name", "last", "doe");
- cm0.put("name", "first", "john");
- cm0.put("tx", "seq", "1");
- Assert.assertEquals(Status.ACCEPTED, cw.write(cm0).getStatus());
- Assert.assertEquals(Status.REJECTED, cw.write(cm0).getStatus());
-
- // mutation conditional on column tx:seq being 1
- ConditionalMutation cm1 = new ConditionalMutation("99006", new Condition("tx", "seq").setValue("1"));
- cm1.put("name", "last", "Doe");
- cm1.put("tx", "seq", "2");
- Assert.assertEquals(Status.ACCEPTED, cw.write(cm1).getStatus());
-
- // test condition where value differs
- ConditionalMutation cm2 = new ConditionalMutation("99006", new Condition("tx", "seq").setValue("1"));
- cm2.put("name", "last", "DOE");
- cm2.put("tx", "seq", "2");
- Assert.assertEquals(Status.REJECTED, cw.write(cm2).getStatus());
-
- // test condition where column does not exists
- ConditionalMutation cm3 = new ConditionalMutation("99006", new Condition("txtypo", "seq").setValue("1"));
- cm3.put("name", "last", "deo");
- cm3.put("tx", "seq", "2");
- Assert.assertEquals(Status.REJECTED, cw.write(cm3).getStatus());
-
- // test two conditions, where one should fail
- ConditionalMutation cm4 = new ConditionalMutation("99006", new Condition("tx", "seq").setValue("2"), new Condition("name", "last").setValue("doe"));
- cm4.put("name", "last", "deo");
- cm4.put("tx", "seq", "3");
- Assert.assertEquals(Status.REJECTED, cw.write(cm4).getStatus());
-
- // test two conditions, where one should fail
- ConditionalMutation cm5 = new ConditionalMutation("99006", new Condition("tx", "seq").setValue("1"), new Condition("name", "last").setValue("Doe"));
- cm5.put("name", "last", "deo");
- cm5.put("tx", "seq", "3");
- Assert.assertEquals(Status.REJECTED, cw.write(cm5).getStatus());
-
- // ensure rejected mutations did not write
- Scanner scanner = conn.createScanner(tableName, Authorizations.EMPTY);
- scanner.fetchColumn(new Text("name"), new Text("last"));
- scanner.setRange(new Range("99006"));
- Entry<Key,Value> entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("Doe", entry.getValue().toString());
-
- // test w/ two conditions that are met
- ConditionalMutation cm6 = new ConditionalMutation("99006", new Condition("tx", "seq").setValue("2"), new Condition("name", "last").setValue("Doe"));
- cm6.put("name", "last", "DOE");
- cm6.put("tx", "seq", "3");
- Assert.assertEquals(Status.ACCEPTED, cw.write(cm6).getStatus());
-
- entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("DOE", entry.getValue().toString());
-
- // test a conditional mutation that deletes
- ConditionalMutation cm7 = new ConditionalMutation("99006", new Condition("tx", "seq").setValue("3"));
- cm7.putDelete("name", "last");
- cm7.putDelete("name", "first");
- cm7.putDelete("tx", "seq");
- Assert.assertEquals(Status.ACCEPTED, cw.write(cm7).getStatus());
-
- Assert.assertFalse("Did not expect to find any results", scanner.iterator().hasNext());
-
- // add the row back
- Assert.assertEquals(Status.ACCEPTED, cw.write(cm0).getStatus());
- Assert.assertEquals(Status.REJECTED, cw.write(cm0).getStatus());
-
- entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("doe", entry.getValue().toString());
- }
-
- @Test
- public void testFields() throws Exception {
-
- Connector conn = getConnector();
- String tableName = getUniqueNames(1)[0];
-
- String user = null;
- ClientConfiguration clientConf = cluster.getClientConfig();
- final boolean saslEnabled = clientConf.getBoolean(ClientProperty.INSTANCE_RPC_SASL_ENABLED.getKey(), false);
-
- ClusterUser user1 = getUser(0);
- user = user1.getPrincipal();
- if (saslEnabled) {
- // The token is pointless for kerberos
- conn.securityOperations().createLocalUser(user, null);
- } else {
- conn.securityOperations().createLocalUser(user, new PasswordToken(user1.getPassword()));
- }
-
- Authorizations auths = new Authorizations("A", "B");
-
- conn.securityOperations().changeUserAuthorizations(user, auths);
- conn.securityOperations().grantSystemPermission(user, SystemPermission.CREATE_TABLE);
-
- conn = conn.getInstance().getConnector(user, user1.getToken());
-
- conn.tableOperations().create(tableName);
-
- ConditionalWriter cw = conn.createConditionalWriter(tableName, new ConditionalWriterConfig().setAuthorizations(auths));
-
- ColumnVisibility cva = new ColumnVisibility("A");
- ColumnVisibility cvb = new ColumnVisibility("B");
-
- ConditionalMutation cm0 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cva));
- cm0.put("name", "last", cva, "doe");
- cm0.put("name", "first", cva, "john");
- cm0.put("tx", "seq", cva, "1");
- Assert.assertEquals(Status.ACCEPTED, cw.write(cm0).getStatus());
-
- Scanner scanner = conn.createScanner(tableName, auths);
- scanner.setRange(new Range("99006"));
- // TODO verify all columns
- scanner.fetchColumn(new Text("tx"), new Text("seq"));
- Entry<Key,Value> entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("1", entry.getValue().toString());
- long ts = entry.getKey().getTimestamp();
-
- // test wrong colf
- ConditionalMutation cm1 = new ConditionalMutation("99006", new Condition("txA", "seq").setVisibility(cva).setValue("1"));
- cm1.put("name", "last", cva, "Doe");
- cm1.put("name", "first", cva, "John");
- cm1.put("tx", "seq", cva, "2");
- Assert.assertEquals(Status.REJECTED, cw.write(cm1).getStatus());
-
- // test wrong colq
- ConditionalMutation cm2 = new ConditionalMutation("99006", new Condition("tx", "seqA").setVisibility(cva).setValue("1"));
- cm2.put("name", "last", cva, "Doe");
- cm2.put("name", "first", cva, "John");
- cm2.put("tx", "seq", cva, "2");
- Assert.assertEquals(Status.REJECTED, cw.write(cm2).getStatus());
-
- // test wrong colv
- ConditionalMutation cm3 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvb).setValue("1"));
- cm3.put("name", "last", cva, "Doe");
- cm3.put("name", "first", cva, "John");
- cm3.put("tx", "seq", cva, "2");
- Assert.assertEquals(Status.REJECTED, cw.write(cm3).getStatus());
-
- // test wrong timestamp
- ConditionalMutation cm4 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cva).setTimestamp(ts + 1).setValue("1"));
- cm4.put("name", "last", cva, "Doe");
- cm4.put("name", "first", cva, "John");
- cm4.put("tx", "seq", cva, "2");
- Assert.assertEquals(Status.REJECTED, cw.write(cm4).getStatus());
-
- // test wrong timestamp
- ConditionalMutation cm5 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cva).setTimestamp(ts - 1).setValue("1"));
- cm5.put("name", "last", cva, "Doe");
- cm5.put("name", "first", cva, "John");
- cm5.put("tx", "seq", cva, "2");
- Assert.assertEquals(Status.REJECTED, cw.write(cm5).getStatus());
-
- // ensure no updates were made
- entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("1", entry.getValue().toString());
-
- // set all columns correctly
- ConditionalMutation cm6 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cva).setTimestamp(ts).setValue("1"));
- cm6.put("name", "last", cva, "Doe");
- cm6.put("name", "first", cva, "John");
- cm6.put("tx", "seq", cva, "2");
- Assert.assertEquals(Status.ACCEPTED, cw.write(cm6).getStatus());
-
- entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("2", entry.getValue().toString());
-
- }
-
- @Test
- public void testBadColVis() throws Exception {
- // test when a user sets a col vis in a condition that can never be seen
-
- Connector conn = getConnector();
- String tableName = getUniqueNames(1)[0];
-
- conn.tableOperations().create(tableName);
-
- Authorizations auths = new Authorizations("A", "B");
-
- conn.securityOperations().changeUserAuthorizations(getAdminPrincipal(), auths);
-
- Authorizations filteredAuths = new Authorizations("A");
-
- ConditionalWriter cw = conn.createConditionalWriter(tableName, new ConditionalWriterConfig().setAuthorizations(filteredAuths));
-
- ColumnVisibility cva = new ColumnVisibility("A");
- ColumnVisibility cvb = new ColumnVisibility("B");
- ColumnVisibility cvc = new ColumnVisibility("C");
-
- // User has authorization, but didn't include it in the writer
- ConditionalMutation cm0 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvb));
- cm0.put("name", "last", cva, "doe");
- cm0.put("name", "first", cva, "john");
- cm0.put("tx", "seq", cva, "1");
- Assert.assertEquals(Status.INVISIBLE_VISIBILITY, cw.write(cm0).getStatus());
-
- ConditionalMutation cm1 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvb).setValue("1"));
- cm1.put("name", "last", cva, "doe");
- cm1.put("name", "first", cva, "john");
- cm1.put("tx", "seq", cva, "1");
- Assert.assertEquals(Status.INVISIBLE_VISIBILITY, cw.write(cm1).getStatus());
-
- // User does not have the authorization
- ConditionalMutation cm2 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvc));
- cm2.put("name", "last", cva, "doe");
- cm2.put("name", "first", cva, "john");
- cm2.put("tx", "seq", cva, "1");
- Assert.assertEquals(Status.INVISIBLE_VISIBILITY, cw.write(cm2).getStatus());
-
- ConditionalMutation cm3 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvc).setValue("1"));
- cm3.put("name", "last", cva, "doe");
- cm3.put("name", "first", cva, "john");
- cm3.put("tx", "seq", cva, "1");
- Assert.assertEquals(Status.INVISIBLE_VISIBILITY, cw.write(cm3).getStatus());
-
- // if any visibility is bad, good visibilities don't override
- ConditionalMutation cm4 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvb), new Condition("tx", "seq").setVisibility(cva));
-
- cm4.put("name", "last", cva, "doe");
- cm4.put("name", "first", cva, "john");
- cm4.put("tx", "seq", cva, "1");
- Assert.assertEquals(Status.INVISIBLE_VISIBILITY, cw.write(cm4).getStatus());
-
- ConditionalMutation cm5 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvb).setValue("1"), new Condition("tx", "seq")
- .setVisibility(cva).setValue("1"));
- cm5.put("name", "last", cva, "doe");
- cm5.put("name", "first", cva, "john");
- cm5.put("tx", "seq", cva, "1");
- Assert.assertEquals(Status.INVISIBLE_VISIBILITY, cw.write(cm5).getStatus());
-
- ConditionalMutation cm6 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvb).setValue("1"),
- new Condition("tx", "seq").setVisibility(cva));
- cm6.put("name", "last", cva, "doe");
- cm6.put("name", "first", cva, "john");
- cm6.put("tx", "seq", cva, "1");
- Assert.assertEquals(Status.INVISIBLE_VISIBILITY, cw.write(cm6).getStatus());
-
- ConditionalMutation cm7 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvb), new Condition("tx", "seq").setVisibility(cva)
- .setValue("1"));
- cm7.put("name", "last", cva, "doe");
- cm7.put("name", "first", cva, "john");
- cm7.put("tx", "seq", cva, "1");
- Assert.assertEquals(Status.INVISIBLE_VISIBILITY, cw.write(cm7).getStatus());
-
- cw.close();
-
- // test passing auths that exceed users configured auths
-
- Authorizations exceedingAuths = new Authorizations("A", "B", "D");
- ConditionalWriter cw2 = conn.createConditionalWriter(tableName, new ConditionalWriterConfig().setAuthorizations(exceedingAuths));
-
- ConditionalMutation cm8 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvb), new Condition("tx", "seq").setVisibility(cva)
- .setValue("1"));
- cm8.put("name", "last", cva, "doe");
- cm8.put("name", "first", cva, "john");
- cm8.put("tx", "seq", cva, "1");
-
- try {
- Status status = cw2.write(cm8).getStatus();
- Assert.fail("Writing mutation with Authorizations the user doesn't have should fail. Got status: " + status);
- } catch (AccumuloSecurityException ase) {
- // expected, check specific failure?
- } finally {
- cw2.close();
- }
- }
-
- @Test
- public void testConstraints() throws Exception {
- // ensure constraint violations are properly reported
-
- Connector conn = getConnector();
- String tableName = getUniqueNames(1)[0];
-
- conn.tableOperations().create(tableName);
- conn.tableOperations().addConstraint(tableName, AlphaNumKeyConstraint.class.getName());
- conn.tableOperations().clone(tableName, tableName + "_clone", true, new HashMap<String,String>(), new HashSet<String>());
-
- Scanner scanner = conn.createScanner(tableName + "_clone", new Authorizations());
-
- ConditionalWriter cw = conn.createConditionalWriter(tableName + "_clone", new ConditionalWriterConfig());
-
- ConditionalMutation cm0 = new ConditionalMutation("99006+", new Condition("tx", "seq"));
- cm0.put("tx", "seq", "1");
-
- Assert.assertEquals(Status.VIOLATED, cw.write(cm0).getStatus());
- Assert.assertFalse("Should find no results in the table is mutation result was violated", scanner.iterator().hasNext());
-
- ConditionalMutation cm1 = new ConditionalMutation("99006", new Condition("tx", "seq"));
- cm1.put("tx", "seq", "1");
-
- Assert.assertEquals(Status.ACCEPTED, cw.write(cm1).getStatus());
- Assert.assertTrue("Accepted result should be returned when reading table", scanner.iterator().hasNext());
-
- cw.close();
- }
-
- @Test
- public void testIterators() throws Exception {
-
- Connector conn = getConnector();
- String tableName = getUniqueNames(1)[0];
-
- conn.tableOperations().create(tableName, new NewTableConfiguration().withoutDefaultIterators());
-
- BatchWriter bw = conn.createBatchWriter(tableName, new BatchWriterConfig());
-
- Mutation m = new Mutation("ACCUMULO-1000");
- m.put("count", "comments", "1");
- bw.addMutation(m);
- bw.addMutation(m);
- bw.addMutation(m);
-
- m = new Mutation("ACCUMULO-1001");
- m.put("count2", "comments", "1");
- bw.addMutation(m);
- bw.addMutation(m);
-
- m = new Mutation("ACCUMULO-1002");
- m.put("count2", "comments", "1");
- bw.addMutation(m);
- bw.addMutation(m);
-
- bw.close();
-
- IteratorSetting iterConfig = new IteratorSetting(10, SummingCombiner.class);
- SummingCombiner.setEncodingType(iterConfig, Type.STRING);
- SummingCombiner.setColumns(iterConfig, Collections.singletonList(new IteratorSetting.Column("count")));
-
- IteratorSetting iterConfig2 = new IteratorSetting(10, SummingCombiner.class);
- SummingCombiner.setEncodingType(iterConfig2, Type.STRING);
- SummingCombiner.setColumns(iterConfig2, Collections.singletonList(new IteratorSetting.Column("count2", "comments")));
-
- IteratorSetting iterConfig3 = new IteratorSetting(5, VersioningIterator.class);
- VersioningIterator.setMaxVersions(iterConfig3, 1);
-
- Scanner scanner = conn.createScanner(tableName, new Authorizations());
- scanner.addScanIterator(iterConfig);
- scanner.setRange(new Range("ACCUMULO-1000"));
- scanner.fetchColumn(new Text("count"), new Text("comments"));
-
- Entry<Key,Value> entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("3", entry.getValue().toString());
-
- ConditionalWriter cw = conn.createConditionalWriter(tableName, new ConditionalWriterConfig());
-
- ConditionalMutation cm0 = new ConditionalMutation("ACCUMULO-1000", new Condition("count", "comments").setValue("3"));
- cm0.put("count", "comments", "1");
- Assert.assertEquals(Status.REJECTED, cw.write(cm0).getStatus());
- entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("3", entry.getValue().toString());
-
- ConditionalMutation cm1 = new ConditionalMutation("ACCUMULO-1000", new Condition("count", "comments").setIterators(iterConfig).setValue("3"));
- cm1.put("count", "comments", "1");
- Assert.assertEquals(Status.ACCEPTED, cw.write(cm1).getStatus());
- entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("4", entry.getValue().toString());
-
- ConditionalMutation cm2 = new ConditionalMutation("ACCUMULO-1000", new Condition("count", "comments").setValue("4"));
- cm2.put("count", "comments", "1");
- Assert.assertEquals(Status.REJECTED, cw.write(cm1).getStatus());
- entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("4", entry.getValue().toString());
-
- // run test with multiple iterators passed in same batch and condition with two iterators
-
- ConditionalMutation cm3 = new ConditionalMutation("ACCUMULO-1000", new Condition("count", "comments").setIterators(iterConfig).setValue("4"));
- cm3.put("count", "comments", "1");
-
- ConditionalMutation cm4 = new ConditionalMutation("ACCUMULO-1001", new Condition("count2", "comments").setIterators(iterConfig2).setValue("2"));
- cm4.put("count2", "comments", "1");
-
- ConditionalMutation cm5 = new ConditionalMutation("ACCUMULO-1002", new Condition("count2", "comments").setIterators(iterConfig2, iterConfig3).setValue("2"));
- cm5.put("count2", "comments", "1");
-
- Iterator<Result> results = cw.write(Arrays.asList(cm3, cm4, cm5).iterator());
- Map<String,Status> actual = new HashMap<String,Status>();
-
- while (results.hasNext()) {
- Result result = results.next();
- String k = new String(result.getMutation().getRow());
- Assert.assertFalse("Did not expect to see multiple resultus for the row: " + k, actual.containsKey(k));
- actual.put(k, result.getStatus());
- }
-
- Map<String,Status> expected = new HashMap<String,Status>();
- expected.put("ACCUMULO-1000", Status.ACCEPTED);
- expected.put("ACCUMULO-1001", Status.ACCEPTED);
- expected.put("ACCUMULO-1002", Status.REJECTED);
-
- Assert.assertEquals(expected, actual);
-
- // TODO test w/ table that has iterators configured
-
- cw.close();
- }
-
- @Test
- public void testBatch() throws Exception {
-
- Connector conn = getConnector();
- String tableName = getUniqueNames(1)[0];
-
- conn.tableOperations().create(tableName);
-
- conn.securityOperations().changeUserAuthorizations(getAdminPrincipal(), new Authorizations("A", "B"));
-
- ColumnVisibility cvab = new ColumnVisibility("A|B");
-
- ArrayList<ConditionalMutation> mutations = new ArrayList<ConditionalMutation>();
-
- ConditionalMutation cm0 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvab));
- cm0.put("name", "last", cvab, "doe");
- cm0.put("name", "first", cvab, "john");
- cm0.put("tx", "seq", cvab, "1");
- mutations.add(cm0);
-
- ConditionalMutation cm1 = new ConditionalMutation("59056", new Condition("tx", "seq").setVisibility(cvab));
- cm1.put("name", "last", cvab, "doe");
- cm1.put("name", "first", cvab, "jane");
- cm1.put("tx", "seq", cvab, "1");
- mutations.add(cm1);
-
- ConditionalMutation cm2 = new ConditionalMutation("19059", new Condition("tx", "seq").setVisibility(cvab));
- cm2.put("name", "last", cvab, "doe");
- cm2.put("name", "first", cvab, "jack");
- cm2.put("tx", "seq", cvab, "1");
- mutations.add(cm2);
-
- ConditionalWriter cw = conn.createConditionalWriter(tableName, new ConditionalWriterConfig().setAuthorizations(new Authorizations("A")));
- Iterator<Result> results = cw.write(mutations.iterator());
- int count = 0;
- while (results.hasNext()) {
- Result result = results.next();
- Assert.assertEquals(Status.ACCEPTED, result.getStatus());
- count++;
- }
-
- Assert.assertEquals(3, count);
-
- Scanner scanner = conn.createScanner(tableName, new Authorizations("A"));
- scanner.fetchColumn(new Text("tx"), new Text("seq"));
-
- for (String row : new String[] {"99006", "59056", "19059"}) {
- scanner.setRange(new Range(row));
- Entry<Key,Value> entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("1", entry.getValue().toString());
- }
-
- TreeSet<Text> splits = new TreeSet<Text>();
- splits.add(new Text("7"));
- splits.add(new Text("3"));
- conn.tableOperations().addSplits(tableName, splits);
-
- mutations.clear();
-
- ConditionalMutation cm3 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvab).setValue("1"));
- cm3.put("name", "last", cvab, "Doe");
- cm3.put("tx", "seq", cvab, "2");
- mutations.add(cm3);
-
- ConditionalMutation cm4 = new ConditionalMutation("59056", new Condition("tx", "seq").setVisibility(cvab));
- cm4.put("name", "last", cvab, "Doe");
- cm4.put("tx", "seq", cvab, "1");
- mutations.add(cm4);
-
- ConditionalMutation cm5 = new ConditionalMutation("19059", new Condition("tx", "seq").setVisibility(cvab).setValue("2"));
- cm5.put("name", "last", cvab, "Doe");
- cm5.put("tx", "seq", cvab, "3");
- mutations.add(cm5);
-
- results = cw.write(mutations.iterator());
- int accepted = 0;
- int rejected = 0;
- while (results.hasNext()) {
- Result result = results.next();
- if (new String(result.getMutation().getRow()).equals("99006")) {
- Assert.assertEquals(Status.ACCEPTED, result.getStatus());
- accepted++;
- } else {
- Assert.assertEquals(Status.REJECTED, result.getStatus());
- rejected++;
- }
- }
-
- Assert.assertEquals("Expected only one accepted conditional mutation", 1, accepted);
- Assert.assertEquals("Expected two rejected conditional mutations", 2, rejected);
-
- for (String row : new String[] {"59056", "19059"}) {
- scanner.setRange(new Range(row));
- Entry<Key,Value> entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("1", entry.getValue().toString());
- }
-
- scanner.setRange(new Range("99006"));
- Entry<Key,Value> entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("2", entry.getValue().toString());
-
- scanner.clearColumns();
- scanner.fetchColumn(new Text("name"), new Text("last"));
- entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("Doe", entry.getValue().toString());
-
- cw.close();
- }
-
- @Test
- public void testBigBatch() throws Exception {
-
- Connector conn = getConnector();
- String tableName = getUniqueNames(1)[0];
-
- conn.tableOperations().create(tableName);
- conn.tableOperations().addSplits(tableName, nss("2", "4", "6"));
-
- UtilWaitThread.sleep(2000);
-
- int num = 100;
-
- ArrayList<byte[]> rows = new ArrayList<byte[]>(num);
- ArrayList<ConditionalMutation> cml = new ArrayList<ConditionalMutation>(num);
-
- Random r = new Random();
- byte[] e = new byte[0];
-
- for (int i = 0; i < num; i++) {
- rows.add(FastFormat.toZeroPaddedString(abs(r.nextLong()), 16, 16, e));
- }
-
- for (int i = 0; i < num; i++) {
- ConditionalMutation cm = new ConditionalMutation(rows.get(i), new Condition("meta", "seq"));
-
- cm.put("meta", "seq", "1");
- cm.put("meta", "tx", UUID.randomUUID().toString());
-
- cml.add(cm);
- }
-
- ConditionalWriter cw = conn.createConditionalWriter(tableName, new ConditionalWriterConfig());
-
- Iterator<Result> results = cw.write(cml.iterator());
-
- int count = 0;
-
- // TODO check got each row back
- while (results.hasNext()) {
- Result result = results.next();
- Assert.assertEquals(Status.ACCEPTED, result.getStatus());
- count++;
- }
-
- Assert.assertEquals("Did not receive the expected number of results", num, count);
-
- ArrayList<ConditionalMutation> cml2 = new ArrayList<ConditionalMutation>(num);
-
- for (int i = 0; i < num; i++) {
- ConditionalMutation cm = new ConditionalMutation(rows.get(i), new Condition("meta", "seq").setValue("1"));
-
- cm.put("meta", "seq", "2");
- cm.put("meta", "tx", UUID.randomUUID().toString());
-
- cml2.add(cm);
- }
-
- count = 0;
-
- results = cw.write(cml2.iterator());
-
- while (results.hasNext()) {
- Result result = results.next();
- Assert.assertEquals(Status.ACCEPTED, result.getStatus());
- count++;
- }
-
- Assert.assertEquals("Did not receive the expected number of results", num, count);
-
- cw.close();
- }
-
- @Test
- public void testBatchErrors() throws Exception {
-
- Connector conn = getConnector();
- String tableName = getUniqueNames(1)[0];
-
- conn.tableOperations().create(tableName);
- conn.tableOperations().addConstraint(tableName, AlphaNumKeyConstraint.class.getName());
- conn.tableOperations().clone(tableName, tableName + "_clone", true, new HashMap<String,String>(), new HashSet<String>());
-
- conn.securityOperations().changeUserAuthorizations(getAdminPrincipal(), new Authorizations("A", "B"));
-
- ColumnVisibility cvaob = new ColumnVisibility("A|B");
- ColumnVisibility cvaab = new ColumnVisibility("A&B");
-
- switch ((new Random()).nextInt(3)) {
- case 1:
- conn.tableOperations().addSplits(tableName, nss("6"));
- break;
- case 2:
- conn.tableOperations().addSplits(tableName, nss("2", "95"));
- break;
- }
-
- ArrayList<ConditionalMutation> mutations = new ArrayList<ConditionalMutation>();
-
- ConditionalMutation cm0 = new ConditionalMutation("99006", new Condition("tx", "seq").setVisibility(cvaob));
- cm0.put("name+", "last", cvaob, "doe");
- cm0.put("name", "first", cvaob, "john");
- cm0.put("tx", "seq", cvaob, "1");
- mutations.add(cm0);
-
- ConditionalMutation cm1 = new ConditionalMutation("59056", new Condition("tx", "seq").setVisibility(cvaab));
- cm1.put("name", "last", cvaab, "doe");
- cm1.put("name", "first", cvaab, "jane");
- cm1.put("tx", "seq", cvaab, "1");
- mutations.add(cm1);
-
- ConditionalMutation cm2 = new ConditionalMutation("19059", new Condition("tx", "seq").setVisibility(cvaob));
- cm2.put("name", "last", cvaob, "doe");
- cm2.put("name", "first", cvaob, "jack");
- cm2.put("tx", "seq", cvaob, "1");
- mutations.add(cm2);
-
- ConditionalMutation cm3 = new ConditionalMutation("90909", new Condition("tx", "seq").setVisibility(cvaob).setValue("1"));
- cm3.put("name", "last", cvaob, "doe");
- cm3.put("name", "first", cvaob, "john");
- cm3.put("tx", "seq", cvaob, "2");
- mutations.add(cm3);
-
- ConditionalWriter cw = conn.createConditionalWriter(tableName, new ConditionalWriterConfig().setAuthorizations(new Authorizations("A")));
- Iterator<Result> results = cw.write(mutations.iterator());
- HashSet<String> rows = new HashSet<String>();
- while (results.hasNext()) {
- Result result = results.next();
- String row = new String(result.getMutation().getRow());
- if (row.equals("19059")) {
- Assert.assertEquals(Status.ACCEPTED, result.getStatus());
- } else if (row.equals("59056")) {
- Assert.assertEquals(Status.INVISIBLE_VISIBILITY, result.getStatus());
- } else if (row.equals("99006")) {
- Assert.assertEquals(Status.VIOLATED, result.getStatus());
- } else if (row.equals("90909")) {
- Assert.assertEquals(Status.REJECTED, result.getStatus());
- }
- rows.add(row);
- }
-
- Assert.assertEquals(4, rows.size());
-
- Scanner scanner = conn.createScanner(tableName, new Authorizations("A"));
- scanner.fetchColumn(new Text("tx"), new Text("seq"));
-
- Entry<Key,Value> entry = Iterables.getOnlyElement(scanner);
- Assert.assertEquals("1", entry.getValue().toString());
-
- cw.close();
- }
-
- @Test
- public void testSameRow() throws Exception {
- // test multiple mutations for same row in same batch
-
- Connector conn = getConnector();
- String tableName = getUniqueNames(1)[0];
-
- conn.tableOperations().create(tableName);
-
- ConditionalWriter cw = conn.createConditionalWriter(tableName, new ConditionalWriterConfig());
-
- ConditionalMutation cm1 = new ConditionalMutation("r1", new Condition("tx", "seq"));
- cm1.put("tx", "seq", "1");
- cm1.put("data", "x", "a");
-
- Assert.assertEquals(Status.ACCEPTED, cw.write(cm1).getStatus());
-
- ConditionalMutation cm2 = new ConditionalMutation("r1", new Condition("tx", "seq").setValue("1"));
- cm2.put("tx", "seq", "2");
- cm2.put("data", "x", "b");
-
- ConditionalMutation cm3 = new ConditionalMutation("r1", new Condition("tx", "seq").setValue("1"));
- cm3.put("tx", "seq", "2");
- cm3.put("data", "x", "c");
-
- ConditionalMutation cm4 = new ConditionalMutation("r1", new Condition("tx", "seq").setValue("1"));
- cm4.put("tx", "seq", "2");
- cm4.put("data", "x", "d");
-
- Iterator<Result> results = cw.write(Arrays.asList(cm2, cm3, cm4).iterator());
-
- int accepted = 0;
- int rejected = 0;
- int total = 0;
-
- while (results.hasNext()) {
- Status status = results.next().getStatus();
- if (status == Status.ACCEPTED)
- accepted++;
- if (status == Status.REJECTED)
- rejected++;
- total++;
- }
-
- Assert.assertEquals("Expected one accepted result", 1, accepted);
- Assert.assertEquals("Expected two rejected results", 2, rejected);
- Assert.assertEquals("Expected three total results", 3, total);
-
- cw.close();
- }
-
- private static class Stats {
-
- ByteSequence row = null;
- int seq;
- long sum;
- int data[] = new int[10];
-
- public Stats(Iterator<Entry<Key,Value>> iterator) {
- while (iterator.hasNext()) {
- Entry<Key,Value> entry = iterator.next();
-
- if (row == null)
- row = entry.getKey().getRowData();
-
- String cf = entry.getKey().getColumnFamilyData().toString();
- String cq = entry.getKey().getColumnQualifierData().toString();
-
- if (cf.equals("data")) {
- data[Integer.parseInt(cq)] = Integer.parseInt(entry.getValue().toString());
- } else if (cf.equals("meta")) {
- if (cq.equals("sum")) {
- sum = Long.parseLong(entry.getValue().toString());
- } else if (cq.equals("seq")) {
- seq = Integer.parseInt(entry.getValue().toString());
- }
- }
- }
-
- long sum2 = 0;
-
- for (int datum : data) {
- sum2 += datum;
- }
-
- Assert.assertEquals(sum2, sum);
- }
-
- public Stats(ByteSequence row) {
- this.row = row;
- for (int i = 0; i < data.length; i++) {
- this.data[i] = 0;
- }
- this.seq = -1;
- this.sum = 0;
- }
-
- void set(int index, int value) {
- sum -= data[index];
- sum += value;
- data[index] = value;
- }
-
- ConditionalMutation toMutation() {
- Condition cond = new Condition("meta", "seq");
- if (seq >= 0)
- cond.setValue(seq + "");
-
- ConditionalMutation cm = new ConditionalMutation(row, cond);
-
- cm.put("meta", "seq", (seq + 1) + "");
- cm.put("meta", "sum", (sum) + "");
-
- for (int i = 0; i < data.length; i++) {
- cm.put("data", i + "", data[i] + "");
- }
-
- return cm;
- }
-
- @Override
- public String toString() {
- return row + " " + seq + " " + sum;
- }
- }
-
- private static class MutatorTask implements Runnable {
- String table;
- ArrayList<ByteSequence> rows;
- ConditionalWriter cw;
- Connector conn;
- AtomicBoolean failed;
-
- public MutatorTask(String table, Connector conn, ArrayList<ByteSequence> rows, ConditionalWriter cw, AtomicBoolean failed) {
- this.table = table;
- this.rows = rows;
- this.conn = conn;
- this.cw = cw;
- this.failed = failed;
- }
-
- @Override
- public void run() {
- try {
- Random rand = new Random();
-
- Scanner scanner = new IsolatedScanner(conn.createScanner(table, Authorizations.EMPTY));
-
- for (int i = 0; i < 20; i++) {
- int numRows = rand.nextInt(10) + 1;
-
- ArrayList<ByteSequence> changes = new ArrayList<ByteSequence>(numRows);
- ArrayList<ConditionalMutation> mutations = new ArrayList<ConditionalMutation>();
-
- for (int j = 0; j < numRows; j++)
- changes.add(rows.get(rand.nextInt(rows.size())));
-
- for (ByteSequence row : changes) {
- scanner.setRange(new Range(row.toString()));
- Stats stats = new Stats(scanner.iterator());
- stats.set(rand.nextInt(10), rand.nextInt(Integer.MAX_VALUE));
- mutations.add(stats.toMutation());
- }
-
- ArrayList<ByteSequence> changed = new ArrayList<ByteSequence>(numRows);
- Iterator<Result> results = cw.write(mutations.iterator());
- while (results.hasNext()) {
- Result result = results.next();
- changed.add(new ArrayByteSequence(result.getMutation().getRow()));
- }
-
- Collections.sort(changes);
- Collections.sort(changed);
-
- Assert.assertEquals(changes, changed);
-
- }
-
- } catch (Exception e) {
- log.error("{}", e.getMessage(), e);
- failed.set(true);
- }
- }
- }
-
- @Test
- public void testThreads() throws Exception {
- // test multiple threads using a single conditional writer
-
- String table = getUniqueNames(1)[0];
- Connector conn = getConnector();
-
- conn.tableOperations().create(table);
-
- Random rand = new Random();
-
- switch (rand.nextInt(3)) {
- case 1:
- conn.tableOperations().addSplits(table, nss("4"));
- break;
- case 2:
- conn.tableOperations().addSplits(table, nss("3", "5"));
- break;
- }
-
- ConditionalWriter cw = conn.createConditionalWriter(table, new ConditionalWriterConfig());
-
- ArrayList<ByteSequence> rows = new ArrayList<ByteSequence>();
-
- for (int i = 0; i < 1000; i++) {
- rows.add(new ArrayByteSequence(FastFormat.toZeroPaddedString(abs(rand.nextLong()), 16, 16, new byte[0])));
- }
-
- ArrayList<ConditionalMutation> mutations = new ArrayList<ConditionalMutation>();
-
- for (ByteSequence row : rows)
- mutations.add(new Stats(row).toMutation());
-
- ArrayList<ByteSequence> rows2 = new ArrayList<ByteSequence>();
- Iterator<Result> results = cw.write(mutations.iterator());
- while (results.hasNext()) {
- Result result = results.next();
- Assert.assertEquals(Status.ACCEPTED, result.getStatus());
- rows2.add(new ArrayByteSequence(result.getMutation().getRow()));
- }
-
- Collections.sort(rows);
- Collections.sort(rows2);
-
- Assert.assertEquals(rows, rows2);
-
- AtomicBoolean failed = new AtomicBoolean(false);
-
- ExecutorService tp = Executors.newFixedThreadPool(5);
- for (int i = 0; i < 5; i++) {
- tp.submit(new MutatorTask(table, conn, rows, cw, failed));
- }
-
- tp.shutdown();
-
- while (!tp.isTerminated()) {
- tp.awaitTermination(1, TimeUnit.MINUTES);
- }
-
- Assert.assertFalse("A MutatorTask failed with an exception", failed.get());
-
- Scanner scanner = conn.createScanner(table, Authorizations.EMPTY);
-
- RowIterator rowIter = new RowIterator(scanner);
-
- while (rowIter.hasNext()) {
- Iterator<Entry<Key,Value>> row = rowIter.next();
- new Stats(row);
- }
- }
-
- private SortedSet<Text> nss(String... splits) {
- TreeSet<Text> ret = new TreeSet<Text>();
- for (String split : splits)
- ret.add(new Text(split));
-
- return ret;
- }
-
- @Test
- public void testSecurity() throws Exception {
- // test against table user does not have read and/or write permissions for
- Connector conn = getConnector();
- String user = null;
- ClientConfiguration clientConf = cluster.getClientConfig();
- final boolean saslEnabled = clientConf.getBoolean(ClientProperty.INSTANCE_RPC_SASL_ENABLED.getKey(), false);
-
- // Create a new user
- ClusterUser user1 = getUser(0);
- user = user1.getPrincipal();
- if (saslEnabled) {
- conn.securityOperations().createLocalUser(user, null);
- } else {
- conn.securityOperations().createLocalUser(user, new PasswordToken(user1.getPassword()));
- }
-
- String[] tables = getUniqueNames(3);
- String table1 = tables[0], table2 = tables[1], table3 = tables[2];
-
- // Create three tables
- conn.tableOperations().create(table1);
- conn.tableOperations().create(table2);
- conn.tableOperations().create(table3);
-
- // Grant R on table1, W on table2, R/W on table3
- conn.securityOperations().grantTablePermission(user, table1, TablePermission.READ);
- conn.securityOperations().grantTablePermission(user, table2, TablePermission.WRITE);
- conn.securityOperations().grantTablePermission(user, table3, TablePermission.READ);
- conn.securityOperations().grantTablePermission(user, table3, TablePermission.WRITE);
-
- // Login as the user
- Connector conn2 = conn.getInstance().getConnector(user, user1.getToken());
-
- ConditionalMutation cm1 = new ConditionalMutation("r1", new Condition("tx", "seq"));
- cm1.put("tx", "seq", "1");
- cm1.put("data", "x", "a");
-
- ConditionalWriter cw1 = conn2.createConditionalWriter(table1, new ConditionalWriterConfig());
- ConditionalWriter cw2 = conn2.createConditionalWriter(table2, new ConditionalWriterConfig());
- ConditionalWriter cw3 = conn2.createConditionalWriter(table3, new ConditionalWriterConfig());
-
- // Should be able to conditional-update a table we have R/W on
- Assert.assertEquals(Status.ACCEPTED, cw3.write(cm1).getStatus());
-
- // Conditional-update to a table we only have read on should fail
- try {
- Status status = cw1.write(cm1).getStatus();
- Assert.fail("Expected exception writing conditional mutation to table the user doesn't have write access to, Got status: " + status);
- } catch (AccumuloSecurityException ase) {
-
- }
-
- // Conditional-update to a table we only have writer on should fail
- try {
- Status status = cw2.write(cm1).getStatus();
- Assert.fail("Expected exception writing conditional mutation to table the user doesn't have read access to. Got status: " + status);
- } catch (AccumuloSecurityException ase) {
-
- }
- }
-
- @Test
- public void testTimeout() throws Exception {
- Connector conn = getConnector();
-
- String table = getUniqueNames(1)[0];
-
- conn.tableOperations().create(table);
-
- ConditionalWriter cw = conn.createConditionalWriter(table, new ConditionalWriterConfig().setTimeout(3, TimeUnit.SECONDS));
-
- ConditionalMutation cm1 = new ConditionalMutation("r1", new Condition("tx", "seq"));
- cm1.put("tx", "seq", "1");
- cm1.put("data", "x", "a");
-
- Assert.assertEquals(cw.write(cm1).getStatus(), Status.ACCEPTED);
-
- IteratorSetting is = new IteratorSetting(5, SlowIterator.class);
- SlowIterator.setSeekSleepTime(is, 5000);
-
- ConditionalMutation cm2 = new ConditionalMutation("r1", new Condition("tx", "seq").setValue("1").setIterators(is));
- cm2.put("tx", "seq", "2");
- cm2.put("data", "x", "b");
-
- Assert.assertEquals(cw.write(cm2).getStatus(), Status.UNKNOWN);
-
- Scanner scanner = conn.createScanner(table, Authorizations.EMPTY);
-
- for (Entry<Key,Value> entry : scanner) {
- String cf = entry.getKey().getColumnFamilyData().toString();
- String cq = entry.getKey().getColumnQualifierData().toString();
- String val = entry.getValue().toString();
-
- if (cf.equals("tx") && cq.equals("seq"))
- Assert.assertEquals("Unexpected value in tx:seq", "1", val);
- else if (cf.equals("data") && cq.equals("x"))
- Assert.assertEquals("Unexpected value in data:x", "a", val);
- else
- Assert.fail("Saw unexpected column family and qualifier: " + entry);
- }
-
- ConditionalMutation cm3 = new ConditionalMutation("r1", new Condition("tx", "seq").setValue("1"));
- cm3.put("tx", "seq", "2");
- cm3.put("data", "x", "b");
-
- Assert.assertEquals(cw.write(cm3).getStatus(), Status.ACCEPTED);
-
- cw.close();
- }
-
- @Test
- public void testDeleteTable() throws Exception {
- String table = getUniqueNames(1)[0];
- Connector conn = getConnector();
-
- try {
- conn.createConditionalWriter(table, new ConditionalWriterConfig());
- Assert.fail("Creating conditional writer for table that doesn't exist should fail");
- } catch (TableNotFoundException e) {}
-
- conn.tableOperations().create(table);
-
- ConditionalWriter cw = conn.createConditionalWriter(table, new ConditionalWriterConfig());
-
- conn.tableOperations().delete(table);
-
- ConditionalMutation cm1 = new ConditionalMutation("r1", new Condition("tx", "seq"));
- cm1.put("tx", "seq", "1");
- cm1.put("data", "x", "a");
-
- Result result = cw.write(cm1);
-
- try {
- Status status = result.getStatus();
- Assert.fail("Expected exception writing conditional mutation to deleted table. Got status: " + status);
- } catch (AccumuloException ae) {
- Assert.assertEquals(TableDeletedException.class, ae.getCause().getClass());
- }
- }
-
- @Test
- public void testOffline() throws Exception {
- String table = getUniqueNames(1)[0];
- Connector conn = getConnector();
-
- conn.tableOperations().create(table);
-
- ConditionalWriter cw = conn.createConditionalWriter(table, new ConditionalWriterConfig());
-
- conn.tableOperations().offline(table, true);
-
- ConditionalMutation cm1 = new ConditionalMutation("r1", new Condition("tx", "seq"));
- cm1.put("tx", "seq", "1");
- cm1.put("data", "x", "a");
-
- Result result = cw.write(cm1);
-
- try {
- Status status = result.getStatus();
- Assert.fail("Expected exception writing conditional mutation to offline table. Got status: " + status);
- } catch (AccumuloException ae) {
- Assert.assertEquals(TableOfflineException.class, ae.getCause().getClass());
- }
-
- cw.close();
-
- try {
- conn.createConditionalWriter(table, new ConditionalWriterConfig());
- Assert.fail("Expected exception creating conditional writer to offline table");
- } catch (TableOfflineException e) {}
- }
-
- @Test
- public void testError() throws Exception {
- String table = getUniqueNames(1)[0];
- Connector conn = getConnector();
-
- conn.tableOperations().create(table);
-
- ConditionalWriter cw = conn.createConditionalWriter(table, new ConditionalWriterConfig());
-
- IteratorSetting iterSetting = new IteratorSetting(5, BadIterator.class);
-
- ConditionalMutation cm1 = new ConditionalMutation("r1", new Condition("tx", "seq").setIterators(iterSetting));
- cm1.put("tx", "seq", "1");
- cm1.put("data", "x", "a");
-
- Result result = cw.write(cm1);
-
- try {
- Status status = result.getStatus();
- Assert.fail("Expected exception using iterator which throws an error, Got status: " + status);
- } catch (AccumuloException ae) {
-
- }
-
- cw.close();
- }
-
- @Test(expected = IllegalArgumentException.class)
- public void testNoConditions() throws AccumuloException, AccumuloSecurityException, TableExistsException, TableNotFoundException {
- String table = getUniqueNames(1)[0];
- Connector conn = getConnector();
-
- conn.tableOperations().create(table);
-
- ConditionalWriter cw = conn.createConditionalWriter(table, new ConditionalWriterConfig());
-
- ConditionalMutation cm1 = new ConditionalMutation("r1");
- cm1.put("tx", "seq", "1");
- cm1.put("data", "x", "a");
-
- cw.write(cm1);
- }
-
- @Test
- public void testTrace() throws Exception {
- // Need to add a getClientConfig() to AccumuloCluster
- Assume.assumeTrue(getClusterType() == ClusterType.MINI);
- Process tracer = null;
- Connector conn = getConnector();
- AccumuloCluster cluster = getCluster();
- MiniAccumuloClusterImpl mac = (MiniAccumuloClusterImpl) cluster;
- if (!conn.tableOperations().exists("trace")) {
- tracer = mac.exec(TraceServer.class);
- while (!conn.tableOperations().exists("trace")) {
- UtilWaitThread.sleep(1000);
- }
- }
-
- String tableName = getUniqueNames(1)[0];
- conn.tableOperations().create(tableName);
-
- DistributedTrace.enable("localhost", "testTrace", mac.getClientConfig());
- UtilWaitThread.sleep(1000);
- Span root = Trace.on("traceTest");
- ConditionalWriter cw = conn.createConditionalWriter(tableName, new ConditionalWriterConfig());
-
- // mutation conditional on column tx:seq not exiting
- ConditionalMutation cm0 = new ConditionalMutation("99006", new Condition("tx", "seq"));
- cm0.put("name", "last", "doe");
- cm0.put("name", "first", "john");
- cm0.put("tx", "seq", "1");
- Assert.assertEquals(Status.ACCEPTED, cw.write(cm0).getStatus());
- root.stop();
-
- final Scanner scanner = conn.createScanner("trace", Authorizations.EMPTY);
- scanner.setRange(new Range(new Text(Long.toHexString(root.traceId()))));
- loop: while (true) {
- final StringBuffer finalBuffer = new StringBuffer();
- int traceCount = TraceDump.printTrace(scanner, new Printer() {
- @Override
- public void print(final String line) {
- try {
- finalBuffer.append(line).append("\n");
- } catch (Exception ex) {
- throw new RuntimeException(ex);
- }
- }
- });
- String traceOutput = finalBuffer.toString();
- log.info("Trace output:" + traceOutput);
- if (traceCount > 0) {
- int lastPos = 0;
- for (String part : "traceTest, startScan,startConditionalUpdate,conditionalUpdate,Check conditions,apply conditional mutations".split(",")) {
- log.info("Looking in trace output for '" + part + "'");
- int pos = traceOutput.indexOf(part);
- if (-1 == pos) {
- log.info("Trace output doesn't contain '" + part + "'");
- Thread.sleep(1000);
- break loop;
- }
- assertTrue("Did not find '" + part + "' in output", pos > 0);
- assertTrue("'" + part + "' occurred earlier than the previous element unexpectedly", pos > lastPos);
- lastPos = pos;
- }
- break;
- } else {
- log.info("Ignoring trace output as traceCount not greater than zero: " + traceCount);
- Thread.sleep(1000);
- }
- }
- if (tracer != null) {
- tracer.destroy();
- }
- }
-}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/01ae5b85/test/src/test/java/org/apache/accumulo/test/ConfigurableMajorCompactionIT.java
----------------------------------------------------------------------
diff --git a/test/src/test/java/org/apache/accumulo/test/ConfigurableMajorCompactionIT.java b/test/src/test/java/org/apache/accumulo/test/ConfigurableMajorCompactionIT.java
deleted file mode 100644
index bc45dda..0000000
--- a/test/src/test/java/org/apache/accumulo/test/ConfigurableMajorCompactionIT.java
+++ /dev/null
@@ -1,119 +0,0 @@
-/*
- * 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.accumulo.test;
-
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertTrue;
-
-import java.io.IOException;
-import java.util.HashMap;
-import java.util.Map;
-
-import org.apache.accumulo.core.client.BatchWriter;
-import org.apache.accumulo.core.client.BatchWriterConfig;
-import org.apache.accumulo.core.client.Connector;
-import org.apache.accumulo.core.client.Scanner;
-import org.apache.accumulo.core.conf.Property;
-import org.apache.accumulo.core.data.Mutation;
-import org.apache.accumulo.core.metadata.MetadataTable;
-import org.apache.accumulo.core.metadata.schema.MetadataSchema;
-import org.apache.accumulo.core.security.Authorizations;
-import org.apache.accumulo.fate.util.UtilWaitThread;
-import org.apache.accumulo.minicluster.impl.MiniAccumuloConfigImpl;
-import org.apache.accumulo.test.functional.ConfigurableMacBase;
-import org.apache.accumulo.tserver.compaction.CompactionPlan;
-import org.apache.accumulo.tserver.compaction.CompactionStrategy;
-import org.apache.accumulo.tserver.compaction.MajorCompactionRequest;
-import org.apache.accumulo.tserver.compaction.WriteParameters;
-import org.apache.hadoop.conf.Configuration;
-import org.junit.Test;
-
-import com.google.common.collect.Iterators;
-
-public class ConfigurableMajorCompactionIT extends ConfigurableMacBase {
-
- @Override
- public int defaultTimeoutSeconds() {
- return 30;
- }
-
- @Override
- public void configure(MiniAccumuloConfigImpl cfg, Configuration hadoopCoreSite) {
- Map<String,String> siteConfig = new HashMap<String,String>();
- siteConfig.put(Property.TSERV_MAJC_DELAY.getKey(), "1s");
- cfg.setSiteConfig(siteConfig);
- }
-
- public static class TestCompactionStrategy extends CompactionStrategy {
-
- @Override
- public boolean shouldCompact(MajorCompactionRequest request) throws IOException {
- return request.getFiles().size() == 5;
- }
-
- @Override
- public CompactionPlan getCompactionPlan(MajorCompactionRequest request) throws IOException {
- CompactionPlan plan = new CompactionPlan();
- plan.inputFiles.addAll(request.getFiles().keySet());
- plan.writeParameters = new WriteParameters();
- plan.writeParameters.setBlockSize(1024 * 1024);
- plan.writeParameters.setCompressType("none");
- plan.writeParameters.setHdfsBlockSize(1024 * 1024);
- plan.writeParameters.setIndexBlockSize(10);
- plan.writeParameters.setReplication(7);
- return plan;
- }
- }
-
- @Test
- public void test() throws Exception {
- Connector conn = getConnector();
- String tableName = getUniqueNames(1)[0];
- conn.tableOperations().create(tableName);
- conn.tableOperations().setProperty(tableName, Property.TABLE_COMPACTION_STRATEGY.getKey(), TestCompactionStrategy.class.getName());
- writeFile(conn, tableName);
- writeFile(conn, tableName);
- writeFile(conn, tableName);
- writeFile(conn, tableName);
- UtilWaitThread.sleep(2 * 1000);
- assertEquals(4, countFiles(conn));
- writeFile(conn, tableName);
- int count = countFiles(conn);
- assertTrue(count == 1 || count == 5);
- while (count != 1) {
- UtilWaitThread.sleep(250);
- count = countFiles(conn);
- }
- }
-
- private int countFiles(Connector conn) throws Exception {
- Scanner s = conn.createScanner(MetadataTable.NAME, Authorizations.EMPTY);
- s.setRange(MetadataSchema.TabletsSection.getRange());
- s.fetchColumnFamily(MetadataSchema.TabletsSection.DataFileColumnFamily.NAME);
- return Iterators.size(s.iterator());
- }
-
- private void writeFile(Connector conn, String tableName) throws Exception {
- BatchWriter bw = conn.createBatchWriter(tableName, new BatchWriterConfig());
- Mutation m = new Mutation("row");
- m.put("cf", "cq", "value");
- bw.addMutation(m);
- bw.close();
- conn.tableOperations().flush(tableName, null, null, true);
- }
-
-}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/01ae5b85/test/src/test/java/org/apache/accumulo/test/CreateTableWithNewTableConfigIT.java
----------------------------------------------------------------------
diff --git a/test/src/test/java/org/apache/accumulo/test/CreateTableWithNewTableConfigIT.java b/test/src/test/java/org/apache/accumulo/test/CreateTableWithNewTableConfigIT.java
deleted file mode 100644
index b80bcb7..0000000
--- a/test/src/test/java/org/apache/accumulo/test/CreateTableWithNewTableConfigIT.java
+++ /dev/null
@@ -1,193 +0,0 @@
-/*
- * 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.accumulo.test;
-
-import java.util.HashMap;
-import java.util.Map;
-import java.util.Map.Entry;
-
-import org.apache.accumulo.core.client.AccumuloException;
-import org.apache.accumulo.core.client.Connector;
-import org.apache.accumulo.core.client.Scanner;
-import org.apache.accumulo.core.client.TableNotFoundException;
-import org.apache.accumulo.core.client.admin.NewTableConfiguration;
-import org.apache.accumulo.core.client.admin.TimeType;
-import org.apache.accumulo.core.conf.Property;
-import org.apache.accumulo.core.data.Key;
-import org.apache.accumulo.core.data.Value;
-import org.apache.accumulo.core.metadata.MetadataTable;
-import org.apache.accumulo.core.metadata.schema.MetadataSchema.TabletsSection.ServerColumnFamily;
-import org.apache.accumulo.core.security.Authorizations;
-import org.apache.accumulo.harness.SharedMiniClusterBase;
-import org.junit.Assert;
-import org.junit.Test;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-
-import com.google.common.collect.Iterators;
-
-/**
- *
- */
-public class CreateTableWithNewTableConfigIT extends SharedMiniClusterBase {
- static private final Logger log = LoggerFactory.getLogger(CreateTableWithNewTableConfigIT.class);
-
- @Override
- protected int defaultTimeoutSeconds() {
- return 30;
- };
-
- public int numProperties(Connector connector, String tableName) throws AccumuloException, TableNotFoundException {
- return Iterators.size(connector.tableOperations().getProperties(tableName).iterator());
- }
-
- public int compareProperties(Connector connector, String tableNameOrig, String tableName, String changedProp) throws AccumuloException,
- TableNotFoundException {
- boolean inNew = false;
- int countOrig = 0;
- for (Entry<String,String> orig : connector.tableOperations().getProperties(tableNameOrig)) {
- countOrig++;
- for (Entry<String,String> entry : connector.tableOperations().getProperties(tableName)) {
- if (entry.equals(orig)) {
- inNew = true;
- break;
- } else if (entry.getKey().equals(orig.getKey()) && !entry.getKey().equals(changedProp))
- Assert.fail("Property " + orig.getKey() + " has different value than deprecated method");
- }
- if (!inNew)
- Assert.fail("Original property missing after using the new create method");
- }
- return countOrig;
- }
-
- public boolean checkTimeType(Connector connector, String tableName, TimeType expectedTimeType) throws TableNotFoundException {
- final Scanner scanner = connector.createScanner(MetadataTable.NAME, Authorizations.EMPTY);
- String tableID = connector.tableOperations().tableIdMap().get(tableName) + "<";
- for (Entry<Key,Value> entry : scanner) {
- Key k = entry.getKey();
-
- if (k.getRow().toString().equals(tableID) && k.getColumnQualifier().toString().equals(ServerColumnFamily.TIME_COLUMN.getColumnQualifier().toString())) {
- if (expectedTimeType == TimeType.MILLIS && entry.getValue().toString().charAt(0) == 'M')
- return true;
- if (expectedTimeType == TimeType.LOGICAL && entry.getValue().toString().charAt(0) == 'L')
- return true;
- }
- }
- return false;
- }
-
- @SuppressWarnings("deprecation")
- @Test
- public void tableNameOnly() throws Exception {
- log.info("Starting tableNameOnly");
-
- // Create a table with the initial properties
- Connector connector = getConnector();
- String tableName = getUniqueNames(2)[0];
- connector.tableOperations().create(tableName, new NewTableConfiguration());
-
- String tableNameOrig = "original";
- connector.tableOperations().create(tableNameOrig, true);
-
- int countNew = numProperties(connector, tableName);
- int countOrig = compareProperties(connector, tableNameOrig, tableName, null);
-
- Assert.assertEquals("Extra properties using the new create method", countOrig, countNew);
- Assert.assertTrue("Wrong TimeType", checkTimeType(connector, tableName, TimeType.MILLIS));
- }
-
- @SuppressWarnings("deprecation")
- @Test
- public void tableNameAndLimitVersion() throws Exception {
- log.info("Starting tableNameAndLimitVersion");
-
- // Create a table with the initial properties
- Connector connector = getConnector();
- String tableName = getUniqueNames(2)[0];
- boolean limitVersion = false;
- connector.tableOperations().create(tableName, new NewTableConfiguration().withoutDefaultIterators());
-
- String tableNameOrig = "originalWithLimitVersion";
- connector.tableOperations().create(tableNameOrig, limitVersion);
-
- int countNew = numProperties(connector, tableName);
- int countOrig = compareProperties(connector, tableNameOrig, tableName, null);
-
- Assert.assertEquals("Extra properties using the new create method", countOrig, countNew);
- Assert.assertTrue("Wrong TimeType", checkTimeType(connector, tableName, TimeType.MILLIS));
- }
-
- @SuppressWarnings("deprecation")
- @Test
- public void tableNameLimitVersionAndTimeType() throws Exception {
- log.info("Starting tableNameLimitVersionAndTimeType");
-
- // Create a table with the initial properties
- Connector connector = getConnector();
- String tableName = getUniqueNames(2)[0];
- boolean limitVersion = false;
- TimeType tt = TimeType.LOGICAL;
- connector.tableOperations().create(tableName, new NewTableConfiguration().withoutDefaultIterators().setTimeType(tt));
-
- String tableNameOrig = "originalWithLimitVersionAndTimeType";
- connector.tableOperations().create(tableNameOrig, limitVersion, tt);
-
- int countNew = numProperties(connector, tableName);
- int countOrig = compareProperties(connector, tableNameOrig, tableName, null);
-
- Assert.assertEquals("Extra properties using the new create method", countOrig, countNew);
- Assert.assertTrue("Wrong TimeType", checkTimeType(connector, tableName, tt));
- }
-
- @SuppressWarnings("deprecation")
- @Test
- public void addCustomPropAndChangeExisting() throws Exception {
- log.info("Starting addCustomPropAndChangeExisting");
-
- // Create and populate initial properties map for creating table 1
- Map<String,String> properties = new HashMap<String,String>();
- String propertyName = Property.TABLE_SPLIT_THRESHOLD.getKey();
- String volume = "10K";
- properties.put(propertyName, volume);
-
- String propertyName2 = "table.custom.testProp";
- String volume2 = "Test property";
- properties.put(propertyName2, volume2);
-
- // Create a table with the initial properties
- Connector connector = getConnector();
- String tableName = getUniqueNames(2)[0];
- connector.tableOperations().create(tableName, new NewTableConfiguration().setProperties(properties));
-
- String tableNameOrig = "originalWithTableName";
- connector.tableOperations().create(tableNameOrig, true);
-
- int countNew = numProperties(connector, tableName);
- int countOrig = compareProperties(connector, tableNameOrig, tableName, propertyName);
-
- for (Entry<String,String> entry : connector.tableOperations().getProperties(tableName)) {
- if (entry.getKey().equals(Property.TABLE_SPLIT_THRESHOLD.getKey()))
- Assert.assertTrue("TABLE_SPLIT_THRESHOLD has been changed", entry.getValue().equals("10K"));
- if (entry.getKey().equals("table.custom.testProp"))
- Assert.assertTrue("table.custom.testProp has been changed", entry.getValue().equals("Test property"));
- }
-
- Assert.assertEquals("Extra properties using the new create method", countOrig + 1, countNew);
- Assert.assertTrue("Wrong TimeType", checkTimeType(connector, tableName, TimeType.MILLIS));
-
- }
-}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/01ae5b85/test/src/test/java/org/apache/accumulo/test/DumpConfigIT.java
----------------------------------------------------------------------
diff --git a/test/src/test/java/org/apache/accumulo/test/DumpConfigIT.java b/test/src/test/java/org/apache/accumulo/test/DumpConfigIT.java
deleted file mode 100644
index 5cc37a5..0000000
--- a/test/src/test/java/org/apache/accumulo/test/DumpConfigIT.java
+++ /dev/null
@@ -1,69 +0,0 @@
-/*
- * 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.accumulo.test;
-
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertFalse;
-import static org.junit.Assert.assertTrue;
-
-import java.io.File;
-import java.io.FileInputStream;
-import java.util.Collections;
-
-import org.apache.accumulo.core.conf.Property;
-import org.apache.accumulo.core.metadata.MetadataTable;
-import org.apache.accumulo.minicluster.impl.MiniAccumuloConfigImpl;
-import org.apache.accumulo.server.util.Admin;
-import org.apache.accumulo.test.functional.ConfigurableMacBase;
-import org.apache.accumulo.test.functional.FunctionalTestUtils;
-import org.apache.hadoop.conf.Configuration;
-import org.junit.Rule;
-import org.junit.Test;
-import org.junit.rules.TemporaryFolder;
-
-public class DumpConfigIT extends ConfigurableMacBase {
-
- @Rule
- public TemporaryFolder folder = new TemporaryFolder(new File(System.getProperty("user.dir") + "/target"));
-
- @Override
- public int defaultTimeoutSeconds() {
- return 2 * 60;
- }
-
- @Override
- public void configure(MiniAccumuloConfigImpl cfg, Configuration hadoopCoreSite) {
- cfg.setSiteConfig(Collections.singletonMap(Property.TABLE_FILE_BLOCK_SIZE.getKey(), "1234567"));
- }
-
- @Test
- public void test() throws Exception {
- File siteFileBackup = new File(folder.getRoot(), "accumulo-site.xml.bak");
- assertFalse(siteFileBackup.exists());
- assertEquals(0, exec(Admin.class, new String[] {"dumpConfig", "-a", "-d", folder.getRoot().getPath()}).waitFor());
- assertTrue(siteFileBackup.exists());
- String site = FunctionalTestUtils.readAll(new FileInputStream(siteFileBackup));
- assertTrue(site.contains(Property.TABLE_FILE_BLOCK_SIZE.getKey()));
- assertTrue(site.contains("1234567"));
- String meta = FunctionalTestUtils.readAll(new FileInputStream(new File(folder.getRoot(), MetadataTable.NAME + ".cfg")));
- assertTrue(meta.contains(Property.TABLE_FILE_REPLICATION.getKey()));
- String systemPerm = FunctionalTestUtils.readAll(new FileInputStream(new File(folder.getRoot(), "root_user.cfg")));
- assertTrue(systemPerm.contains("grant System.ALTER_USER -s -u root"));
- assertTrue(systemPerm.contains("grant Table.READ -t " + MetadataTable.NAME + " -u root"));
- assertFalse(systemPerm.contains("grant Table.DROP -t " + MetadataTable.NAME + " -u root"));
- }
-}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/01ae5b85/test/src/test/java/org/apache/accumulo/test/ExistingMacIT.java
----------------------------------------------------------------------
diff --git a/test/src/test/java/org/apache/accumulo/test/ExistingMacIT.java b/test/src/test/java/org/apache/accumulo/test/ExistingMacIT.java
deleted file mode 100644
index 52d2086..0000000
--- a/test/src/test/java/org/apache/accumulo/test/ExistingMacIT.java
+++ /dev/null
@@ -1,169 +0,0 @@
-/*
- * 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.accumulo.test;
-
-import static org.junit.Assert.assertTrue;
-
-import java.io.File;
-import java.io.FileOutputStream;
-import java.io.IOException;
-import java.io.OutputStream;
-import java.util.Collection;
-import java.util.Map.Entry;
-import java.util.Set;
-
-import org.apache.accumulo.core.client.BatchWriter;
-import org.apache.accumulo.core.client.BatchWriterConfig;
-import org.apache.accumulo.core.client.Connector;
-import org.apache.accumulo.core.client.Scanner;
-import org.apache.accumulo.core.client.security.tokens.PasswordToken;
-import org.apache.accumulo.core.conf.Property;
-import org.apache.accumulo.core.data.Key;
-import org.apache.accumulo.core.data.Mutation;
-import org.apache.accumulo.core.data.Value;
-import org.apache.accumulo.core.metadata.MetadataTable;
-import org.apache.accumulo.core.metadata.RootTable;
-import org.apache.accumulo.core.security.Authorizations;
-import org.apache.accumulo.core.util.UtilWaitThread;
-import org.apache.accumulo.minicluster.ServerType;
-import org.apache.accumulo.minicluster.impl.MiniAccumuloClusterImpl;
-import org.apache.accumulo.minicluster.impl.MiniAccumuloConfigImpl;
-import org.apache.accumulo.minicluster.impl.ProcessReference;
-import org.apache.accumulo.test.functional.ConfigurableMacBase;
-import org.apache.commons.io.FileUtils;
-import org.apache.hadoop.conf.Configuration;
-import org.apache.hadoop.fs.RawLocalFileSystem;
-import org.junit.Assert;
-import org.junit.Test;
-
-public class ExistingMacIT extends ConfigurableMacBase {
- @Override
- public int defaultTimeoutSeconds() {
- return 2 * 60;
- }
-
- @Override
- public void configure(MiniAccumuloConfigImpl cfg, Configuration hadoopCoreSite) {
- cfg.setProperty(Property.INSTANCE_ZK_TIMEOUT, "5s");
-
- // use raw local file system so walogs sync and flush will work
- hadoopCoreSite.set("fs.file.impl", RawLocalFileSystem.class.getName());
- }
-
- private void createEmptyConfig(File confFile) throws IOException {
- Configuration conf = new Configuration(false);
- OutputStream hcOut = new FileOutputStream(confFile);
- conf.writeXml(hcOut);
- hcOut.close();
- }
-
- @Test
- public void testExistingInstance() throws Exception {
-
- Connector conn = getCluster().getConnector("root", new PasswordToken(ROOT_PASSWORD));
-
- conn.tableOperations().create("table1");
-
- BatchWriter bw = conn.createBatchWriter("table1", new BatchWriterConfig());
-
- Mutation m1 = new Mutation("00081");
- m1.put("math", "sqroot", "9");
- m1.put("math", "sq", "6560");
-
- bw.addMutation(m1);
- bw.close();
-
- conn.tableOperations().flush("table1", null, null, true);
- // TOOD use constants
- conn.tableOperations().flush(MetadataTable.NAME, null, null, true);
- conn.tableOperations().flush(RootTable.NAME, null, null, true);
-
- Set<Entry<ServerType,Collection<ProcessReference>>> procs = getCluster().getProcesses().entrySet();
- for (Entry<ServerType,Collection<ProcessReference>> entry : procs) {
- if (entry.getKey() == ServerType.ZOOKEEPER)
- continue;
- for (ProcessReference pr : entry.getValue())
- getCluster().killProcess(entry.getKey(), pr);
- }
-
- // TODO clean out zookeeper? following sleep waits for ephemeral nodes to go away
- UtilWaitThread.sleep(10000);
-
- File hadoopConfDir = createTestDir(ExistingMacIT.class.getSimpleName() + "_hadoop_conf");
- FileUtils.deleteQuietly(hadoopConfDir);
- assertTrue(hadoopConfDir.mkdirs());
- createEmptyConfig(new File(hadoopConfDir, "core-site.xml"));
- createEmptyConfig(new File(hadoopConfDir, "hdfs-site.xml"));
-
- File testDir2 = createTestDir(ExistingMacIT.class.getSimpleName() + "_2");
- FileUtils.deleteQuietly(testDir2);
-
- MiniAccumuloConfigImpl macConfig2 = new MiniAccumuloConfigImpl(testDir2, "notused");
- macConfig2.useExistingInstance(new File(getCluster().getConfig().getConfDir(), "accumulo-site.xml"), hadoopConfDir);
-
- MiniAccumuloClusterImpl accumulo2 = new MiniAccumuloClusterImpl(macConfig2);
- accumulo2.start();
-
- conn = accumulo2.getConnector("root", new PasswordToken(ROOT_PASSWORD));
-
- Scanner scanner = conn.createScanner("table1", Authorizations.EMPTY);
-
- int sum = 0;
- for (Entry<Key,Value> entry : scanner) {
- sum += Integer.parseInt(entry.getValue().toString());
- }
-
- Assert.assertEquals(6569, sum);
-
- accumulo2.stop();
- }
-
- @Test
- public void testExistingRunningInstance() throws Exception {
- final String table = getUniqueNames(1)[0];
- Connector conn = getConnector();
- // Ensure that a master and tserver are up so the existing instance check won't fail.
- conn.tableOperations().create(table);
- BatchWriter bw = conn.createBatchWriter(table, new BatchWriterConfig());
- Mutation m = new Mutation("foo");
- m.put("cf", "cq", "value");
- bw.addMutation(m);
- bw.close();
-
- File hadoopConfDir = createTestDir(ExistingMacIT.class.getSimpleName() + "_hadoop_conf_2");
- FileUtils.deleteQuietly(hadoopConfDir);
- assertTrue(hadoopConfDir.mkdirs());
- createEmptyConfig(new File(hadoopConfDir, "core-site.xml"));
- createEmptyConfig(new File(hadoopConfDir, "hdfs-site.xml"));
-
- File testDir2 = createTestDir(ExistingMacIT.class.getSimpleName() + "_3");
- FileUtils.deleteQuietly(testDir2);
-
- MiniAccumuloConfigImpl macConfig2 = new MiniAccumuloConfigImpl(testDir2, "notused");
- macConfig2.useExistingInstance(new File(getCluster().getConfig().getConfDir(), "accumulo-site.xml"), hadoopConfDir);
-
- System.out.println("conf " + new File(getCluster().getConfig().getConfDir(), "accumulo-site.xml"));
-
- MiniAccumuloClusterImpl accumulo2 = new MiniAccumuloClusterImpl(macConfig2);
- try {
- accumulo2.start();
- Assert.fail("A 2nd MAC instance should not be able to start over an existing MAC instance");
- } catch (RuntimeException e) {
- // TODO check message or throw more explicit exception
- }
- }
-}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/01ae5b85/test/src/test/java/org/apache/accumulo/test/FileArchiveIT.java
----------------------------------------------------------------------
diff --git a/test/src/test/java/org/apache/accumulo/test/FileArchiveIT.java b/test/src/test/java/org/apache/accumulo/test/FileArchiveIT.java
deleted file mode 100644
index 8e51984..0000000
--- a/test/src/test/java/org/apache/accumulo/test/FileArchiveIT.java
+++ /dev/null
@@ -1,271 +0,0 @@
-/*
- * 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.accumulo.test;
-
-import java.util.Map.Entry;
-
-import org.apache.accumulo.core.client.BatchWriter;
-import org.apache.accumulo.core.client.BatchWriterConfig;
-import org.apache.accumulo.core.client.Connector;
-import org.apache.accumulo.core.client.Scanner;
-import org.apache.accumulo.core.conf.Property;
-import org.apache.accumulo.core.data.Key;
-import org.apache.accumulo.core.data.Mutation;
-import org.apache.accumulo.core.data.Value;
-import org.apache.accumulo.core.metadata.MetadataTable;
-import org.apache.accumulo.core.metadata.schema.MetadataSchema;
-import org.apache.accumulo.core.security.Authorizations;
-import org.apache.accumulo.minicluster.impl.MiniAccumuloConfigImpl;
-import org.apache.accumulo.server.ServerConstants;
-import org.apache.accumulo.test.functional.ConfigurableMacBase;
-import org.apache.hadoop.conf.Configuration;
-import org.apache.hadoop.fs.FileSystem;
-import org.apache.hadoop.fs.Path;
-import org.junit.Assert;
-import org.junit.Test;
-
-import com.google.common.collect.Iterables;
-
-/**
- * Tests that files are archived instead of deleted when configured.
- */
-public class FileArchiveIT extends ConfigurableMacBase {
-
- @Override
- public int defaultTimeoutSeconds() {
- return 2 * 60;
- }
-
- @Override
- public void configure(MiniAccumuloConfigImpl cfg, Configuration coreSite) {
- cfg.setProperty(Property.GC_FILE_ARCHIVE, "true");
- cfg.setProperty(Property.GC_CYCLE_DELAY, "1s");
- cfg.setProperty(Property.GC_CYCLE_START, "1s");
- }
-
- @Test
- public void testUnusuedFilesAreArchived() throws Exception {
- final Connector conn = getConnector();
- final String tableName = getUniqueNames(1)[0];
-
- conn.tableOperations().create(tableName);
-
- final String tableId = conn.tableOperations().tableIdMap().get(tableName);
- Assert.assertNotNull("Could not get table ID", tableId);
-
- BatchWriter bw = conn.createBatchWriter(tableName, new BatchWriterConfig());
- Mutation m = new Mutation("row");
- m.put("", "", "value");
- bw.addMutation(m);
- bw.close();
-
- // Compact memory to disk
- conn.tableOperations().compact(tableName, null, null, true, true);
-
- Scanner s = conn.createScanner(MetadataTable.NAME, Authorizations.EMPTY);
- s.setRange(MetadataSchema.TabletsSection.getRange(tableId));
- s.fetchColumnFamily(MetadataSchema.TabletsSection.DataFileColumnFamily.NAME);
-
- Entry<Key,Value> entry = Iterables.getOnlyElement(s);
- final String file = entry.getKey().getColumnQualifier().toString();
- final Path p = new Path(file);
-
- // Then force another to make an unreferenced file
- conn.tableOperations().compact(tableName, null, null, true, true);
-
- log.info("File for table: " + file);
-
- FileSystem fs = getCluster().getFileSystem();
- int i = 0;
- while (fs.exists(p)) {
- i++;
- Thread.sleep(1000);
- if (0 == i % 10) {
- log.info("Waited " + i + " iterations, file still exists");
- }
- }
-
- log.info("File was removed");
-
- String filePath = p.toUri().getPath().substring(getCluster().getConfig().getAccumuloDir().toString().length());
-
- log.info("File relative to accumulo dir: " + filePath);
-
- Path fileArchiveDir = new Path(getCluster().getConfig().getAccumuloDir().toString(), ServerConstants.FILE_ARCHIVE_DIR);
-
- Assert.assertTrue("File archive directory didn't exist", fs.exists(fileArchiveDir));
-
- // Remove the leading '/' to make sure Path treats the 2nd arg as a child.
- Path archivedFile = new Path(fileArchiveDir, filePath.substring(1));
-
- Assert.assertTrue("File doesn't exists in archive directory: " + archivedFile, fs.exists(archivedFile));
- }
-
- @Test
- public void testDeletedTableIsArchived() throws Exception {
- final Connector conn = getConnector();
- final String tableName = getUniqueNames(1)[0];
-
- conn.tableOperations().create(tableName);
-
- final String tableId = conn.tableOperations().tableIdMap().get(tableName);
- Assert.assertNotNull("Could not get table ID", tableId);
-
- BatchWriter bw = conn.createBatchWriter(tableName, new BatchWriterConfig());
- Mutation m = new Mutation("row");
- m.put("", "", "value");
- bw.addMutation(m);
- bw.close();
-
- // Compact memory to disk
- conn.tableOperations().compact(tableName, null, null, true, true);
-
- Scanner s = conn.createScanner(MetadataTable.NAME, Authorizations.EMPTY);
- s.setRange(MetadataSchema.TabletsSection.getRange(tableId));
- s.fetchColumnFamily(MetadataSchema.TabletsSection.DataFileColumnFamily.NAME);
-
- Entry<Key,Value> entry = Iterables.getOnlyElement(s);
- final String file = entry.getKey().getColumnQualifier().toString();
- final Path p = new Path(file);
-
- conn.tableOperations().delete(tableName);
-
- log.info("File for table: " + file);
-
- FileSystem fs = getCluster().getFileSystem();
- int i = 0;
- while (fs.exists(p)) {
- i++;
- Thread.sleep(1000);
- if (0 == i % 10) {
- log.info("Waited " + i + " iterations, file still exists");
- }
- }
-
- log.info("File was removed");
-
- String filePath = p.toUri().getPath().substring(getCluster().getConfig().getAccumuloDir().toString().length());
-
- log.info("File relative to accumulo dir: " + filePath);
-
- Path fileArchiveDir = new Path(getCluster().getConfig().getAccumuloDir().toString(), ServerConstants.FILE_ARCHIVE_DIR);
-
- Assert.assertTrue("File archive directory didn't exist", fs.exists(fileArchiveDir));
-
- // Remove the leading '/' to make sure Path treats the 2nd arg as a child.
- Path archivedFile = new Path(fileArchiveDir, filePath.substring(1));
-
- Assert.assertTrue("File doesn't exists in archive directory: " + archivedFile, fs.exists(archivedFile));
- }
-
- @Test
- public void testUnusuedFilesAndDeletedTable() throws Exception {
- final Connector conn = getConnector();
- final String tableName = getUniqueNames(1)[0];
-
- conn.tableOperations().create(tableName);
-
- final String tableId = conn.tableOperations().tableIdMap().get(tableName);
- Assert.assertNotNull("Could not get table ID", tableId);
-
- BatchWriter bw = conn.createBatchWriter(tableName, new BatchWriterConfig());
- Mutation m = new Mutation("row");
- m.put("", "", "value");
- bw.addMutation(m);
- bw.close();
-
- // Compact memory to disk
- conn.tableOperations().compact(tableName, null, null, true, true);
-
- Scanner s = conn.createScanner(MetadataTable.NAME, Authorizations.EMPTY);
- s.setRange(MetadataSchema.TabletsSection.getRange(tableId));
- s.fetchColumnFamily(MetadataSchema.TabletsSection.DataFileColumnFamily.NAME);
-
- Entry<Key,Value> entry = Iterables.getOnlyElement(s);
- final String file = entry.getKey().getColumnQualifier().toString();
- final Path p = new Path(file);
-
- // Then force another to make an unreferenced file
- conn.tableOperations().compact(tableName, null, null, true, true);
-
- log.info("File for table: " + file);
-
- FileSystem fs = getCluster().getFileSystem();
- int i = 0;
- while (fs.exists(p)) {
- i++;
- Thread.sleep(1000);
- if (0 == i % 10) {
- log.info("Waited " + i + " iterations, file still exists");
- }
- }
-
- log.info("File was removed");
-
- String filePath = p.toUri().getPath().substring(getCluster().getConfig().getAccumuloDir().toString().length());
-
- log.info("File relative to accumulo dir: " + filePath);
-
- Path fileArchiveDir = new Path(getCluster().getConfig().getAccumuloDir().toString(), ServerConstants.FILE_ARCHIVE_DIR);
-
- Assert.assertTrue("File archive directory didn't exist", fs.exists(fileArchiveDir));
-
- // Remove the leading '/' to make sure Path treats the 2nd arg as a child.
- Path archivedFile = new Path(fileArchiveDir, filePath.substring(1));
-
- Assert.assertTrue("File doesn't exists in archive directory: " + archivedFile, fs.exists(archivedFile));
-
- // Offline the table so we can be sure there is a single file
- conn.tableOperations().offline(tableName, true);
-
- // See that the file in metadata currently is
- s = conn.createScanner(MetadataTable.NAME, Authorizations.EMPTY);
- s.setRange(MetadataSchema.TabletsSection.getRange(tableId));
- s.fetchColumnFamily(MetadataSchema.TabletsSection.DataFileColumnFamily.NAME);
-
- entry = Iterables.getOnlyElement(s);
- final String finalFile = entry.getKey().getColumnQualifier().toString();
- final Path finalPath = new Path(finalFile);
-
- conn.tableOperations().delete(tableName);
-
- log.info("File for table: " + finalPath);
-
- i = 0;
- while (fs.exists(finalPath)) {
- i++;
- Thread.sleep(1000);
- if (0 == i % 10) {
- log.info("Waited " + i + " iterations, file still exists");
- }
- }
-
- log.info("File was removed");
-
- String finalFilePath = finalPath.toUri().getPath().substring(getCluster().getConfig().getAccumuloDir().toString().length());
-
- log.info("File relative to accumulo dir: " + finalFilePath);
-
- Assert.assertTrue("File archive directory didn't exist", fs.exists(fileArchiveDir));
-
- // Remove the leading '/' to make sure Path treats the 2nd arg as a child.
- Path finalArchivedFile = new Path(fileArchiveDir, finalFilePath.substring(1));
-
- Assert.assertTrue("File doesn't exists in archive directory: " + finalArchivedFile, fs.exists(finalArchivedFile));
- }
-
-}