def check_allocate_free(self):
cluster = JsonCluster({
"nodes": [{
"hostname": "localhost1"
}, {
"hostname": "localhost2"
}, {
"hostname": "localhost3"
}]
})
assert len(cluster) == 3
assert (cluster.num_available_nodes() == 3)
nodes = cluster.request(1)
nodes_hostnames = self.cluster_hostnames(nodes)
assert len(cluster) == 3
assert (cluster.num_available_nodes() == 2)
nodes2 = cluster.request(2)
nodes2_hostnames = self.cluster_hostnames(nodes2)
assert len(cluster) == 3
assert (cluster.num_available_nodes() == 0)
assert (nodes_hostnames.isdisjoint(nodes2_hostnames))
cluster.free(nodes)
assert (cluster.num_available_nodes() == 1)
cluster.free(nodes2)
assert (cluster.num_available_nodes() == 3)
def check_allocate_free(self):
cluster = JsonCluster({"nodes":[{"hostname": "localhost1"}, {"hostname": "localhost2"}, {"hostname": "localhost3"}]})
assert(cluster.num_available_nodes() == 3)
nodes = cluster.request(1)
nodes_hostnames = self.cluster_hostnames(nodes)
assert(cluster.num_available_nodes() == 2)
nodes2 = cluster.request(2)
nodes2_hostnames = self.cluster_hostnames(nodes2)
assert(cluster.num_available_nodes() == 0)
assert(nodes_hostnames.isdisjoint(nodes2_hostnames))
cluster.free(nodes)
assert(cluster.num_available_nodes() == 1)
cluster.free(nodes2)
assert(cluster.num_available_nodes() == 3)
def check_exhausts_supply(self):
cluster = JsonCluster(self.single_node_cluster_json)
with pytest.raises(RuntimeError):
cluster.request(2)
def check_exhausts_supply(self):
cluster = JsonCluster(self.single_node_cluster_json)
with pytest.raises(RuntimeError):
cluster.request(2)
