소스 검색

[libzookeeper] Add a live test to the ZooKeeper API

Romain Rigaux 10 년 전
부모
커밋
c2ec3e4

+ 1 - 1
desktop/libs/libzookeeper/src/libzookeeper/conf.py

@@ -19,7 +19,7 @@ import logging
 
 from urlparse import urlparse
 
-from desktop.lib.conf import Config, UnspecifiedConfigSection, ConfigSection
+from desktop.lib.conf import Config
 
 
 LOG = logging.getLogger(__name__)

+ 0 - 2
desktop/libs/libzookeeper/src/libzookeeper/models.py

@@ -15,8 +15,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import json
-
 from kazoo.client import KazooClient
 from libzookeeper.conf import PRINCIPAL_NAME
 

+ 45 - 0
desktop/libs/libzookeeper/src/libzookeeper/tests.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python
+# Licensed to Cloudera, Inc. under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  Cloudera, Inc. 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.
+
+from nose.plugins.skip import SkipTest
+from nose.tools import assert_equal, assert_true, assert_false
+
+from django.contrib.auth.models import User
+
+from desktop.lib.django_test_util import make_logged_in_client
+from desktop.lib.test_utils import add_to_group, grant_access
+from hadoop.pseudo_hdfs4 import is_live_cluster
+
+from libzookeeper.models import get_children_data
+from libzookeeper.conf import zkensemble
+
+
+class TestWithZooKeeper:
+
+  @classmethod
+  def setup_class(cls):
+
+    if not is_live_cluster():
+      raise SkipTest()
+
+    cls.client = make_logged_in_client(username='test', is_superuser=False)
+    cls.user = User.objects.get(username='test')
+    add_to_group('test')
+    grant_access("test", "test", "libzookeeper")
+
+  def test_get_collections(self):
+    db = get_children_data(zkensemble(), '/solr')