public static Connection getConnection(final String databaseName, final ContextManager contextManager) {
        return STATES.get(contextManager.getComputeNodeInstanceContext().getInstance().getState().getCurrentState()).getConnection(databaseName, contextManager);
    }
@Test
    void assertGetConnectionWithOkState() {
        ContextManager contextManager = mockContextManager(InstanceState.OK);
        assertThat(DriverStateContext.getConnection(DefaultDatabase.LOGIC_NAME, contextManager), instanceOf(ShardingSphereConnection.class));
    }