From 1fc34a9f41a8a0fbddd83691f076874e5979dc37 Mon Sep 17 00:00:00 2001 From: Chris Meyers Date: Tue, 8 Nov 2016 10:19:06 -0500 Subject: [PATCH] mock capacity --- awx/main/tests/unit/scheduler/conftest.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/awx/main/tests/unit/scheduler/conftest.py b/awx/main/tests/unit/scheduler/conftest.py index 6b07649fd0..97eec0dbb4 100644 --- a/awx/main/tests/unit/scheduler/conftest.py +++ b/awx/main/tests/unit/scheduler/conftest.py @@ -22,9 +22,10 @@ def epoch(): @pytest.fixture def scheduler_factory(mocker, epoch): + mocker.patch('awx.main.models.Instance.objects.total_capacity', return_value=999999999) + def fn(tasks=[], inventory_sources=[], latest_project_updates=[], latest_inventory_updates=[], create_project_update=None, create_inventory_update=None): sched = TaskManager() - sched.capacity_total = 999999999 sched.graph.get_now = lambda: epoch