Examples with TypeSignature io.crate.types.TypeSignature used on opensource projects

Example 6 with TypeSignature

use of io.crate.types.TypeSignature in project crate by crate.

the class UserDefinedFunctionsIntegrationTest method test_pg_get_function_result.

@Test
public void test_pg_get_function_result() throws Exception {
    TypeSignature returnTypeSig = TypeSignature.parseTypeSignature("array(array(integer))");
    String returnType = returnTypeSig.toString();
    Signature signature = Signature.builder().kind(FunctionType.SCALAR).name(new FunctionName(Schemas.DOC_SCHEMA_NAME, "make_2d_array")).argumentTypes(DataTypes.INTEGER.getTypeSignature()).returnType(returnTypeSig).build();
    int functionOid = OidHash.functionOid(signature);
    execute("select pg_get_function_result(?)", new Object[] { functionOid });
    assertThat(response.rows()[0][0], nullValue());
    execute("create function doc.make_2d_array(integer) returns array(array(integer)) language dummy_lang as ?", new Object[] { returnType });
    execute("select pg_get_function_result(" + functionOid + ")");
    assertThat(response.rows()[0][0], is(returnType));
    execute("drop function doc.make_2d_array(integer)");
    execute("select pg_get_function_result(" + functionOid + ")");
    assertThat(response.rows()[0][0], nullValue());
}