def test_can_select_nodes_without_label(cl, client, has_conf):
has_conf.return_value = False
v1 = MagicMock()
v1.list_node.return_value = io.BytesIO(
json.dumps({"hey": "there"}).encode('utf-8'))
client.CoreV1Api.return_value = v1
resp = get_nodes()
v1.list_node.assert_called_with(_preload_content=False)
assert resp == {"hey": "there"}
def test_can_select_nodes_by_label(cl, client, has_conf):
has_conf.return_value = False
v1 = MagicMock()
v1.list_node.return_value = io.BytesIO(
json.dumps({"hey": "there"}).encode('utf-8'))
client.CoreV1Api.return_value = v1
label_selector = 'beta.kubernetes.io/instance-type=m5.large'
resp = get_nodes(label_selector=label_selector)
v1.list_node.assert_called_with(
label_selector=label_selector, _preload_content=False)
assert resp == {"hey": "there"}