def get_node(self):
cloud_provider = MagicMock()
node = Node(self.name, self.node_kind, cloud_provider,
self.user_key_public, self.user_key_private,
self.user_key_name, self.image_user, self.security_group,
self.image, self.flavor, self.image_userdata)
return node
def node():
with NamedTemporaryFile() as tmp:
node = Node( # pylint: disable=redefined-outer-name
name='test',
cluster_name='cluster',
kind='frontend',
cloud_provider=MagicMock(),
user_key_public=tmp.name,
user_key_private=tmp.name,
user_key_name='key',
image_user='******',
security_group='secgroup',
image_id='ami-000000',
flavor='m1.tiny',
image_userdata=None)
yield node
def create_node(self, cluster_template, node_type, cloud_provider, name):
"""
:param cluster_template: name of the cluster template
:param node_type: type of the node, string defining the <group> in
the configuration
:param cloud_provider: instance of :py:class:`elasticluster
.providers.AbstractCloudProvider`
:param name: name of the node
:return: :py:class:`elasticluster.cluster.node` instance
"""
conf = self.cluster_conf[cluster_template]['nodes'][node_type]
conf_login = self.cluster_conf[cluster_template]['login']
return Node(name,
node_type,
cloud_provider,
conf_login['user_key_public'],
conf_login["user_key_private"],
conf_login['user_key_name'],
conf_login['image_user'],
conf['security_group'],
conf['image_id'],
conf['flavor'],
image_userdata=conf.get('image_userdata', ''))