def test__frameworks(mock_fetch): mesos_master = master.MesosMaster({}) mock_frameworks = Mock() mock_fetch.return_value = Mock(json=Mock(return_value=mock_frameworks)) ret = mesos_master._frameworks mock_fetch.assert_called_with(mesos_master, "/master/frameworks") assert ret == mock_frameworks
async def test__frameworks(): with patch.object(master.MesosMaster, 'fetch', autospec=True) as mock_fetch: mesos_master = master.MesosMaster({}) mock_frameworks = Mock() mock_fetch.return_value = CoroutineMock(json=CoroutineMock(return_value=mock_frameworks)) ret = await mesos_master._frameworks() mock_fetch.assert_called_with( mesos_master, "/master/frameworks", cached=True, ) assert ret == mock_frameworks
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()
def test_tasks(mock__task_list, mock_task): mock_task_1 = {'id': 'aaa'} mock_task_2 = {'id': 'bbb'} mock__task_list.return_value = [mock_task_1, mock_task_2] mock_task.return_value = Mock() mesos_master = master.MesosMaster({}) ret = mesos_master.tasks() mock_task.assert_has_calls( [call(mesos_master, mock_task_1), call(mesos_master, mock_task_2)]) mock__task_list.assert_called_with(mesos_master, False) expected = [mock_task.return_value, mock_task.return_value] assert list(ret) == expected
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()
def test__framework_list(mock__frameworks): mock_frameworks = Mock() mock_completed = Mock() mock__frameworks.__get__ = Mock(return_value={ 'frameworks': [mock_frameworks], 'completed_frameworks': [mock_completed], }) mesos_master = master.MesosMaster({}) ret = mesos_master._framework_list() expected = [mock_frameworks, mock_completed] assert list(ret) == expected ret = mesos_master._framework_list(active_only=True) expected = [mock_frameworks] assert list(ret) == expected
async def test_tasks(): with patch.object( master.MesosMaster, "_task_list", autospec=True ) as mock__task_list, patch.object(task, "Task", autospec=True) as mock_task: mock_task_1 = {"id": "aaa"} mock_task_2 = {"id": "bbb"} mock__task_list.return_value = [mock_task_1, mock_task_2] mock_task.return_value = Mock() mesos_master = master.MesosMaster({}) ret = await mesos_master.tasks() mock_task.assert_has_calls( [call(mesos_master, mock_task_1), call(mesos_master, mock_task_2)] ) mock__task_list.assert_called_with(mesos_master, False) expected = [mock_task.return_value, mock_task.return_value] assert list(ret) == expected
def test__task_list(mock__frameworks_list): mock_task_1 = Mock() mock_task_2 = Mock() mock_framework = {'tasks': [mock_task_1], 'completed_tasks': [mock_task_2]} mock__frameworks_list.return_value = [mock_framework] mesos_master = master.MesosMaster({}) ret = mesos_master._task_list() mock__frameworks_list.assert_called_with(mesos_master, False) expected = [mock_task_1, mock_task_2] assert list(ret) == expected ret = mesos_master._task_list(active_only=True) expected = [mock_task_1] assert list(ret) == expected ret = mesos_master._task_list(active_only=False) expected = [mock_task_1, mock_task_2] assert list(ret) == expected
async def test__framework_list(): mock_frameworks = Mock() mock_completed = Mock() with patch.object( master.MesosMaster, '_frameworks', autospec=True, return_value={ 'frameworks': [mock_frameworks], 'completed_frameworks': [mock_completed], }, ): mesos_master = master.MesosMaster({}) ret = await mesos_master._framework_list() expected = [mock_frameworks, mock_completed] assert list(ret) == expected ret = await mesos_master._framework_list(active_only=True) expected = [mock_frameworks] assert list(ret) == expected
async def test_tasks(): with patch.object( master.MesosMaster, '_task_list', autospec=True, ) as mock__task_list, patch.object( task, 'Task', autospec=True, ) as mock_task: mock_task_1 = {'id': 'aaa'} mock_task_2 = {'id': 'bbb'} mock__task_list.return_value = [mock_task_1, mock_task_2] mock_task.return_value = Mock() mesos_master = master.MesosMaster({}) ret = await mesos_master.tasks() mock_task.assert_has_calls([ call(mesos_master, mock_task_1), call(mesos_master, mock_task_2), ]) mock__task_list.assert_called_with(mesos_master, False) expected = [mock_task.return_value, mock_task.return_value] assert list(ret) == expected
async def test__task_list(): mock_task_1 = Mock() mock_task_2 = Mock() mock_framework = { 'tasks': [mock_task_1], 'completed_tasks': [mock_task_2], } with patch.object( master.MesosMaster, '_framework_list', autospec=True, return_value=[mock_framework], ) as mock__frameworks_list: mesos_master = master.MesosMaster({}) ret = await mesos_master._task_list() mock__frameworks_list.assert_called_with(mesos_master, False) expected = [mock_task_1, mock_task_2] assert list(ret) == expected ret = await mesos_master._task_list(active_only=True) expected = [mock_task_1] assert list(ret) == expected ret = await mesos_master._task_list(active_only=False) expected = [mock_task_1, mock_task_2] assert list(ret) == expected
async def test_orphan_tasks(): mesos_master = master.MesosMaster({}) mock_task_1 = Mock() mesos_master.state = CoroutineMock(return_value={'orphan_tasks': [mock_task_1]}) assert await mesos_master.orphan_tasks() == [mock_task_1]
def test_orphan_tasks(): mesos_master = master.MesosMaster({}) mock_task_1 = Mock() mesos_master.state = {'orphan_tasks': [mock_task_1]} assert mesos_master.orphan_tasks() == [mock_task_1]