def test_extract_labels(self):
     # Test when labels are not empty
     labels = {'label_a': 'val_a', 'label_b': 'val_b'}
     pod = Pod('v3.14', {}, [], labels=labels)
     self.expected['metadata']['labels'] = labels
     KubernetesRequestFactory.extract_labels(pod, self.input_req)
     self.assertEqual(self.input_req, self.expected)
 def test_extract_labels(self):
     # Test when labels are not empty
     labels = {'label_a': 'val_a', 'label_b': 'val_b'}
     pod = Pod('v3.14', {}, [], labels=labels)
     self.expected['metadata']['labels'] = labels
     KubernetesRequestFactory.extract_labels(pod, self.input_req)
     self.assertEqual(self.input_req, self.expected)
 def extract_labels(pod, req):
     KubernetesRequestFactory.extract_labels(pod, req)
     req['metadata']['labels']['cluster'] = 'airflow'
     req['metadata']['labels']['component'] = 'airflow-worker'