Revision 244c552b snf-app/synnefo/api/tests.py

b/snf-app/synnefo/api/tests.py
49 49

  
50 50
class AaiClient(Client):
51 51
    def request(self, **request):
52
        request['HTTP_X_AUTH_TOKEN'] = \
53
            settings.BYPASS_AUTHENTICATION_SECRET_TOKEN
52
        request['HTTP_X_AUTH_TOKEN'] = '0000'
54 53
        return super(AaiClient, self).request(**request)
55 54

  
56 55

  
......
160 159
        request = {
161 160
                    "server": {
162 161
                        "name": "new-server-test",
163
                        "owner": 1,
162
                        "userid": "test",
164 163
                        "imageRef": 1,
165 164
                        "flavorRef": 1,
166 165
                        "metadata": {
......
342 341
        self.assertEqual(response.status_code, 201)
343 342

  
344 343

  
345
def create_users(n=1):
346
    for i in range(n):
347
        SynnefoUser.objects.create(name='User %d' % i)
348

  
349

  
350 344
def create_flavors(n=1):
351 345
    for i in range(n):
352 346
        Flavor.objects.create(
......
356 350

  
357 351

  
358 352
def create_images(n=1):
359
    owner = SynnefoUser.objects.all()[0]
360 353
    for i in range(n):
361 354
        Image.objects.create(
362 355
            name='Image %d' % (i + 1),
363 356
            state='ACTIVE',
364
            owner=owner)
357
            owner='test')
365 358

  
366 359

  
367 360
def create_image_metadata(n=1):
......
374 367

  
375 368

  
376 369
def create_servers(n=1):
377
    owner = SynnefoUser.objects.all()[0]
378 370
    flavors = Flavor.objects.all()
379 371
    images = Image.objects.all()
380 372
    for i in range(n):
381 373
        VirtualMachine.objects.create(
382 374
            name='Server %d' % (i + 1),
383
            owner=owner,
375
            owner='test',
384 376
            imageid=choice(images).id,
385 377
            hostid=str(i),
386 378
            flavor=choice(flavors))
......
410 402

  
411 403

  
412 404
class BaseTestCase(TestCase):
413
    USERS = 0
414 405
    FLAVORS = 1
415 406
    IMAGES = 1
416 407
    SERVERS = 1
......
420 411

  
421 412
    def setUp(self):
422 413
        self.client = AaiClient()
423
        create_users(self.USERS)
424 414
        create_flavors(self.FLAVORS)
425 415
        create_images(self.IMAGES)
426 416
        create_image_metadata(self.IMAGE_METADATA)

Also available in: Unified diff