def test_node_affinity_with_required(self): model = _constants.pod_with_node_affinity() pod = Pod(model) config = _utils.create_config() k8s = K8sPod(config=config, name="yo") k8s.model = pod if _utils.is_reachable(config): # ensure we have enough nodes nodes = K8sNode(config=config, name="yo").list() self.assertGreaterEqual(len(nodes), 3) # tag the nodes for i in range(len(nodes)): node = nodes[i] if i == 0: node.labels.update( {'kubernetes.io/e2e-az-name': 'e2e-az1'}) if i == 1: node.labels.update( {'kubernetes.io/e2e-az-name': 'e2e-az2'}) if i == 2: node.labels.update( {'kubernetes.io/e2e-az-name': 'e2e-az3'}) node.update() nodes = K8sNode(config=config, name="yo").list() for node in nodes: self.assertIn('kubernetes.io/e2e-az-name', node.labels) # create the pod k8s.create() # untag the nodes for node in nodes: node.labels.pop('kubernetes.io/e2e-az-name', None) node.update() pass
def list(self, pattern=None, labels=None): ls = super(K8sPod, self).list(labels=labels) pods = list(map(lambda pod: Pod(pod), ls)) if pattern is not None: pods = list(filter(lambda pod: pattern in pod.name, pods)) k8s = list() for x in pods: p = K8sPod(config=self.config, name=x.name).from_model(m=x) k8s.append(p) return k8s
def list(self, pattern=None): ls = super(K8sPod, self).list() pods = map(lambda x: Pod(x), ls) if pattern is not None: pods = filter(lambda x: pattern in x.name, pods) k8s = [] for x in pods: p = K8sPod(config=self.config, name=x.name) p.model = x k8s.append(p) return k8s
def test_pod_affinity_no_good_nodes(self): model = _constants.pod_with_pod_affinity() pod = Pod(model) config = _utils.create_config() k8s = K8sPod(config=config, name="yo") k8s.model = pod if _utils.is_reachable(config): nodes = K8sNode(config=config, name="yo").list() # untag all nodes for node in nodes: node.labels.pop('kubernetes.io/e2e-az-name', None) node.update() with self.assertRaises(TimedOutException): # timeout because of required k8s.create()
def test_pod_affinities(self): config = _utils.create_config() container = _utils.create_container(name="nginx", image="nginx:latest") pod_s1 = _utils.create_pod(config=config, name="s1") pod_s1.labels.update({'security': 'S1'}) pod_s1.node_selector = {'kubernetes.io/e2e-az-name': 'e2e-az1'} pod_s1.add_container(container) pod_s2 = _utils.create_pod(config=config, name="s2") pod_s2.labels.update({'security': 'S2'}) pod_s2.node_selector = {'kubernetes.io/e2e-az-name': 'e2e-az2'} pod_s2.add_container(container) model = _constants.pod_with_pod_affinity() pod = Pod(model) config = _utils.create_config() k8s = K8sPod(config=config, name="yo") k8s.model = pod if _utils.is_reachable(config): # ensure we have enough nodes nodes = K8sNode(config=config, name="yo").list() if len(nodes) < 3: self.skipTest(reason='Not enough nodes to perform test.') else: self.assertGreaterEqual(len(nodes), 3) # tag the nodes for i in range(len(nodes)): node = nodes[i] if i == 0: node.labels.update( {'kubernetes.io/e2e-az-name': 'e2e-az1'}) if i == 1: node.labels.update( {'kubernetes.io/e2e-az-name': 'e2e-az2'}) if i == 2: node.labels.update( {'kubernetes.io/e2e-az-name': 'e2e-az3'}) node.update() # create the pods for affinity lookup pod_s1.create() pod_s2.create() # create the pod with affinities k8s.create() pass
def get_by_name(config=None, name=None): if config is None: config = K8sConfig() if not is_valid_string(name): raise SyntaxError( 'K8sPod.get_by_name(): name: [ {0} ] is invalid.'.format(name)) pod_list = [] data = {'labelSelector': 'name={0}'.format(name)} pods = K8sPod(config=config, name=name).get_with_params(data=data) for pod in pods: try: p = Pod(pod) k8s_pod = K8sPod(config=config, name=p.metadata.name).get() pod_list.append(k8s_pod) except NotFoundException: pass return pod_list
def get_by_labels(config=None, labels=None): if config is None: config = K8sConfig() if not is_valid_dict(labels): raise SyntaxError( 'K8sPod.get_by_labels(): labels: [ {} ] is invalid.'.format( labels)) pod_list = [] selector = ",".join( ['%s=%s' % (key, value) for (key, value) in labels.items()]) data = {'labelSelector': selector} p = K8sPod(config=config, name=labels['name']) pods = p.get_with_params(data=data) for pod in pods: try: p = Pod(pod) k8s_pod = K8sPod(config=config, name=p.metadata.name).get() pod_list.append(k8s_pod) except NotFoundException: pass return pod_list
def get(self): self.model = Pod(self.get_model()) return self