def test_node_label_present__add(self): node_data = self.make_node() # Remove some of the defaults to make it simpler node_data['metadata']['labels'] = { 'beta.kubernetes.io/os': 'linux', } labels = node_data['metadata']['labels'] with self.mock_func('node_labels', return_value=labels): with self.mock_func('node_add_label', return_value=node_data): actual = kubernetes.node_label_present( name='failure-domain.beta.kubernetes.io/zone', node='minikube', value='us-central1-a', ) self.assertDictEqual( { 'comment': '', 'changes': { 'minikube.failure-domain.beta.kubernetes.io/zone': { 'new': { 'failure-domain.beta.kubernetes.io/zone': 'us-central1-a', 'beta.kubernetes.io/os': 'linux' }, 'old': { 'beta.kubernetes.io/os': 'linux', }, }, }, 'name': 'failure-domain.beta.kubernetes.io/zone', 'result': True, }, actual, )
def test_node_label_present__update(self): node_data = self.make_node() # Remove some of the defaults to make it simpler node_data['metadata']['labels'] = { 'failure-domain.beta.kubernetes.io/region': 'us-west-1', } labels = node_data['metadata']['labels'] with self.mock_func('node_labels', return_value=labels): with self.mock_func('node_add_label', return_value=node_data): actual = kubernetes.node_label_present( name='failure-domain.beta.kubernetes.io/region', node='minikube', value='us-east-1', ) self.assertDictEqual( { 'changes': { 'minikube.failure-domain.beta.kubernetes.io/region': { 'new': {'failure-domain.beta.kubernetes.io/region': 'us-east-1'}, 'old': {'failure-domain.beta.kubernetes.io/region': 'us-west-1'}, } }, 'result': True, 'name': 'failure-domain.beta.kubernetes.io/region', 'comment': 'The label is already set, changing the value', }, actual, )
def test_node_label_present__update(self): node_data = self.make_node() # Remove some of the defaults to make it simpler node_data["metadata"]["labels"] = { "failure-domain.beta.kubernetes.io/region": "us-west-1", } labels = node_data["metadata"]["labels"] with self.mock_func("node_labels", return_value=labels): with self.mock_func("node_add_label", return_value=node_data): actual = kubernetes.node_label_present( name="failure-domain.beta.kubernetes.io/region", node="minikube", value="us-east-1", ) self.assertDictEqual( { "changes": { "minikube.failure-domain.beta.kubernetes.io/region": { "new": { "failure-domain.beta.kubernetes.io/region": "us-east-1" }, "old": { "failure-domain.beta.kubernetes.io/region": "us-west-1" }, } }, "result": True, "name": "failure-domain.beta.kubernetes.io/region", "comment": "The label is already set, changing the value", }, actual, )
def test_node_label_present__add(self): node_data = self.make_node() # Remove some of the defaults to make it simpler node_data["metadata"]["labels"] = { "beta.kubernetes.io/os": "linux", } labels = node_data["metadata"]["labels"] with self.mock_func("node_labels", return_value=labels): with self.mock_func("node_add_label", return_value=node_data): actual = kubernetes.node_label_present( name="failure-domain.beta.kubernetes.io/zone", node="minikube", value="us-central1-a", ) self.assertDictEqual( { "comment": "", "changes": { "minikube.failure-domain.beta.kubernetes.io/zone": { "new": { "failure-domain.beta.kubernetes.io/zone": "us-central1-a", "beta.kubernetes.io/os": "linux", }, "old": {"beta.kubernetes.io/os": "linux"}, }, }, "name": "failure-domain.beta.kubernetes.io/zone", "result": True, }, actual, )
def test_node_label_present__add_test_true(self): labels = self.make_node_labels() with self.mock_func("node_labels", return_value=labels, test=True): actual = kubernetes.node_label_present( name="com.zoo-animal", node="minikube", value="monkey", ) self.assertDictEqual( { "changes": {}, "result": None, "name": "com.zoo-animal", "comment": "The label is going to be set", }, actual, )
def test_node_label_present__add_test_true(self): labels = self.make_node_labels() with self.mock_func('node_labels', return_value=labels, test=True): actual = kubernetes.node_label_present( name='com.zoo-animal', node='minikube', value='monkey', ) self.assertDictEqual( { 'changes': {}, 'result': None, 'name': 'com.zoo-animal', 'comment': 'The label is going to be set', }, actual, )
def test_node_label_present__update_test_true(self): node_data = self.make_node() labels = node_data['metadata']['labels'] with self.mock_func('node_labels', return_value=labels): with self.mock_func('node_add_label', return_value=node_data, test=True): actual = kubernetes.node_label_present( name='failure-domain.beta.kubernetes.io/region', node='minikube', value='us-east-1', ) self.assertDictEqual( { 'changes': {}, 'result': None, 'name': 'failure-domain.beta.kubernetes.io/region', 'comment': 'The label is going to be updated', }, actual, )
def test_node_label_present__already_set(self): node_data = self.make_node() labels = node_data['metadata']['labels'] with self.mock_func('node_labels', return_value=labels): with self.mock_func('node_add_label', return_value=node_data): actual = kubernetes.node_label_present( name='failure-domain.beta.kubernetes.io/region', node='minikube', value='us-west-1', ) self.assertDictEqual( { 'changes': {}, 'result': True, 'name': 'failure-domain.beta.kubernetes.io/region', 'comment': 'The label is already set and has the specified value', }, actual, )
def test_node_label_present__update_test_true(self): node_data = self.make_node() labels = node_data["metadata"]["labels"] with self.mock_func("node_labels", return_value=labels): with self.mock_func("node_add_label", return_value=node_data, test=True): actual = kubernetes.node_label_present( name="failure-domain.beta.kubernetes.io/region", node="minikube", value="us-east-1", ) self.assertDictEqual( { "changes": {}, "result": None, "name": "failure-domain.beta.kubernetes.io/region", "comment": "The label is going to be updated", }, actual, )
def test_node_label_present__already_set(self): node_data = self.make_node() labels = node_data["metadata"]["labels"] with self.mock_func("node_labels", return_value=labels): with self.mock_func("node_add_label", return_value=node_data): actual = kubernetes.node_label_present( name="failure-domain.beta.kubernetes.io/region", node="minikube", value="us-west-1", ) self.assertDictEqual( { "changes": {}, "result": True, "name": "failure-domain.beta.kubernetes.io/region", "comment": "The label is already set and has the specified value", }, actual, )