def test_scaler_always_upscale(monkeypatch):
api = MagicMock()
monkeypatch.setattr('kube_downscaler.scaler.helper.get_kube_api', MagicMock(return_value=api))
def get(url, version, **kwargs):
if url == 'pods':
data = {'items': []}
elif url == 'deployments':
data = {'items': [{'metadata': {'name': 'deploy-1', 'namespace': 'ns-1'}, 'spec': {'replicas': 1}}]}
elif url == 'statefulsets':
data = {'items': []}
elif url == 'stacksets':
data = {'items': []}
elif url == 'namespaces/ns-1':
data = {'metadata': {}}
else:
raise Exception(f'unexpected call: {url}, {version}, {kwargs}')
response = MagicMock()
response.json.return_value = data
return response
api.get = get
kinds = frozenset(['statefulset', 'deployment', 'stackset'])
scale(namespace=None, upscale_period='always', downscale_period='never', default_uptime='never', default_downtime='always', kinds=kinds,
exclude_namespaces=[], exclude_deployments=[], exclude_statefulsets=[], dry_run=False, grace_period=300, downtime_replicas=0)
api.patch.assert_not_called()
def test_scaler_down_to(monkeypatch):
api = MagicMock()
monkeypatch.setattr("kube_downscaler.scaler.helper.get_kube_api",
MagicMock(return_value=api))
monkeypatch.setattr("kube_downscaler.scaler.helper.add_event",
MagicMock(return_value=None))
SCALE_TO = 1
def get(url, version, **kwargs):
if url == "pods":
data = {"items": []}
elif url == "deployments":
data = {
"items": [
{
"metadata": {
"name": "deploy-1",
"namespace": "default",
"creationTimestamp": "2019-03-01T16:38:00Z",
"annotations": {
DOWNTIME_REPLICAS_ANNOTATION: SCALE_TO
},
},
"spec": {
"replicas": 5
},
},
]
}
elif url == "namespaces/default":
data = {"metadata": {}}
else:
raise Exception(f"unexpected call: {url}, {version}, {kwargs}")
response = MagicMock()
response.json.return_value = data
return response
api.get = get
include_resources = frozenset(["deployments"])
scale(
namespace=None,
upscale_period="never",
downscale_period="never",
default_uptime="never",
default_downtime="always",
include_resources=include_resources,
exclude_namespaces=[],
exclude_deployments=[],
dry_run=False,
grace_period=300,
downtime_replicas=0,
enable_events=True,
)
assert api.patch.call_count == 1
assert api.patch.call_args[1]["url"] == "/deployments/deploy-1"
assert json.loads(
api.patch.call_args[1]["data"])["spec"]["replicas"] == SCALE_TO
def test_scaler_down_to(monkeypatch):
api = MagicMock()
monkeypatch.setattr('kube_downscaler.scaler.helper.get_kube_api', MagicMock(return_value=api))
SCALE_TO = 1
def get(url, version, **kwargs):
if url == 'pods':
data = {'items': []}
elif url == 'deployments':
data = {'items': [
{
'metadata': {
'name': 'deploy-1', 'namespace': 'default', 'creationTimestamp': '2019-03-01T16:38:00Z',
'annotations': {DOWNTIME_REPLICAS_ANNOTATION: SCALE_TO},
}, 'spec': {'replicas': 5}
},
]}
elif url == 'namespaces/default':
data = {'metadata': {}}
else:
raise Exception(f'unexpected call: {url}, {version}, {kwargs}')
response = MagicMock()
response.json.return_value = data
return response
api.get = get
kinds = frozenset(['deployment'])
scale(namespace=None, upscale_period='never', downscale_period='never', default_uptime='never', default_downtime='always', kinds=kinds,
exclude_namespaces=[], exclude_deployments=[], exclude_statefulsets=[], dry_run=False, grace_period=300, downtime_replicas=0)
assert api.patch.call_count == 1
assert api.patch.call_args[1]['url'] == 'deployments/deploy-1'
assert json.loads(api.patch.call_args[1]['data'])['spec']['replicas'] == SCALE_TO
def test_scaler_upscale_on_exclude_namespace(monkeypatch):
api = MagicMock()
monkeypatch.setattr(
"kube_downscaler.scaler.helper.get_kube_api", MagicMock(return_value=api)
)
ORIGINAL_REPLICAS = 2
def get(url, version, **kwargs):
if url == "pods":
data = {"items": []}
elif url == "deployments":
data = {
"items": [
{
"metadata": {
"name": "deploy-1",
"namespace": "default",
"annotations": {
ORIGINAL_REPLICAS_ANNOTATION: ORIGINAL_REPLICAS,
},
},
"spec": {"replicas": 0},
},
]
}
elif url == "namespaces/default":
data = {"metadata": {"annotations": {EXCLUDE_ANNOTATION: "true"}}}
else:
raise Exception(f"unexpected call: {url}, {version}, {kwargs}")
response = MagicMock()
response.json.return_value = data
return response
api.get = get
include_resources = frozenset(["deployments"])
scale(
namespace=None,
upscale_period="never",
downscale_period="never",
default_uptime="never",
default_downtime="always",
include_resources=include_resources,
exclude_namespaces=[],
exclude_deployments=[],
dry_run=False,
grace_period=300,
downtime_replicas=0,
)
assert api.patch.call_count == 1
assert api.patch.call_args[1]["url"] == "/deployments/deploy-1"
assert (
json.loads(api.patch.call_args[1]["data"])["spec"]["replicas"]
== ORIGINAL_REPLICAS
)
assert not json.loads(api.patch.call_args[1]["data"])["metadata"]["annotations"][
ORIGINAL_REPLICAS_ANNOTATION
]
def test_scaler_namespace_excluded(monkeypatch):
api = MagicMock()
monkeypatch.setattr('kube_downscaler.scaler.helper.get_kube_api', MagicMock(return_value=api))
def get(url, version, **kwargs):
if url == 'pods':
data = {'items': []}
elif url == 'deployments':
data = {'items': [
{'metadata': {'name': 'sysdep-1', 'namespace': 'system-ns', 'creationTimestamp': '2019-03-01T16:38:00Z'}, 'spec': {'replicas': 1}},
{'metadata': {'name': 'deploy-2', 'namespace': 'default', 'creationTimestamp': '2019-03-01T16:38:00Z'}, 'spec': {'replicas': 2}}
]}
elif url == 'namespaces/default':
data = {'metadata': {}}
else:
raise Exception(f'unexpected call: {url}, {version}, {kwargs}')
response = MagicMock()
response.json.return_value = data
return response
api.get = get
include_resources = frozenset(['deployments'])
scale(namespace=None, upscale_period='never', downscale_period='never', default_uptime='never', default_downtime='always',
include_resources=include_resources, exclude_namespaces=['system-ns'], exclude_deployments=[], exclude_statefulsets=[], dry_run=False,
grace_period=300, downtime_replicas=0)
assert api.patch.call_count == 1
# make sure that deploy-2 was updated (namespace of sysdep-1 was excluded)
patch_data = {"metadata": {"name": "deploy-2", "namespace": "default", "creationTimestamp": "2019-03-01T16:38:00Z",
'annotations': {ORIGINAL_REPLICAS_ANNOTATION: '2'}}, "spec": {"replicas": 0}}
assert api.patch.call_args[1]['url'] == 'deployments/deploy-2'
assert json.loads(api.patch.call_args[1]['data']) == patch_data
def test_scaler_upscale_on_exclude(monkeypatch):
api = MagicMock()
monkeypatch.setattr('kube_downscaler.scaler.helper.get_kube_api',
MagicMock(return_value=api))
ORIGINAL_REPLICAS = 2
def get(url, version, **kwargs):
if url == 'pods':
data = {'items': []}
elif url == 'deployments':
data = {
'items': [
{
'metadata': {
'name': 'deploy-1',
'namespace': 'default',
'annotations': {
EXCLUDE_ANNOTATION: 'true',
ORIGINAL_REPLICAS_ANNOTATION:
ORIGINAL_REPLICAS,
}
},
'spec': {
'replicas': 0
}
},
]
}
elif url == 'namespaces/default':
data = {'metadata': {}}
else:
raise Exception(f'unexpected call: {url}, {version}, {kwargs}')
response = MagicMock()
response.json.return_value = data
return response
api.get = get
kinds = frozenset(['deployment'])
scale(namespace=None,
upscale_period='never',
downscale_period='never',
default_uptime='never',
default_downtime='always',
kinds=kinds,
exclude_namespaces=[],
exclude_deployments=[],
exclude_statefulsets=[],
dry_run=False,
grace_period=300,
downtime_replicas=0)
assert api.patch.call_count == 1
assert api.patch.call_args[1]['url'] == 'deployments/deploy-1'
assert json.loads(api.patch.call_args[1]
['data'])["spec"]["replicas"] == ORIGINAL_REPLICAS
assert not json.loads(
api.patch.call_args[1]
['data'])["metadata"]["annotations"][ORIGINAL_REPLICAS_ANNOTATION]
def run_loop(run_once, namespace, kinds, upscale_period, downscale_period,
default_uptime, default_downtime, exclude_namespaces,
exclude_deployments, exclude_statefulsets, grace_period, interval,
dry_run, downtime_replicas):
handler = shutdown.GracefulShutdown()
while True:
try:
scale(namespace,
upscale_period,
downscale_period,
default_uptime,
default_downtime,
kinds=frozenset(kinds),
exclude_namespaces=frozenset(exclude_namespaces.split(',')),
exclude_deployments=frozenset(
exclude_deployments.split(',')),
exclude_statefulsets=frozenset(
exclude_statefulsets.split(',')),
dry_run=dry_run,
grace_period=grace_period,
downtime_replicas=downtime_replicas)
except Exception as e:
logger.exception('Failed to autoscale : %s', e)
if run_once or handler.shutdown_now:
return
with handler.safe_exit():
time.sleep(interval)
def test_scaler_always_up(monkeypatch):
api = MagicMock()
monkeypatch.setattr("kube_downscaler.scaler.helper.get_kube_api",
MagicMock(return_value=api))
def get(url, version, **kwargs):
if url == "pods":
data = {"items": []}
elif url == "deployments":
data = {
"items": [{
"metadata": {
"name": "deploy-1",
"namespace": "ns-1"
},
"spec": {
"replicas": 1
},
}]
}
elif url == "statefulsets":
data = {"items": []}
elif url == "stacks":
data = {"items": []}
elif url == "cronjobs":
data = {"items": []}
elif url == "namespaces/ns-1":
data = {"metadata": {}}
else:
raise Exception(f"unexpected call: {url}, {version}, {kwargs}")
response = MagicMock()
response.json.return_value = data
return response
api.get = get
include_resources = frozenset(
["statefulsets", "deployments", "stacks", "cronjobs"])
scale(
namespace=None,
upscale_period="never",
downscale_period="never",
default_uptime="always",
default_downtime="never",
include_resources=include_resources,
exclude_namespaces=[],
exclude_deployments=[],
exclude_statefulsets=[],
exclude_cronjobs=[],
dry_run=False,
grace_period=300,
downtime_replicas=0,
)
api.patch.assert_not_called()
def test_scaler_namespace_force_uptime_true(monkeypatch):
api = MagicMock()
monkeypatch.setattr("kube_downscaler.scaler.helper.get_kube_api",
MagicMock(return_value=api))
def get(url, version, **kwargs):
if url == "pods":
data = {"items": []}
elif url == "deployments":
data = {
"items": [
{
"metadata": {
"name": "deploy-1",
"namespace": "ns-1",
"creationTimestamp": "2019-03-01T16:38:00Z",
},
"spec": {
"replicas": 1
},
},
]
}
elif url == "namespaces/ns-1":
data = {
"metadata": {
"annotations": {
"downscaler/force-uptime": "true"
}
}
}
else:
raise Exception(f"unexpected call: {url}, {version}, {kwargs}")
response = MagicMock()
response.json.return_value = data
return response
api.get = get
include_resources = frozenset(["deployments"])
scale(
namespace=None,
upscale_period="never",
downscale_period="never",
default_uptime="never",
default_downtime="always",
include_resources=include_resources,
exclude_namespaces=[],
exclude_deployments=[],
dry_run=False,
grace_period=300,
)
assert api.patch.call_count == 0
def test_scaler_downscale_period_no_error(monkeypatch, caplog):
api = MagicMock()
monkeypatch.setattr("kube_downscaler.scaler.helper.get_kube_api",
MagicMock(return_value=api))
def get(url, version, **kwargs):
if url == "pods":
data = {"items": []}
elif url == "cronjobs":
data = {
"items": [
{
"metadata": {
"name": "cronjob-1",
"namespace": "default",
"creationTimestamp": "2019-03-01T16:38:00Z",
"annotations": {},
},
"spec": {
"suspend": False
},
},
]
}
elif url == "namespaces/default":
data = {"metadata": {}}
else:
raise Exception(f"unexpected call: {url}, {version}, {kwargs}")
response = MagicMock()
response.json.return_value = data
return response
api.get = get
include_resources = frozenset(["cronjobs"])
scale(
namespace=None,
upscale_period="never",
downscale_period="Mon-Tue 19:00-19:00 UTC",
default_uptime="always",
default_downtime="never",
include_resources=include_resources,
exclude_namespaces=[],
exclude_deployments=[],
exclude_statefulsets=[],
exclude_cronjobs=[],
dry_run=False,
grace_period=300,
downtime_replicas=0,
)
assert api.patch.call_count == 0
for record in caplog.records:
assert record.levelname != "ERROR"
def run_loop(
run_once,
namespace,
include_resources,
upscale_period,
downscale_period,
default_uptime,
default_downtime,
exclude_namespaces,
exclude_deployments,
grace_period,
interval,
dry_run,
downtime_replicas,
deployment_time_annotation=None,
enable_events=False,
upscale_step_size=0,
):
handler = shutdown.GracefulShutdown()
while True:
try:
scale(
namespace,
upscale_period,
downscale_period,
default_uptime,
default_downtime,
include_resources=frozenset(include_resources.split(",")),
exclude_namespaces=frozenset(
re.compile(pattern)
for pattern in exclude_namespaces.split(",")),
exclude_deployments=frozenset(exclude_deployments.split(",")),
dry_run=dry_run,
grace_period=grace_period,
downtime_replicas=downtime_replicas,
deployment_time_annotation=deployment_time_annotation,
enable_events=enable_events,
upscale_step_size=upscale_step_size,
)
except Exception as e:
logger.exception(f"Failed to autoscale: {e}")
if run_once or handler.shutdown_now:
return
with handler.safe_exit():
time.sleep(interval)
def run_loop(
run_once,
namespace,
include_resources,
upscale_period,
downscale_period,
default_uptime,
default_downtime,
exclude_namespaces,
exclude_deployments,
exclude_statefulsets,
exclude_cronjobs,
grace_period,
interval,
dry_run,
downtime_replicas,
deployment_time_annotation=None,
):
handler = shutdown.GracefulShutdown()
while True:
try:
scale(
namespace,
upscale_period,
downscale_period,
default_uptime,
default_downtime,
include_resources=frozenset(include_resources.split(",")),
exclude_namespaces=frozenset(exclude_namespaces.split(",")),
exclude_deployments=frozenset(exclude_deployments.split(",")),
exclude_statefulsets=frozenset(
exclude_statefulsets.split(",")),
exclude_cronjobs=frozenset(exclude_cronjobs.split(",")),
dry_run=dry_run,
grace_period=grace_period,
downtime_replicas=downtime_replicas,
deployment_time_annotation=deployment_time_annotation,
)
except Exception as e:
logger.exception("Failed to autoscale : %s", e)
if run_once or handler.shutdown_now:
return
with handler.safe_exit():
time.sleep(interval)
def test_scaler_name_excluded(monkeypatch):
api = MagicMock()
monkeypatch.setattr(
"kube_downscaler.scaler.helper.get_kube_api", MagicMock(return_value=api)
)
def get(url, version, **kwargs):
if url == "pods":
data = {"items": []}
elif url == "deployments":
data = {
"items": [
{
"metadata": {
"name": "sysdep-1",
"namespace": "system-ns",
"creationTimestamp": "2019-03-01T16:38:00Z",
},
"spec": {"replicas": 1},
},
{
"metadata": {
"name": "deploy-2",
"namespace": "default",
"creationTimestamp": "2019-03-01T16:38:00Z",
},
"spec": {"replicas": 2},
},
]
}
elif url == "namespaces/default":
data = {"metadata": {}}
else:
raise Exception(f"unexpected call: {url}, {version}, {kwargs}")
response = MagicMock()
response.json.return_value = data
return response
api.get = get
include_resources = frozenset(["deployments"])
scale(
namespace=None,
upscale_period="never",
downscale_period="never",
default_uptime="never",
default_downtime="always",
include_resources=include_resources,
exclude_namespaces=[],
exclude_deployments=["sysdep-1"],
dry_run=False,
grace_period=300,
)
assert api.patch.call_count == 1
# make sure that deploy-2 was updated (sysdep-1 was excluded)
patch_data = {
"metadata": {
"name": "deploy-2",
"namespace": "default",
"creationTimestamp": "2019-03-01T16:38:00Z",
"annotations": {ORIGINAL_REPLICAS_ANNOTATION: "2"},
},
"spec": {"replicas": 0},
}
assert api.patch.call_args[1]["url"] == "/deployments/deploy-2"
assert json.loads(api.patch.call_args[1]["data"]) == patch_data
def test_scaler_deployment_excluded_until(monkeypatch):
api = MagicMock()
monkeypatch.setattr(
"kube_downscaler.scaler.helper.get_kube_api", MagicMock(return_value=api)
)
one_day_in_future = datetime.datetime.utcnow() + datetime.timedelta(days=1)
def get(url, version, **kwargs):
if url == "pods":
data = {"items": []}
elif url == "deployments":
data = {
"items": [
{
"metadata": {
"name": "deploy-1",
"namespace": "my-ns",
"creationTimestamp": "2020-04-04T16:38:00Z",
"annotations": {"downscaler/exclude-until": "2040-01-01"},
},
"spec": {"replicas": 1},
},
{
"metadata": {
"name": "deploy-2",
"namespace": "my-ns",
"creationTimestamp": "2020-04-04T16:38:00Z",
"annotations": {"downscaler/exclude-until": "2020-04-04"},
},
"spec": {"replicas": 2},
},
{
"metadata": {
"name": "deploy-3",
"namespace": "my-ns",
"creationTimestamp": "2020-04-04T16:38:00Z",
"annotations": {
"downscaler/exclude-until": one_day_in_future.strftime(
"%Y-%m-%dT%H:%M:%SZ"
)
},
},
"spec": {"replicas": 3},
},
]
}
elif url == "namespaces/my-ns":
data = {"metadata": {}}
else:
raise Exception(f"unexpected call: {url}, {version}, {kwargs}")
response = MagicMock()
response.json.return_value = data
return response
api.get = get
include_resources = frozenset(["deployments"])
scale(
namespace=None,
upscale_period="never",
downscale_period="never",
default_uptime="never",
default_downtime="always",
include_resources=include_resources,
exclude_namespaces=[],
exclude_deployments=[],
dry_run=False,
grace_period=300,
)
assert api.patch.call_count == 1
# make sure that deploy-2 was updated (deploy-1 was excluded via annotation)
patch_data = {
"metadata": {
"name": "deploy-2",
"namespace": "my-ns",
"creationTimestamp": "2020-04-04T16:38:00Z",
"annotations": {
ORIGINAL_REPLICAS_ANNOTATION: "2",
"downscaler/exclude-until": "2020-04-04",
},
},
"spec": {"replicas": 0},
}
assert api.patch.call_args[1]["url"] == "/deployments/deploy-2"
assert json.loads(api.patch.call_args[1]["data"]) == patch_data
def test_scaler_cronjob_unsuspend(monkeypatch):
api = MagicMock()
monkeypatch.setattr(
"kube_downscaler.scaler.helper.get_kube_api", MagicMock(return_value=api)
)
def get(url, version, **kwargs):
if url == "pods":
data = {"items": []}
elif url == "cronjobs":
data = {
"items": [
{
"metadata": {
"name": "cronjob-1",
"namespace": "default",
"creationTimestamp": "2019-03-01T16:38:00Z",
"annotations": {ORIGINAL_REPLICAS_ANNOTATION: "1"},
},
"spec": {"suspend": True},
},
]
}
elif url == "namespaces/default":
data = {
"metadata": {
"annotations": {
"downscaler/uptime": "always",
"downscaler/downtime": "never",
}
}
}
# data = {'metadata': {}}
else:
raise Exception(f"unexpected call: {url}, {version}, {kwargs}")
response = MagicMock()
response.json.return_value = data
return response
api.get = get
include_resources = frozenset(["cronjobs"])
scale(
namespace=None,
upscale_period="never",
downscale_period="never",
default_uptime="never",
default_downtime="always",
include_resources=include_resources,
exclude_namespaces=[],
exclude_deployments=[],
dry_run=False,
grace_period=300,
downtime_replicas=0,
)
assert api.patch.call_count == 1
assert api.patch.call_args[1]["url"] == "/cronjobs/cronjob-1"
patch_data = {
"metadata": {
"name": "cronjob-1",
"namespace": "default",
"creationTimestamp": "2019-03-01T16:38:00Z",
"annotations": {ORIGINAL_REPLICAS_ANNOTATION: None},
},
"spec": {"suspend": False},
}
assert json.loads(api.patch.call_args[1]["data"]) == patch_data
def test_scaler_namespace_excluded_via_annotation(monkeypatch):
api = MagicMock()
monkeypatch.setattr('kube_downscaler.scaler.helper.get_kube_api',
MagicMock(return_value=api))
def get(url, version, **kwargs):
if url == 'pods':
data = {'items': []}
elif url == 'deployments':
data = {
'items': [{
'metadata': {
'name': 'deploy-1',
'namespace': 'ns-1',
'creationTimestamp': '2019-03-01T16:38:00Z'
},
'spec': {
'replicas': 1
}
}, {
'metadata': {
'name': 'deploy-2',
'namespace': 'ns-2',
'creationTimestamp': '2019-03-01T16:38:00Z'
},
'spec': {
'replicas': 2
}
}]
}
elif url == 'namespaces/ns-1':
data = {
'metadata': {
'annotations': {
'downscaler/exclude': 'true'
}
}
}
elif url == 'namespaces/ns-2':
data = {'metadata': {}}
else:
raise Exception(f'unexpected call: {url}, {version}, {kwargs}')
response = MagicMock()
response.json.return_value = data
return response
api.get = get
kinds = frozenset(['deployment'])
scale(namespace=None,
default_uptime='never',
default_downtime='always',
kinds=kinds,
exclude_namespaces=[],
exclude_deployments=[],
exclude_statefulsets=[],
dry_run=False,
grace_period=300)
assert api.patch.call_count == 1
# make sure that deploy-2 was updated (deploy-1 was excluded via annotation on ns-1)
patch_data = {
"metadata": {
"name": "deploy-2",
"namespace": "ns-2",
"creationTimestamp": "2019-03-01T16:38:00Z"
},
"spec": {
"replicas": 0
}
}
assert api.patch.call_args[1]['url'] == 'deployments/deploy-2'
assert json.loads(api.patch.call_args[1]['data']) == patch_data
def test_scaler_namespace_force_uptime_period(monkeypatch):
api = MagicMock()
monkeypatch.setattr("kube_downscaler.scaler.helper.get_kube_api",
MagicMock(return_value=api))
ORIGINAL_REPLICAS = 2
def get(url, version, **kwargs):
if url == "pods":
data = {"items": []}
elif url == "deployments":
data = {
"items": [
{
"metadata": {
"name": "deploy-1",
"namespace": "ns-1",
"creationTimestamp": "2019-03-01T16:38:00Z",
"annotations": {
ORIGINAL_REPLICAS_ANNOTATION:
ORIGINAL_REPLICAS,
},
},
"spec": {
"replicas": 0
},
},
{
"metadata": {
"name": "deploy-2",
"namespace": "ns-2",
"creationTimestamp": "2019-03-01T16:38:00Z",
"annotations": {
ORIGINAL_REPLICAS_ANNOTATION:
ORIGINAL_REPLICAS,
},
},
"spec": {
"replicas": 0
},
},
{
"metadata": {
"name": "deploy-3",
"namespace": "ns-3",
"creationTimestamp": "2019-03-01T16:38:00Z",
"annotations": {
ORIGINAL_REPLICAS_ANNOTATION:
ORIGINAL_REPLICAS,
},
},
"spec": {
"replicas": 0
},
},
]
}
elif url == "namespaces/ns-1":
# past period
data = {
"metadata": {
"annotations": {
"downscaler/force-uptime":
"2020-04-04T16:00:00+00:00-2020-04-05T16:00:00+00:00"
}
}
}
elif url == "namespaces/ns-2":
# current period
data = {
"metadata": {
"annotations": {
"downscaler/force-uptime":
"2020-04-04T16:00:00+00:00-2040-04-05T16:00:00+00:00"
}
}
}
elif url == "namespaces/ns-3":
# future period
data = {
"metadata": {
"annotations": {
"downscaler/force-uptime":
"2040-04-04T16:00:00+00:00-2040-04-05T16:00:00+00:00"
}
}
}
else:
raise Exception(f"unexpected call: {url}, {version}, {kwargs}")
response = MagicMock()
response.json.return_value = data
return response
api.get = get
include_resources = frozenset(["deployments"])
scale(
namespace=None,
upscale_period="never",
downscale_period="never",
default_uptime="never",
default_downtime="always",
include_resources=include_resources,
exclude_namespaces=[],
exclude_deployments=[],
dry_run=False,
grace_period=300,
)
# make sure that deploy-2 was updated
assert api.patch.call_count == 1
assert api.patch.call_args[1]["url"] == "/deployments/deploy-2"
assert (json.loads(api.patch.call_args[1]["data"])["spec"]["replicas"] ==
ORIGINAL_REPLICAS)
assert not json.loads(
api.patch.call_args[1]
["data"])["metadata"]["annotations"][ORIGINAL_REPLICAS_ANNOTATION]
