ceph/src/arrow/java/tools/src/test/java/org/apache/arrow/tools/TestFileRoundtrip.java

   1 /*
   2  * Licensed to the Apache Software Foundation (ASF) under one or more
   3  * contributor license agreements.  See the NOTICE file distributed with
   4  * this work for additional information regarding copyright ownership.
   5  * The ASF licenses this file to You under the Apache License, Version 2.0
   6  * (the "License"); you may not use this file except in compliance with
   7  * the License.  You may obtain a copy of the License at
   8  *
   9  *    http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17
  18 package org.apache.arrow.tools;
  19
  20 import static org.apache.arrow.tools.ArrowFileTestFixtures.validateOutput;
  21 import static org.apache.arrow.tools.ArrowFileTestFixtures.writeInput;
  22 import static org.junit.Assert.assertEquals;
  23
  24 import java.io.File;
  25
  26 import org.apache.arrow.memory.BufferAllocator;
  27 import org.apache.arrow.memory.RootAllocator;
  28 import org.junit.After;
  29 import org.junit.Before;
  30 import org.junit.Rule;
  31 import org.junit.Test;
  32 import org.junit.rules.TemporaryFolder;
  33
  34 public class TestFileRoundtrip {
  35
  36   @Rule
  37   public TemporaryFolder testFolder = new TemporaryFolder();
  38
  39   private BufferAllocator allocator;
  40
  41   @Before
  42   public void init() {
  43     allocator = new RootAllocator(Integer.MAX_VALUE);
  44   }
  45
  46   @After
  47   public void tearDown() {
  48     allocator.close();
  49   }
  50
  51   @Test
  52   public void test() throws Exception {
  53     File testInFile = testFolder.newFile("testIn.arrow");
  54     File testOutFile = testFolder.newFile("testOut.arrow");
  55
  56     writeInput(testInFile, allocator);
  57
  58     String[] args = {"-i", testInFile.getAbsolutePath(), "-o", testOutFile.getAbsolutePath()};
  59     int result = new FileRoundtrip(System.out, System.err).run(args);
  60     assertEquals(0, result);
  61
  62     validateOutput(testOutFile, allocator);
  63   }
  64
  65 }