Search in sources :

Example 1 with SESSION

use of com.facebook.presto.testing.TestingConnectorSession.SESSION in project presto by prestodb.

the class TestLocalFileRecordSet method assertData.

private static void assertData(LocalFileTables localFileTables, LocalFileMetadata metadata) {
    SchemaTableName tableName = getSchemaTableName();
    List<LocalFileColumnHandle> columnHandles = metadata.getColumnHandles(SESSION, new LocalFileTableHandle(tableName, OptionalInt.of(0), OptionalInt.of(-1))).values().stream().map(column -> (LocalFileColumnHandle) column).collect(Collectors.toList());
    LocalFileRecordSet recordSet = new LocalFileRecordSet(localFileTables, new LocalFileSplit(address, tableName, TupleDomain.all()), columnHandles);
    RecordCursor cursor = recordSet.cursor();
    for (int i = 0; i < columnHandles.size(); i++) {
        assertEquals(cursor.getType(i), columnHandles.get(i).getColumnType());
    }
    // test one row
    assertTrue(cursor.advanceNextPosition());
    assertEquals(cursor.getSlice(0).toStringUtf8(), address.toString());
    assertEquals(cursor.getSlice(2).toStringUtf8(), "127.0.0.1");
    assertEquals(cursor.getSlice(3).toStringUtf8(), "POST");
    assertEquals(cursor.getSlice(4).toStringUtf8(), "/v1/memory");
    assertTrue(cursor.isNull(5));
    assertTrue(cursor.isNull(6));
    assertEquals(cursor.getLong(7), 200);
    assertEquals(cursor.getLong(8), 0);
    assertEquals(cursor.getLong(9), 1000);
    assertEquals(cursor.getLong(10), 10);
    assertTrue(cursor.isNull(11));
    assertTrue(cursor.advanceNextPosition());
    assertEquals(cursor.getSlice(0).toStringUtf8(), address.toString());
    assertEquals(cursor.getSlice(2).toStringUtf8(), "127.0.0.1");
    assertEquals(cursor.getSlice(3).toStringUtf8(), "GET");
    assertEquals(cursor.getSlice(4).toStringUtf8(), "/v1/service/presto/general");
    assertEquals(cursor.getSlice(5).toStringUtf8(), "foo");
    assertEquals(cursor.getSlice(6).toStringUtf8(), "ffffffff-ffff-ffff-ffff-ffffffffffff");
    assertEquals(cursor.getLong(7), 200);
    assertEquals(cursor.getLong(8), 0);
    assertEquals(cursor.getLong(9), 37);
    assertEquals(cursor.getLong(10), 1094);
    assertEquals(cursor.getSlice(11).toStringUtf8(), "a7229d56-5cbd-4e23-81ff-312ba6be0f12");
}
Also used : TupleDomain(com.facebook.presto.spi.predicate.TupleDomain) RecordCursor(com.facebook.presto.spi.RecordCursor) List(java.util.List) SchemaTableName(com.facebook.presto.spi.SchemaTableName) HttpRequestLogTable.getSchemaTableName(com.facebook.presto.localfile.LocalFileTables.HttpRequestLogTable.getSchemaTableName) SESSION(com.facebook.presto.testing.TestingConnectorSession.SESSION) HostAddress(com.facebook.presto.spi.HostAddress) Assert.assertEquals(org.testng.Assert.assertEquals) Assert.assertTrue(org.testng.Assert.assertTrue) Test(org.testng.annotations.Test) OptionalInt(java.util.OptionalInt) Collectors(java.util.stream.Collectors) RecordCursor(com.facebook.presto.spi.RecordCursor) SchemaTableName(com.facebook.presto.spi.SchemaTableName) HttpRequestLogTable.getSchemaTableName(com.facebook.presto.localfile.LocalFileTables.HttpRequestLogTable.getSchemaTableName)

Aggregations

HttpRequestLogTable.getSchemaTableName (com.facebook.presto.localfile.LocalFileTables.HttpRequestLogTable.getSchemaTableName)1 HostAddress (com.facebook.presto.spi.HostAddress)1 RecordCursor (com.facebook.presto.spi.RecordCursor)1 SchemaTableName (com.facebook.presto.spi.SchemaTableName)1 TupleDomain (com.facebook.presto.spi.predicate.TupleDomain)1 SESSION (com.facebook.presto.testing.TestingConnectorSession.SESSION)1 List (java.util.List)1 OptionalInt (java.util.OptionalInt)1 Collectors (java.util.stream.Collectors)1 Assert.assertEquals (org.testng.Assert.assertEquals)1 Assert.assertTrue (org.testng.Assert.assertTrue)1 Test (org.testng.annotations.Test)1