void
test__aocs_addcol_init(void **state)
{
AOCSAddColumnDesc desc;
RelationData reldata;
FormData_pg_appendonly pgappendonly;
int nattr = 5;
StdRdOptions **opts =
(StdRdOptions **) malloc(sizeof(StdRdOptions *) * nattr);
/* 3 existing columns */
opts[0] = opts[1] = opts[2] = (StdRdOptions *) NULL;
/* 2 newly added columns */
opts[3] = (StdRdOptions *) malloc(sizeof(StdRdOptions));
strcpy(opts[3]->compresstype, "rle_type");
opts[3]->compresslevel = 2;
opts[3]->blocksize = 8192;
opts[4] = (StdRdOptions *) malloc(sizeof(StdRdOptions));
strcpy(opts[4]->compresstype, "none");
opts[4]->compresslevel = 0;
opts[4]->blocksize = 8192 * 2;
/* One call to RelationGetAttributeOptions() */
expect_any(RelationGetAttributeOptions, rel);
will_return(RelationGetAttributeOptions, opts);
/* Two calls to create_datumstreamwrite() */
expect_string(create_datumstreamwrite, compName, "rle_type");
expect_string(create_datumstreamwrite, compName, "none");
expect_value(create_datumstreamwrite, compLevel, 2);
expect_value(create_datumstreamwrite, compLevel, 0);
expect_value_count(create_datumstreamwrite, checksum, true, 2);
expect_value_count(create_datumstreamwrite, safeFSWriteSize, 0, 2);
expect_value(create_datumstreamwrite, maxsz, 8192);
expect_value(create_datumstreamwrite, maxsz, 8192 * 2);
expect_value(create_datumstreamwrite, needsWAL, true);
expect_value(create_datumstreamwrite, needsWAL, true);
expect_any_count(create_datumstreamwrite, attr, 2);
expect_any_count(create_datumstreamwrite, relname, 2);
expect_any_count(create_datumstreamwrite, title, 2);
will_return_count(create_datumstreamwrite, NULL, 2);
pgappendonly.checksum = true;
FormData_pg_class rel;
rel.relpersistence = RELPERSISTENCE_PERMANENT;
reldata.rd_rel = &rel;
reldata.rd_appendonly = &pgappendonly;
reldata.rd_att = (TupleDesc) malloc(sizeof(struct tupleDesc));
reldata.rd_att->attrs =
(Form_pg_attribute *) malloc(sizeof(Form_pg_attribute *) * nattr);
memset(reldata.rd_att->attrs, 0, sizeof(Form_pg_attribute *) * nattr);
reldata.rd_att->natts = 5;
/* 3 existing columns, 2 new columns */
desc = aocs_addcol_init(&reldata, 2);
assert_int_equal(desc->num_newcols, 2);
assert_int_equal(desc->cur_segno, -1);
}
static void
test_secure_channel_listen_start_listen_failed() {
setup();
int listen_fd = 1;
listener_info.listen_fd = -1;
expect_value( mock_socket, domain, PF_INET );
expect_value( mock_socket, type, SOCK_STREAM );
expect_value( mock_socket, protocol, 0 );
will_return( mock_socket, listen_fd );
expect_value_count( mock_setsockopt, sockfd, listen_fd, 1 );
expect_value_count( mock_setsockopt, level, SOL_SOCKET, 1 );
expect_value_count( mock_setsockopt, optname, SO_REUSEADDR, 1 );
expect_not_value_count( mock_setsockopt, optval, NULL, 1 );
expect_value_count( mock_setsockopt, optlen, sizeof( int ), 1 );
will_return_count( mock_setsockopt, 0, 1 );
expect_value_count( mock_setsockopt, sockfd, listen_fd, 1 );
expect_value_count( mock_setsockopt, level, IPPROTO_TCP, 1 );
expect_value_count( mock_setsockopt, optname, TCP_NODELAY, 1 );
expect_not_value_count( mock_setsockopt, optval, NULL, 1 );
expect_value_count( mock_setsockopt, optlen, sizeof( int ), 1 );
will_return_count( mock_setsockopt, 0, 1 );
expect_value( mock_bind, sockfd, listen_fd );
expect_not_value( mock_bind, addr_in, NULL );
expect_value( mock_bind, addrlen, sizeof( struct sockaddr_in ) );
expect_value( mock_bind, sin_family32, AF_INET );
expect_value( mock_bind, sin_port32, ( uint32_t ) htons( OFP_TCP_PORT ) );
expect_value( mock_bind, addr_in->sin_addr.s_addr, htonl( INADDR_ANY ) );
will_return( mock_bind, 0 );
expect_value( mock_listen, sockfd, listen_fd );
expect_value( mock_listen, backlog, 128 );
will_return( mock_listen, -1 );
expect_value( mock_close, fd, listen_fd );
will_return_void( mock_close );
assert_false( secure_channel_listen_start( &listener_info ) );
assert_true( listener_info.listen_fd == -1 );
teardown();
}
void mockLibpq(PGconn *pgConn, int motionListener, int qePid)
{
expect_any_count(PQconnectdbParams, keywords, -1);
expect_any_count(PQconnectdbParams, values, -1);
expect_any_count(PQconnectdbParams, expand_dbname, -1);
will_return_count(PQconnectdbParams, pgConn, TOTOAL_SEGMENTS);
expect_value_count(PQstatus, conn, pgConn, -1);
will_return_count(PQstatus, CONNECTION_OK, -1);
expect_value_count(PQsocket, conn, pgConn, -1);
will_return_count(PQsocket, 100, -1);
expect_value_count(PQsetNoticeReceiver, conn, pgConn, -1);
expect_any_count(PQsetNoticeReceiver, proc, -1);
expect_any_count(PQsetNoticeReceiver, arg, -1);
will_return_count(PQsetNoticeReceiver, CONNECTION_OK, -1);
expect_value_count(PQgetQEdetail, conn, pgConn, -1);
will_return_count(PQgetQEdetail, motionListener, -1);
expect_value_count(PQbackendPID, conn, pgConn, -1);
will_return_count(PQbackendPID, qePid, -1);
}
