Removed print, fixed so inventory tests will still work without a valid license file.

This commit is contained in:
Chris Church
2013-06-30 12:53:25 -04:00
parent 6f0d644f5f
commit adcc402910
2 changed files with 8 additions and 6 deletions

View File

@@ -1,13 +1,16 @@
# Copyright (c) 2013 AnsibleWorks, Inc.
# All Rights Reserved.
import sys
import logging
from django.db.models import Q
from django.contrib.auth.models import User
from rest_framework.exceptions import PermissionDenied
from awx.main.models import *
from awx.main.licenses import LicenseReader
from django.core.exceptions import PermissionDenied
import sys
__all__ = ['get_user_queryset', 'check_user_access']
@@ -261,15 +264,15 @@ class HostAccess(BaseAccess):
reader = LicenseReader()
validation_info = reader.from_file()
if 'test' in sys.argv and 'free_instances' in validation_info:
if 'test' in sys.argv:# and 'free_instances' in validation_info:
# this hack is in here so the test code can function
# but still go down *most* of the license code path.
validation_info['free_instances'] = 99999999
if validation_info['free_instances'] > 0:
if validation_info.get('free_instances', 0) > 0:
# BOOKMARK
return True
instances = validation_info['available_instances']
instances = validation_info.get('available_instances', 0)
raise PermissionDenied("license range of %s instances has been exceed" % instances)
def can_change(self, obj, data):

View File

@@ -54,7 +54,6 @@ class LicenseTests(BaseTest):
vdata = reader.from_string(strdata)
print vdata
assert vdata['available_instances'] == 500
assert vdata['current_instances'] == 6
assert vdata['free_instances'] == 494