/*
 * 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 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 ZooKeeperTest {

    @ArquillianResource
    KubernetesClient client;

    @ArquillianResource
    Session session;

    @Test
    public void testZooKeeper() throws Exception {
        assertThat(client).replicationController("zookeeper-controller").isNotNull();
        assertThat(client).pods().runningStatus().filterLabel(Constants.ARQ_KEY, session.getId()).hasSize(3);
    }
}

@@ -809,16 +809,177 @@
 Client;%0A
+import io.fabric8.kubernetes.api.model.ReplicationControllerSchema;%0Aimport io.fabric8.kubernetes.api.model.ServiceSchema;%0Aimport org.assertj.core.api.Condition;%0A
 import o
@@ -1016,16 +1016,16 @@
 illian;%0A
-
 import o
@@ -1471,45 +1471,853 @@
 ller
-(%22zookeeper-controller%22).isNotNull();
+s().haveAtLeast(3, new Condition%3CReplicationControllerSchema%3E() %7B%0A            @Override%0A            public boolean matches(ReplicationControllerSchema replicationControllerSchema) %7B%0A                return replicationControllerSchema.getId().startsWith(%22zookeeper-controller%22);%0A            %7D%0A        %7D);%0A%0A        assertThat(client).services().haveAtLeast(3, new Condition%3CServiceSchema%3E() %7B%0A            @Override%0A            public boolean matches(ServiceSchema serviceSchema) %7B%0A                return serviceSchema.getId().startsWith(%22zk-peer%22);%0A            %7D%0A        %7D);%0A%0A        assertThat(client).services().haveAtLeast(3, new Condition%3CServiceSchema%3E() %7B%0A            @Override%0A            public boolean matches(ServiceSchema serviceSchema) %7B%0A                return serviceSchema.getId().startsWith(%22zk-election%22);%0A            %7D%0A        %7D);%0A
 %0A   

Fix ZooKeeperTest.