mirror of
https://github.com/didi/KnowStreaming.git
synced 2026-01-07 15:12:14 +08:00
Add km module kafka
This commit is contained in:
39
tests/kafkatest/utils/remote_account.py
Normal file
39
tests/kafkatest/utils/remote_account.py
Normal file
@@ -0,0 +1,39 @@
|
||||
# Licensed to the Apache Software Foundation (ASF) under one or more
|
||||
# contributor license agreements. See the NOTICE file distributed with
|
||||
# this work for additional information regarding copyright ownership.
|
||||
# The ASF 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.
|
||||
|
||||
|
||||
def file_exists(node, file):
|
||||
"""Quick and dirty check for existence of remote file."""
|
||||
try:
|
||||
node.account.ssh("cat " + file, allow_fail=False)
|
||||
return True
|
||||
except:
|
||||
return False
|
||||
|
||||
def path_exists(node, path):
|
||||
"""Quick and dirty check for existence of remote path."""
|
||||
try:
|
||||
node.account.ssh("ls " + path, allow_fail=False)
|
||||
return True
|
||||
except:
|
||||
return False
|
||||
|
||||
def line_count(node, file):
|
||||
"""Return the line count of file on node"""
|
||||
out = [line for line in node.account.ssh_capture("wc -l %s" % file)]
|
||||
if len(out) != 1:
|
||||
raise Exception("Expected single line of output from wc -l")
|
||||
|
||||
return int(out[0].strip().split(" ")[0])
|
||||
Reference in New Issue
Block a user