def check_sizes(self):
empty = NodeContainer()
assert 0 == empty.size()
assert 0 == len(empty)
nodes = [ClusterNode(MockAccount())]
container = NodeContainer(nodes)
assert 1 == container.size()
assert 1 == len(container)
def setup(self):
self.line_num = 6
self.eps = 0.01
self.account = MockAccount()
self.account.ssh("mkdir -p /tmp")
self.temp_file = "/tmp/ducktape-test-" + str(random.randint(0, 100000))
for i in range(self.line_num):
self.account.ssh("echo " + str(i) + " >> " + self.temp_file)
def check_add_and_remove(self):
nodes = [
ClusterNode(MockAccount()),
ClusterNode(MockAccount()),
ClusterNode(MockAccount()),
ClusterNode(MockAccount()),
ClusterNode(MockAccount())
]
container = NodeContainer([])
assert 0 == len(container)
container.add_node(nodes[0])
container.add_node(nodes[1])
container.add_node(nodes[2])
container2 = container.clone()
i = 0
for node in container:
assert nodes[i] == node
i += 1
assert 3 == len(container)
container.remove_node(nodes[0])
with pytest.raises(NodeNotPresentError):
container.remove_node(nodes[0])
assert 2 == len(container)
assert 3 == len(container2)
def setup(self):
self.server = SimpleServer()
self.account = MockAccount()
