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()