@@ -0,0 +1,2394 @@
+/*%0A * Licensed under the Apache License, Version 2.0 (the %22License%22);%0A * you may not use this file except in compliance with the License.%0A * You may obtain a copy of the License at%0A *%0A *     http://www.apache.org/licenses/LICENSE-2.0%0A *%0A * Unless required by applicable law or agreed to in writing, software%0A * distributed under the License is distributed on an %22AS IS%22 BASIS,%0A * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.%0A * See the License for the specific language governing permissions and%0A * limitations under the License.%0A */%0Apackage io.prestosql;%0A%0Aimport io.prestosql.connector.CatalogName;%0Aimport org.testng.annotations.Test;%0A%0Aimport java.util.Map;%0A%0Aimport static io.prestosql.testing.TestingSession.testSessionBuilder;%0Aimport static org.assertj.core.api.Assertions.assertThat;%0A%0Apublic class TestSession%0A%7B%0A    @Test%0A    public void testSetCatalogProperty()%0A    %7B%0A        Session session = Session.builder(testSessionBuilder().build())%0A                .setCatalogSessionProperty(%22some_catalog%22, %22first_property%22, %22some_value%22)%0A                .build();%0A%0A        assertThat(session.getUnprocessedCatalogProperties())%0A                .isEqualTo(Map.of(%22some_catalog%22, Map.of(%22first_property%22, %22some_value%22)));%0A%0A        // empty, will be populated at transaction start%0A        assertThat(session.getConnectorProperties())%0A                .isEqualTo(Map.of());%0A%0A        // empty, will be populated at transaction start%0A        assertThat(session.getConnectorProperties(new CatalogName(%22some_catalog%22)))%0A                .isEqualTo(Map.of());%0A    %7D%0A%0A    @Test%0A    public void testBuildWithCatalogProperty()%0A    %7B%0A        Session session = Session.builder(testSessionBuilder().build())%0A                .setCatalogSessionProperty(%22some_catalog%22, %22first_property%22, %22some_value%22)%0A                .build();%0A        session = Session.builder(session)%0A                .build();%0A%0A        assertThat(session.getUnprocessedCatalogProperties())%0A                .isEqualTo(Map.of(%22some_catalog%22, Map.of(%22first_property%22, %22some_value%22)));%0A%0A        // empty, will be populated at transaction start%0A        assertThat(session.getConnectorProperties())%0A                .isEqualTo(Map.of());%0A%0A        // empty, will be populated at transaction start%0A        assertThat(session.getConnectorProperties(new CatalogName(%22some_catalog%22)))%0A                .isEqualTo(Map.of());%0A    %7D%0A%7D%0A

Add unit test for Session