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