diff --git a/shardingsphere-db-protocol/shardingsphere-db-protocol-postgresql/src/test/java/org/apache/shardingsphere/db/protocol/postgresql/packet/command/query/binary/bind/PostgreSQLBinaryResultSetRowPacketTest.java b/shardingsphere-db-protocol/shardingsphere-db-protocol-postgresql/src/test/java/org/apache/shardingsphere/db/protocol/postgresql/packet/command/query/binary/bind/PostgreSQLBinaryResultSetRowPacketTest.java new file mode 100644 index 0000000000000000000000000000000000000000..bce6eeb0b5bb797f12649d8f175d54c3d2deadb4 --- /dev/null +++ b/shardingsphere-db-protocol/shardingsphere-db-protocol-postgresql/src/test/java/org/apache/shardingsphere/db/protocol/postgresql/packet/command/query/binary/bind/PostgreSQLBinaryResultSetRowPacketTest.java @@ -0,0 +1,70 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.shardingsphere.db.protocol.postgresql.packet.command.query.binary.bind; + +import org.apache.shardingsphere.db.protocol.postgresql.constant.PostgreSQLColumnType; +import org.apache.shardingsphere.db.protocol.postgresql.packet.command.PostgreSQLCommandPacketType; +import org.apache.shardingsphere.db.protocol.postgresql.payload.PostgreSQLPacketPayload; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.junit.MockitoJUnitRunner; + +import java.util.Arrays; + +import static org.hamcrest.CoreMatchers.is; +import static org.junit.Assert.assertThat; +import static org.mockito.Mockito.verify; + +@RunWith(MockitoJUnitRunner.class) +public class PostgreSQLBinaryResultSetRowPacketTest { + + @Mock + private PostgreSQLPacketPayload payload; + + @Test + public void assertWriteStringData() { + PostgreSQLBinaryResultSetRowPacket rowPacket = new PostgreSQLBinaryResultSetRowPacket(Arrays.asList("value", "b"), + Arrays.asList(PostgreSQLColumnType.POSTGRESQL_TYPE_VARCHAR, PostgreSQLColumnType.POSTGRESQL_TYPE_VARCHAR)); + assertThat(rowPacket.getData().size(), is(2)); + rowPacket.write(payload); + verify(payload).writeInt2(2); + verify(payload).writeInt4(5); + verify(payload).writeStringEOF("value"); + verify(payload).writeInt4(1); + verify(payload).writeStringEOF("b"); + } + + @Test + public void assertWriteIntData() { + PostgreSQLBinaryResultSetRowPacket rowPacket = new PostgreSQLBinaryResultSetRowPacket(Arrays.asList(10), + Arrays.asList(PostgreSQLColumnType.POSTGRESQL_TYPE_INT4)); + assertThat(rowPacket.getData().size(), is(1)); + rowPacket.write(payload); + verify(payload).writeInt2(1); + verify(payload).writeInt4(4); + verify(payload).writeInt4(10); + } + + @Test + public void assertGetMessageType() { + PostgreSQLBinaryResultSetRowPacket rowPacket = new PostgreSQLBinaryResultSetRowPacket(null, null); + assertThat(rowPacket.getMessageType(), is(PostgreSQLCommandPacketType.DATA_ROW.getValue())); + } + +} diff --git a/shardingsphere-db-protocol/shardingsphere-db-protocol-postgresql/src/test/java/org/apache/shardingsphere/db/protocol/postgresql/packet/command/query/binary/bind/PostgreSQLBindCompletePacketTest.java b/shardingsphere-db-protocol/shardingsphere-db-protocol-postgresql/src/test/java/org/apache/shardingsphere/db/protocol/postgresql/packet/command/query/binary/bind/PostgreSQLBindCompletePacketTest.java new file mode 100644 index 0000000000000000000000000000000000000000..ad1d597ba2f92013d9fadb350c042a5708f98b1c --- /dev/null +++ b/shardingsphere-db-protocol/shardingsphere-db-protocol-postgresql/src/test/java/org/apache/shardingsphere/db/protocol/postgresql/packet/command/query/binary/bind/PostgreSQLBindCompletePacketTest.java @@ -0,0 +1,53 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.shardingsphere.db.protocol.postgresql.packet.command.query.binary.bind; + +import io.netty.buffer.ByteBuf; +import org.apache.shardingsphere.db.protocol.postgresql.packet.command.PostgreSQLCommandPacketType; +import org.apache.shardingsphere.db.protocol.postgresql.payload.PostgreSQLPacketPayload; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.junit.MockitoJUnitRunner; + +import static org.hamcrest.CoreMatchers.is; +import static org.junit.Assert.assertThat; + +@RunWith(MockitoJUnitRunner.class) +public class PostgreSQLBindCompletePacketTest { + + @Mock + private PostgreSQLPacketPayload payload; + + @Mock + private ByteBuf byteBuf; + + @Test + public void assertWrite() { + PostgreSQLBindCompletePacket rowPacket = new PostgreSQLBindCompletePacket(); + rowPacket.write(payload); + assertThat(byteBuf.writerIndex(), is(0)); + } + + @Test + public void assertGetMessageType() { + PostgreSQLBindCompletePacket rowPacket = new PostgreSQLBindCompletePacket(); + assertThat(rowPacket.getMessageType(), is(PostgreSQLCommandPacketType.BIND_COMPLETE.getValue())); + } + +} diff --git a/shardingsphere-db-protocol/shardingsphere-db-protocol-postgresql/src/test/java/org/apache/shardingsphere/db/protocol/postgresql/packet/command/query/binary/bind/PostgreSQLComBindPacketTest.java b/shardingsphere-db-protocol/shardingsphere-db-protocol-postgresql/src/test/java/org/apache/shardingsphere/db/protocol/postgresql/packet/command/query/binary/bind/PostgreSQLComBindPacketTest.java new file mode 100644 index 0000000000000000000000000000000000000000..942eb25be3e3bba8fd0c9544f2a3d8bd0fbf07ed --- /dev/null +++ b/shardingsphere-db-protocol/shardingsphere-db-protocol-postgresql/src/test/java/org/apache/shardingsphere/db/protocol/postgresql/packet/command/query/binary/bind/PostgreSQLComBindPacketTest.java @@ -0,0 +1,88 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.shardingsphere.db.protocol.postgresql.packet.command.query.binary.bind; + +import org.apache.shardingsphere.db.protocol.postgresql.constant.PostgreSQLColumnType; +import org.apache.shardingsphere.db.protocol.postgresql.packet.command.PostgreSQLCommandPacketType; +import org.apache.shardingsphere.db.protocol.postgresql.packet.command.query.binary.BinaryStatementRegistry; +import org.apache.shardingsphere.db.protocol.postgresql.packet.command.query.binary.PostgreSQLBinaryStatementParameterType; +import org.apache.shardingsphere.db.protocol.postgresql.payload.PostgreSQLPacketPayload; +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.junit.MockitoJUnitRunner; + +import java.sql.SQLException; +import java.util.Arrays; + +import static org.hamcrest.CoreMatchers.is; +import static org.junit.Assert.assertNull; +import static org.junit.Assert.assertThat; +import static org.mockito.Mockito.when; + +@RunWith(MockitoJUnitRunner.class) +public class PostgreSQLComBindPacketTest { + + @Mock + private PostgreSQLPacketPayload payload; + + @Before + public void init() { + when(payload.readInt4()).thenReturn(1); + when(payload.readStringNul()).thenReturn(""); + when(payload.readStringNul()).thenReturn("sts-id"); + when(payload.readInt2()).thenReturn(1); + + BinaryStatementRegistry.getInstance().register(1); + } + + @Test + public void assertWrite() throws SQLException { + //mock parameters + when(payload.readInt2()).thenReturn(1); + when(payload.readInt4()).thenReturn(1); + when(payload.readInt8()).thenReturn(11L); + + String sql = "select * from order where id = ? "; + BinaryStatementRegistry.getInstance().get(1).register("sts-id", sql, 1, Arrays.asList(new PostgreSQLBinaryStatementParameterType(PostgreSQLColumnType.POSTGRESQL_TYPE_INT8))); + + PostgreSQLComBindPacket bindPacket = new PostgreSQLComBindPacket(payload, 1); + bindPacket.write(payload); + + assertThat(bindPacket.getSql(), is(sql)); + assertThat(bindPacket.getParameters().size(), is(1)); + assertThat(bindPacket.isBinaryRowData(), is(true)); + } + + @Test + public void assertWriteWithEmptySql() throws SQLException { + PostgreSQLComBindPacket bindPacket = new PostgreSQLComBindPacket(payload, 1); + bindPacket.write(payload); + assertNull(bindPacket.getSql()); + assertThat(bindPacket.getParameters().size(), is(0)); + assertThat(bindPacket.isBinaryRowData(), is(true)); + } + + @Test + public void getMessageType() throws SQLException { + PostgreSQLComBindPacket bindPacket = new PostgreSQLComBindPacket(payload, 1); + assertThat(bindPacket.getMessageType(), is(PostgreSQLCommandPacketType.BIND.getValue())); + } + +}