2222import org .apache .commons .logging .LogFactory ;
2323import org .apache .curator .framework .CuratorFramework ;
2424import org .apache .curator .framework .CuratorFrameworkFactory ;
25- import org .apache .curator .framework .imps .CuratorFrameworkImpl ;
2625import org .junit .After ;
2726import org .junit .Before ;
2827import org .junit .Test ;
5150 */
5251public class ZookeeperConfigDataCustomizationIntegrationTests {
5352
54- private static final Log log = LogFactory
55- .getLog (ZookeeperConfigDataCustomizationIntegrationTests .class );
56-
5753 public static final String PREFIX = "test__configdata__" ;
58-
5954 public static final String ROOT = "/" + PREFIX + UUID .randomUUID ();
60-
55+ private static final Log log = LogFactory
56+ .getLog (ZookeeperConfigDataCustomizationIntegrationTests .class );
6157 private ConfigurableApplicationContext context ;
6258 private BindHandlerBootstrapper bindHandlerBootstrapper ;
6359
@@ -86,8 +82,9 @@ CuratorFramework curatorFramework(BootstrapContext context) {
8682 ZookeeperProperties properties = context .get (ZookeeperProperties .class );
8783 CuratorFrameworkFactory .Builder builder = CuratorFrameworkFactory .builder ()
8884 .retryPolicy (CuratorFactory .retryPolicy (properties ))
89- .connectString (properties .getConnectString ());
90- TestCuratorFramework curator = new TestCuratorFramework (builder );
85+ .connectString (properties .getConnectString ())
86+ .namespace ("testnamespace" );
87+ CuratorFramework curator = builder .build ();
9188 curator .start ();
9289 try {
9390 curator .blockUntilConnected (properties .getBlockUntilConnectedWait (),
@@ -102,16 +99,11 @@ CuratorFramework curatorFramework(BootstrapContext context) {
10299 @ Test
103100 public void curatorFrameworkIsCustom () {
104101 CuratorFramework curator = context .getBean (CuratorFramework .class );
105- assertThat (curator ).isNotNull ().isInstanceOf (TestCuratorFramework .class );
102+ assertThat (curator ).isNotNull ();
103+ assertThat (curator .getNamespace ()).isEqualTo ("testnamespace" );
106104 assertThat (bindHandlerBootstrapper .onSuccessCount ).isGreaterThan (0 );
107105 }
108106
109- static class TestCuratorFramework extends CuratorFrameworkImpl {
110- TestCuratorFramework (CuratorFrameworkFactory .Builder builder ) {
111- super (builder );
112- }
113- }
114-
115107 @ Configuration
116108 @ EnableAutoConfiguration
117109 static class Config {
0 commit comments