cassandra-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From brandonwilli...@apache.org
Subject [03/10] git commit: add missed test
Date Wed, 13 Aug 2014 01:11:11 GMT
add missed test


Project: http://git-wip-us.apache.org/repos/asf/cassandra/repo
Commit: http://git-wip-us.apache.org/repos/asf/cassandra/commit/67889d6a
Tree: http://git-wip-us.apache.org/repos/asf/cassandra/tree/67889d6a
Diff: http://git-wip-us.apache.org/repos/asf/cassandra/diff/67889d6a

Branch: refs/heads/cassandra-2.1
Commit: 67889d6a639e09b76e0c07216223134a868ab544
Parents: 7d16921
Author: Brandon Williams <brandonwilliams@apache.org>
Authored: Tue Aug 12 20:10:07 2014 -0500
Committer: Brandon Williams <brandonwilliams@apache.org>
Committed: Tue Aug 12 20:10:07 2014 -0500

----------------------------------------------------------------------
 .../cassandra/pig/CqlRecordReaderTest.java      | 132 +++++++++++++++++++
 1 file changed, 132 insertions(+)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/cassandra/blob/67889d6a/test/pig/org/apache/cassandra/pig/CqlRecordReaderTest.java
----------------------------------------------------------------------
diff --git a/test/pig/org/apache/cassandra/pig/CqlRecordReaderTest.java b/test/pig/org/apache/cassandra/pig/CqlRecordReaderTest.java
new file mode 100644
index 0000000..fdb5107
--- /dev/null
+++ b/test/pig/org/apache/cassandra/pig/CqlRecordReaderTest.java
@@ -0,0 +1,132 @@
+/*
+ * 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.cassandra.pig;
+
+import java.io.IOException;
+import java.nio.charset.CharacterCodingException;
+import java.util.Iterator;
+
+import org.junit.Assert;
+import org.junit.BeforeClass;
+import org.junit.Ignore;
+import org.junit.Test;
+
+import org.apache.cassandra.db.marshal.TimeUUIDType;
+import org.apache.cassandra.db.marshal.UUIDType;
+import org.apache.cassandra.exceptions.ConfigurationException;
+import org.apache.cassandra.thrift.AuthenticationException;
+import org.apache.cassandra.thrift.AuthorizationException;
+import org.apache.cassandra.thrift.InvalidRequestException;
+import org.apache.cassandra.thrift.NotFoundException;
+import org.apache.cassandra.thrift.TimedOutException;
+import org.apache.cassandra.thrift.UnavailableException;
+import org.apache.cassandra.utils.Hex;
+import org.apache.pig.data.DataByteArray;
+import org.apache.pig.data.Tuple;
+import org.apache.thrift.TException;
+import org.apache.thrift.transport.TTransportException;
+
+public class CqlRecordReaderTest extends PigTestBase
+{
+    private static String[] statements = {
+        "CREATE KEYSPACE cql3ks WITH replication = {'class': 'SimpleStrategy', 'replication_factor'
: 1}",
+        "USE cql3ks;",
+
+        "CREATE TABLE cqltable (" +
+        "pk1 int," +
+        "pk2 int," +
+        "pk3 int," +
+        "ck1 int," +
+        "ck2 int," +
+        "data text," +
+        "primary key((pk1,pk2,pk3),ck1,ck2));",
+        "INSERT INTO cqltable(pk1, pk2, pk3, ck1, ck2, data) VALUES (11, 12, 13, 14, 15,
'value1');",
+    };
+
+    @BeforeClass
+    public static void setup() throws TTransportException, IOException, InterruptedException,
ConfigurationException,
+    AuthenticationException, AuthorizationException, InvalidRequestException, UnavailableException,
TimedOutException, TException, NotFoundException, CharacterCodingException, ClassNotFoundException,
NoSuchFieldException, IllegalAccessException, InstantiationException
+    {
+        startCassandra();
+        setupDataByCql(statements);
+        startHadoopCluster();
+    }
+
+    @Test
+    public void defaultCqlQueryTest() throws Exception
+    {
+        String initialQuery = "rows = LOAD 'cql://cql3ks/cqltable?" + defaultParameters +
nativeParameters + "' USING CqlNativeStorage();";
+        pig.registerQuery(initialQuery);
+        Iterator<Tuple> it = pig.openIterator("rows");
+        if (it.hasNext()) {
+            Tuple t = it.next();
+            Assert.assertEquals(t.get(0), 11);
+            Assert.assertEquals(t.get(1), 12);
+            Assert.assertEquals(t.get(2), 13);
+            Assert.assertEquals(t.get(3), 14);
+            Assert.assertEquals(t.get(4), 15);
+            Assert.assertEquals(t.get(5), "value1");
+        }
+        else
+        {
+            Assert.fail("Failed to get data for query " + initialQuery);
+        }
+    }
+
+    @Test
+    public void selectColumnsTest() throws Exception
+    {
+        String initialQuery = "rows = LOAD 'cql://cql3ks/cqltable?" + defaultParameters +
nativeParameters + "&columns=ck1%2Cck2%2Cdata' USING CqlNativeStorage();";
+        pig.registerQuery(initialQuery);
+        Iterator<Tuple> it = pig.openIterator("rows");
+        if (it.hasNext()) {
+            Tuple t = it.next();
+            Assert.assertEquals(t.get(0), 11);
+            Assert.assertEquals(t.get(1), 12);
+            Assert.assertEquals(t.get(2), 13);
+            Assert.assertEquals(t.get(3), 14);
+            Assert.assertEquals(t.get(4), 15);
+            Assert.assertEquals(t.get(5), "value1");
+        }
+        else
+        {
+            Assert.fail("Failed to get data for query " + initialQuery);
+        }
+    }
+
+    @Test
+    public void whereClauseTest() throws Exception
+    {
+        String initialQuery = "rows = LOAD 'cql://cql3ks/cqltable?" + defaultParameters +
nativeParameters + "&where_clause=ck1%3d14' USING CqlNativeStorage();";
+        pig.registerQuery(initialQuery);
+        Iterator<Tuple> it = pig.openIterator("rows");
+        if (it.hasNext()) {
+            Tuple t = it.next();
+            Assert.assertEquals(t.get(0), 11);
+            Assert.assertEquals(t.get(1), 12);
+            Assert.assertEquals(t.get(2), 13);
+            Assert.assertEquals(t.get(3), 14);
+            Assert.assertEquals(t.get(4), 15);
+            Assert.assertEquals(t.get(5), "value1");
+        }
+        else
+        {
+            Assert.fail("Failed to get data for query " + initialQuery);
+        }
+    }
+}


Mime
View raw message