External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-03-08 05:01:09 -02:30
Code Issues Packages Projects Releases Wiki Activity

use MagicMock for content_type

Browse Source
This commit is contained in:
Wayne Witzel III
2016-06-22 15:36:26 -04:00
parent 67e516f263
commit 1d329856f9
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
awx/main/tests/unit/api/test_views.py
View File

@@ -4,6 +4,8 @@ import pytest
from rest_framework.test import APIRequestFactory from rest_framework.test import APIRequestFactory
from rest_framework.test import force_authenticate from rest_framework.test import force_authenticate
from django.contrib.contenttypes.models import ContentType
from awx.api.views import ( from awx.api.views import (
ApiV1RootView, ApiV1RootView,
TeamRolesList, TeamRolesList,
@@ -11,6 +13,7 @@ from awx.api.views import (
from awx.main.models import ( from awx.main.models import (
User, User,
Role,
) )
@pytest.fixture @pytest.fixture
@@ -64,10 +67,11 @@ def test_team_roles_list_post_org_roles(url):
with mock.patch('awx.api.views.Role.objects.get', create=True) as role_get, \ with mock.patch('awx.api.views.Role.objects.get', create=True) as role_get, \
mock.patch('awx.api.views.ContentType.objects.get_for_model', create=True) as ct_get: mock.patch('awx.api.views.ContentType.objects.get_for_model', create=True) as ct_get:
role_mock = mock.MagicMock() role_mock = mock.MagicMock(spec=Role)
role_mock.content_type = 1 content_type_mock = mock.MagicMock(spec=ContentType)
role_mock.content_type = content_type_mock
role_get.return_value = role_mock role_get.return_value = role_mock
ct_get.return_value = 1 ct_get.return_value = content_type_mock
factory = APIRequestFactory() factory = APIRequestFactory()
view = TeamRolesList.as_view() view = TeamRolesList.as_view()