def setUp(self): super(CodeResourceViewMockTests, self).setUp() patcher = mocked_relations(KiveUser, CodeResource, CodeResourceRevision, User, Group) patcher.start() self.addCleanup(patcher.stop) # noinspection PyUnresolvedReferences patchers = [patch.object(CodeResource._meta, 'default_manager', CodeResource.objects), patch.object(CodeResourceRevision._meta, 'default_manager', CodeResource.objects)] def dummy_save(r): r.id = id(r) # noinspection PyUnresolvedReferences patchers.append(patch.object(CodeResource, 'save', dummy_save)) patcher = PatcherChain(patchers) patcher.start() self.addCleanup(patcher.stop) self.client = self.create_client() self.dev_group = Group(pk=groups.DEVELOPERS_PK) self.everyone = Group(pk=groups.EVERYONE_PK) Group.objects.add(self.dev_group, self.everyone) self.user = kive_user() self.user.groups.add(self.dev_group) self.content_file = ContentFile('some text', 'existing.txt') self.code_resource = CodeResource(pk='99', user=self.user, name='existing', filename='existing.txt') self.code_resource._state.adding = False self.other_user = User(pk=5) self.other_code_resource = CodeResource(pk='150', user=self.other_user) CodeResource.objects.add(self.code_resource, self.other_code_resource) self.code_resource_revision = CodeResourceRevision( pk='199', user=self.user, content_file=self.content_file) self.code_resource_revision.coderesource = self.code_resource self.other_code_resource_revision = CodeResourceRevision( pk='200', user=self.other_user) self.other_code_resource_revision.coderesource = self.other_code_resource self.other_code_resource.revisions.add(self.other_code_resource_revision) CodeResourceRevision.objects.add(self.code_resource_revision, self.other_code_resource_revision) k = KiveUser(pk=users.KIVE_USER_PK) k.groups.add(self.dev_group) KiveUser.objects.add(k)
def test_start(self): patcher = PatcherChain([PatcherChainTest.patch_mock_max, PatcherChainTest.patch_zero_sum]) mocked = patcher.start() self.addCleanup(patcher.stop) sum_result = sum([1, 2, 3]) mocked[0].return_value = 42 max_result = max([1, 2, 3]) self.assertEqual(0, sum_result) self.assertEqual(42, max_result) self.assertEqual(2, len(mocked)) self.assertIs(zero_sum, mocked[1])
def test_start(self): patcher = PatcherChain( [PatcherChainTest.patch_mock_max, PatcherChainTest.patch_zero_sum]) mocked = patcher.start() self.addCleanup(patcher.stop) sum_result = sum([1, 2, 3]) mocked[0].return_value = 42 max_result = max([1, 2, 3]) self.assertEqual(0, sum_result) self.assertEqual(42, max_result) self.assertEqual(2, len(mocked)) self.assertIs(zero_sum, mocked[1])
def test_context_manager(self): with PatcherChain([PatcherChainTest.patch_mock_max, PatcherChainTest.patch_zero_sum]) as mocked: sum_result = sum([1, 2, 3]) mocked[0].return_value = 42 max_result = max([1, 2, 3]) self.assertEqual(0, sum_result) self.assertEqual(42, max_result) self.assertEqual(2, len(mocked)) self.assertIs(zero_sum, mocked[1])