def test_user_has_permission_product_type_member_success(self, mock_get): other_user = User() other_user.id = 2 product_type_member_other_user = Product_Type_Member() product_type_member_other_user.id = 2 product_type_member_other_user.user = other_user product_type_member_other_user.product_type = self.product_type product_type_member_other_user.role = Roles.Owner mock_get.return_value = product_type_member_other_user result = user_has_permission(other_user, self.product_type_member_reader, Permissions.Product_Type_Member_Delete) self.assertTrue(result) self.assertEqual(mock_get.call_args[1]['user'], other_user) self.assertEqual(mock_get.call_args[1]['product_type'], self.product_type)
def test_user_has_permission_product_type_member_success(self, mock_foo): other_user = User() other_user.id = 2 product_type_member_other_user = Product_Type_Member() product_type_member_other_user.id = 2 product_type_member_other_user.user = other_user product_type_member_other_user.product_type = self.product_type product_type_member_other_user.role = Role.objects.get(id=Roles.Owner) mock_foo.select_related.return_value = mock_foo mock_foo.select_related.return_value = mock_foo mock_foo.filter.return_value = [product_type_member_other_user] result = user_has_permission(other_user, self.product_type_member_reader, Permissions.Product_Type_Member_Delete) self.assertTrue(result) mock_foo.filter.assert_called_with(user=other_user)