/*
 * Copyright 2005-2014 Red Hat, Inc.
 *
 * Red Hat 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 io.fabric8.apps.zookeeper;

import io.fabric8.arquillian.kubernetes.Constants;
import io.fabric8.arquillian.kubernetes.Session;
import io.fabric8.kubernetes.api.KubernetesClient;
import io.fabric8.kubernetes.api.KubernetesHelper;
import io.fabric8.kubernetes.api.model.ReplicationController;
import io.fabric8.kubernetes.api.model.Service;
import io.fabric8.kubernetes.api.model.ServiceList;
import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.CuratorFrameworkFactory;
import org.apache.curator.retry.RetryNTimes;
import org.apache.zookeeper.CreateMode;
import org.assertj.core.api.Condition;
import org.assertj.core.util.Preconditions;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.arquillian.test.api.ArquillianResource;
import org.junit.Test;
import org.junit.runner.RunWith;

import static io.fabric8.kubernetes.assertions.Assertions.assertThat;

@RunWith(Arquillian.class)
public class ZooKeeperKubernetesTest {

    @ArquillianResource
    KubernetesClient client;

    @ArquillianResource
    Session session;

    @ArquillianResource
    ServiceList serviceList;

    @Test
    public void testZooKeeper() throws Exception {
        assertThat(client).replicationControllers().haveAtLeast(3, new Condition<ReplicationController>() {
            @Override
            public boolean matches(ReplicationController replicationController) {
                return replicationController.getId().startsWith("zookeeper-controller");
            }
        });

        assertThat(client).services().haveAtLeast(3, new Condition<Service>() {
            @Override
            public boolean matches(Service serviceSchema) {
                return serviceSchema.getId().startsWith("zk-peer");
            }
        });

        assertThat(client).services().haveAtLeast(3, new Condition<Service>() {
            @Override
            public boolean matches(Service serviceSchema) {
                return serviceSchema.getId().startsWith("zk-election");
            }
        });

        assertThat(client).pods().runningStatus().filterLabel(Constants.ARQ_KEY, session.getId()).hasSize(3);
    }

    @Test
    public void testClient() throws Exception {
        Service service = getRequiredService("zk-client");
        String serviceURL = service.getPortalIP() + ":" + service.getPort();
        try (CuratorFramework curator = CuratorFrameworkFactory.newClient(serviceURL, new RetryNTimes(5, 1000))) {
            curator.start();
            curator.blockUntilConnected();
            curator.create().creatingParentsIfNeeded().withMode(CreateMode.EPHEMERAL).forPath("/fabric8");
        }
    }


    private Service getRequiredService(String id) {
        Preconditions.checkNotNullOrEmpty(id);
        for (Service s : serviceList.getItems()) {
            if (id.equals(s.getId())) {
                return s;
            }
        }
        throw new IllegalStateException("Service with id:"+id+" doesn't exists.");
    }
}

@@ -819,32 +819,43 @@
 port io.fabric8.
+arquillian.
 kubernetes.api.K
@@ -854,27 +854,20 @@
 es.a
-pi.KubernetesHelper
+nnotation.Id
 ;%0Aim
@@ -978,60 +978,8 @@
 ce;%0A
-import io.fabric8.kubernetes.api.model.ServiceList;%0A
 impo
@@ -1217,52 +1217,8 @@
 on;%0A
-import org.assertj.core.util.Preconditions;%0A
 impo
@@ -1611,16 +1611,37 @@
 ssion;%0A%0A
+    @Id(%22zk-client%22)%0A
     @Arq
@@ -1671,23 +1671,16 @@
 vice
-List serviceLis
+ zkClien
 t;%0A%0A
@@ -2765,67 +2765,8 @@
 n %7B%0A
-        Service service = getRequiredService(%22zk-client%22);%0A
     
@@ -2789,23 +2789,24 @@
 ceURL = 
-service
+zkClient
 .getPort
@@ -2820,23 +2820,24 @@
 + %22:%22 + 
-service
+zkClient
 .getPort
@@ -3155,336 +3155,6 @@
  %7D%0A%0A
-%0A    private Service getRequiredService(String id) %7B%0A        Preconditions.checkNotNullOrEmpty(id);%0A        for (Service s : serviceList.getItems()) %7B%0A            if (id.equals(s.getId())) %7B%0A                return s;%0A            %7D%0A        %7D%0A        throw new IllegalStateException(%22Service with id:%22+id+%22 doesn't exists.%22);%0A    %7D%0A
 %7D%0A

Use the @Id annotation for obtaining the zk-client service.