Пример #1
0
 /** To show that it does not have any type specific code */
 private void pickTest(byte t, boolean[] inner) throws ExecException, IOException {
   Random r = new Random();
   switch (t) {
     case DataType.BAG:
       runTest(GenRandomData.genRandSmallTupDataBag(r, 10, 100), inner, DataType.BAG);
       break;
     case DataType.BOOLEAN:
       runTest(r.nextBoolean(), inner, DataType.BOOLEAN);
       break;
     case DataType.BYTEARRAY:
       runTest(GenRandomData.genRandDBA(r), inner, DataType.BYTEARRAY);
       break;
     case DataType.BIGCHARARRAY:
       {
         String s = GenRandomData.genRandString(r);
         for (; s.length() < 65535; ) {
           s += GenRandomData.genRandString(r);
         }
         runTest(s, inner, DataType.CHARARRAY);
         break;
       }
     case DataType.CHARARRAY:
       runTest(GenRandomData.genRandString(r), inner, DataType.CHARARRAY);
       break;
     case DataType.DOUBLE:
       runTest(r.nextDouble(), inner, DataType.DOUBLE);
       break;
     case DataType.FLOAT:
       runTest(r.nextFloat(), inner, DataType.FLOAT);
       break;
     case DataType.INTEGER:
       runTest(r.nextInt(), inner, DataType.INTEGER);
       break;
     case DataType.LONG:
       runTest(r.nextLong(), inner, DataType.LONG);
       break;
     case DataType.DATETIME:
       runTest(new DateTime(r.nextLong()), inner, DataType.DATETIME);
       break;
     case DataType.MAP:
     case DataType.INTERNALMAP:
     case DataType.BYTE:
       return; // map not key type
     case DataType.TUPLE:
       runTest(GenRandomData.genRandSmallBagTuple(r, 10, 100), inner, DataType.TUPLE);
       break;
     case DataType.BIGINTEGER:
       runTest(new BigInteger(256, r), inner, DataType.BIGINTEGER);
       break;
     case DataType.BIGDECIMAL:
       runTest(new BigDecimal(r.nextDouble()), inner, DataType.BIGDECIMAL);
       break;
     default:
       fail("No test case for type " + DataType.findTypeName(t));
   }
 }
Пример #2
0
  private void runTest(Object key, boolean inner[], byte keyType)
      throws ExecException, IOException {
    Random r = new Random();
    DataBag db1 = GenRandomData.genRandSmallTupDataBag(r, 10, 100);
    DataBag db2 = GenRandomData.genRandSmallTupDataBag(r, 10, 100);
    List<NullableTuple> db = new ArrayList<NullableTuple>(200);
    Iterator<Tuple> db1Iter = db1.iterator();
    if (!inner[0]) {
      while (db1Iter.hasNext()) {
        NullableTuple it = new NullableTuple(db1Iter.next());
        it.setIndex((byte) 0);
        db.add(it);
      }
    }
    Iterator<Tuple> db2Iter = db2.iterator();
    while (db2Iter.hasNext()) {
      NullableTuple it = new NullableTuple(db2Iter.next());
      it.setIndex((byte) 1);
      db.add(it);
    }
    // ITIterator iti = new TestPackage.ITIterator(db.iterator());
    POPackage pop = new POPackage(new OperatorKey("", r.nextLong()));
    pop.setNumInps(2);
    pop.getPkgr().setInner(inner);
    PigNullableWritable k = HDataType.getWritableComparableTypes(key, keyType);
    pop.attachInput(k, db.iterator());
    if (keyType != DataType.BAG) {
      // test serialization
      NullablePartitionWritable wr;
      if (keyType == DataType.TUPLE) {
        BinSedesTuple tup =
            (BinSedesTuple) binfactory.newTupleNoCopy(((Tuple) k.getValueAsPigType()).getAll());
        wr = new NullablePartitionWritable(new NullableTuple(tup));
      } else {
        wr = new NullablePartitionWritable(k);
      }
      ByteArrayOutputStream baos = new ByteArrayOutputStream();
      DataOutputStream out = new DataOutputStream(baos);
      wr.write(out);
      byte[] arr = baos.toByteArray();
      ByteArrayInputStream bais = new ByteArrayInputStream(arr);
      DataInputStream in = new DataInputStream(bais);
      NullablePartitionWritable re = new NullablePartitionWritable();
      re.readFields(in);
      assertEquals(re, wr);
    }

    // we are not doing any optimization to remove
    // parts of the "value" which are present in the "key" in this
    // unit test - so set up the "keyInfo" accordingly in
    // the POPackage
    Map<Integer, Pair<Boolean, Map<Integer, Integer>>> keyInfo =
        new HashMap<Integer, Pair<Boolean, Map<Integer, Integer>>>();
    Pair<Boolean, Map<Integer, Integer>> p =
        new Pair<Boolean, Map<Integer, Integer>>(false, new HashMap<Integer, Integer>());
    keyInfo.put(0, p);
    keyInfo.put(1, p);
    pop.getPkgr().setKeyInfo(keyInfo);
    Tuple t = null;
    Result res = null;
    res = pop.getNextTuple();
    if (res.returnStatus == POStatus.STATUS_NULL && inner[0]) return;
    assertEquals(POStatus.STATUS_OK, res.returnStatus);

    t = (Tuple) res.result;
    Object outKey = t.get(0);
    DataBag outDb1 = (DataBag) t.get(1);
    DataBag outDb2 = (DataBag) t.get(2);

    assertEquals(key, outKey);
    assertTrue(TestHelper.compareBags(db1, outDb1));
    assertTrue(TestHelper.compareBags(db2, outDb2));
  }