class CheckLocalhostCluster(object):
def setup_method(self, method):
self.cluster = LocalhostCluster()
def check_size(self):
len(self.cluster) >= 2 ** 31 - 1
def check_request_free(self):
available = self.cluster.num_available_nodes()
initial_size = len(self.cluster)
# Should be able to allocate arbitrarily many nodes
slots = self.cluster.request(100)
assert(len(slots) == 100)
for slot in slots:
assert(slot.account.hostname == 'localhost')
assert(slot.account.user is None)
assert(slot.account.ssh_args is None)
assert(self.cluster.num_available_nodes() == (available - 100))
assert len(self.cluster) == initial_size # This shouldn't change
self.cluster.free(slots)
assert(self.cluster.num_available_nodes() == available)
class CheckLocalhostCluster(object):
def setup_method(self, _):
self.cluster = LocalhostCluster(is_type_based=False)
def check_size(self):
len(self.cluster) >= 2**31 - 1
def check_pickleable(self):
cluster = LocalhostCluster(is_type_based=False)
pickle.dumps(cluster)
def check_request_free(self):
available = self.cluster.num_available_nodes()
initial_size = len(self.cluster)
# Should be able to allocate arbitrarily many nodes
nodes = self.cluster.alloc(Service.setup_cluster_spec(num_nodes=100))
assert (len(nodes) == 100)
for i, node in enumerate(nodes):
assert node.account.hostname == 'localhost%d' % i
assert node.account.ssh_hostname == 'localhost'
assert node.account.ssh_config.hostname == 'localhost'
assert node.account.ssh_config.port == 22
assert node.account.user is None
assert (self.cluster.num_available_nodes() == (available - 100))
assert len(self.cluster) == initial_size # This shouldn't change
self.cluster.free(nodes)
assert (self.cluster.num_available_nodes() == available)
class CheckLocalhostCluster(object):
def setup_method(self, method):
self.cluster = LocalhostCluster()
def check_size(self):
len(self.cluster) >= 2**31 - 1
def check_request_free(self):
available = self.cluster.num_available_nodes()
initial_size = len(self.cluster)
# Should be able to allocate arbitrarily many nodes
slots = self.cluster.request(100)
assert (len(slots) == 100)
for slot in slots:
assert (slot.account.hostname == 'localhost')
assert (slot.account.user is None)
assert (slot.account.ssh_args is None)
assert (self.cluster.num_available_nodes() == (available - 100))
assert len(self.cluster) == initial_size # This shouldn't change
self.cluster.free(slots)
assert (self.cluster.num_available_nodes() == available)
class CheckLocalhostCluster(object):
def setup_method(self, method):
self.cluster = LocalhostCluster()
def check_request_free(self):
available = self.cluster.num_available_nodes()
# Should be able to allocate arbitrarily many nodes
slots = self.cluster.request(100)
assert (len(slots) == 100)
for slot in slots:
assert (slot.account.hostname == 'localhost')
assert (slot.account.user == None)
assert (slot.account.ssh_args == None)
assert (self.cluster.num_available_nodes() == (available - 100))
self.cluster.free(slots)
assert (self.cluster.num_available_nodes() == available)
class CheckLocalhostCluster(object):
def setup_method(self, method):
self.cluster = LocalhostCluster()
def check_request_free(self):
available = self.cluster.num_available_nodes()
# Should be able to allocate arbitrarily many nodes
slots = self.cluster.request(100)
assert(len(slots) == 100)
for slot in slots:
assert(slot.account.hostname == 'localhost')
assert(slot.account.user == None)
assert(slot.account.ssh_args == None)
assert(self.cluster.num_available_nodes() == (available - 100))
self.cluster.free(slots)
assert(self.cluster.num_available_nodes() == available)
