Revision 13bf6cd8 snf-pithos-app/pithos/api/test/__init__.py

b/snf-pithos-app/pithos/api/test/__init__.py
37 37
from urlparse import urlunsplit, urlsplit, urlparse
38 38
from xml.dom import minidom
39 39
from urllib import quote, unquote
40
from mock import patch, PropertyMock
40 41

  
41 42
from snf_django.utils.testing import with_settings, astakos_user
42 43

  
......
46 47

  
47 48
from synnefo.lib.services import get_service_path
48 49
from synnefo.lib import join_urls
50
from synnefo.util import text
49 51

  
52
from django.core.urlresolvers import reverse
50 53
from django.test import TestCase
51 54
from django.test.client import Client, MULTIPART_CONTENT, FakePayload
52 55
from django.test.simple import DjangoTestSuiteRunner
......
56 59

  
57 60
import django.utils.simplejson as json
58 61

  
62

  
59 63
import random
60 64
import functools
61 65

  
......
202 206

  
203 207

  
204 208
class PithosAPITest(TestCase):
209
    def create_patch(self, name, new_callable=None):
210
        patcher = patch(name, new_callable=new_callable)
211
        thing = patcher.start()
212
        self.addCleanup(patcher.stop)
213
        return thing
214

  
205 215
    def setUp(self):
206 216
        self.client = PithosTestClient()
207 217

  
......
213 223
        self.pithos_path = join_urls(get_service_path(
214 224
            pithos_settings.pithos_services, 'object-store'))
215 225

  
226
        # patch astakosclient.AstakosClient.validate_token
227
        mock_validate_token = self.create_patch(
228
            'astakosclient.AstakosClient.validate_token')
229
        mock_validate_token.return_value = {
230
            'access': {'user': {'id': text.udec(self.user, 'utf8')}}}
231

  
232
        # patch astakosclient.AstakosClient.get_token
233
        mock_get_token = self.create_patch(
234
            'astakosclient.AstakosClient.get_token')
235
        mock_get_token.return_value = {'access_token': 'valid_token'}
236

  
237
        # patch astakosclient.AstakosClient.api_oa2_auth
238
        mock_api_oa2_auth = self.create_patch(
239
            'astakosclient.AstakosClient.api_oa2_auth',
240
            new_callable=PropertyMock)
241
        mock_api_oa2_auth.return_value = reverse('oa2_authenticate')
242

  
243
        # patch astakosclient.AstakosClient.api_oa2_token
244
        mock_api_oa2_token = self.create_patch(
245
            'astakosclient.AstakosClient.api_oa2_token',
246
            new_callable=PropertyMock)
247
        mock_api_oa2_token.return_value = reverse('oa2_token')
248

  
216 249
    def tearDown(self):
217 250
        #delete additionally created metadata
218 251
        meta = self.get_account_meta()

Also available in: Unified diff