def __init__(self, config=None, name=None, image=None, replicas=0):
super(K8sDeployment, self).__init__(
config=config,
obj_type='Deployment',
name=name
)
self.desired_replicas = replicas
labels = {'name': name}
sel = LabelSelector()
sel.match_labels = labels
self.selector = sel
self.labels = labels
self.pod_labels = labels
if image is not None:
container = K8sContainer(name=name, image=image)
self.add_container(container)
if self.config.pull_secret is not None:
self.add_image_pull_secrets(self.config.pull_secret)
def create_container(name=None, image="redis"):
return K8sContainer(name=name, image=image)
def test_add_volume_mount_no_args(self):
name = "yomama"
image = "redis"
c = K8sContainer(name=name, image=image)
with self.assertRaises(SyntaxError):
c.add_volume_mount()
def test_init_image_only(self):
image = "redis"
with self.assertRaises(SyntaxError):
K8sContainer(image=image)
def test_init_name_only(self):
name = "yomama"
with self.assertRaises(SyntaxError):
K8sContainer(name=name)
def test_init_no_args(self):
with self.assertRaises(SyntaxError):
K8sContainer()
def test_drop_capabilities_invalid_args(self):
name = "yomama"
image = "redis"
with self.assertRaises(SyntaxError):
c = K8sContainer(name=name, image=image)
c.drop_capabilities('NET_RAW')
