async def test_framework_list_includes_completed_frameworks():
with patch.object(
master.MesosMaster, "_framework_list", autospec=True
) as mock_framework_list:
fake_frameworks = [{"name": "test_framework1"}, {"name": "test_framework2"}]
mock_framework_list.return_value = fake_frameworks
expected_frameworks = [
framework.Framework(config) for config in fake_frameworks
]
mesos_master = master.MesosMaster({})
assert expected_frameworks == await mesos_master.frameworks()
def test_framework_list_includes_completed_frameworks(mock_framework_list):
fake_frameworks = [
{
'name': 'test_framework1',
},
{
'name': 'test_framework2',
},
]
mock_framework_list.return_value = fake_frameworks
expected_frameworks = [framework.Framework(config) for config in fake_frameworks]
mesos_master = master.MesosMaster({})
assert expected_frameworks == mesos_master.frameworks()
async def test_frameworks():
with patch.object(master.MesosMaster, '_framework_list', autospec=True) as mock_framework_list:
fake_frameworks = [
{
'name': 'test_framework1',
},
{
'name': 'test_framework2',
},
]
mock_framework_list.return_value = fake_frameworks
expected_frameworks = [framework.Framework(config) for config in fake_frameworks]
mesos_master = master.MesosMaster({})
assert expected_frameworks == await mesos_master.frameworks()