diff --git a/awx/main/tests/licenses.py b/awx/main/tests/licenses.py index 4e9ce8db03..7606c40200 100644 --- a/awx/main/tests/licenses.py +++ b/awx/main/tests/licenses.py @@ -9,7 +9,7 @@ from django.contrib.auth.models import User as DjangoUser import django.test from django.test.client import Client from django.core.urlresolvers import reverse -from awx.main.models import * +from awx.main.models import Host, Inventory, Organization from awx.main.tests.base import BaseTest from awx.main.licenses import * @@ -17,6 +17,16 @@ class LicenseTests(BaseTest): def setUp(self): super(LicenseTests, self).setUp() + self.setup_users() + u = self.super_django_user + org = Organization.objects.create(name='o1', created_by=u) + inventory = Inventory.objects.create(name='hi', organization=org, created_by=u) + host = Host.objects.create(name='a1', inventory=inventory, created_by=u) + host = Host.objects.create(name='a2', inventory=inventory, created_by=u) + host = Host.objects.create(name='a3', inventory=inventory, created_by=u) + host = Host.objects.create(name='a4', inventory=inventory, created_by=u) + host = Host.objects.create(name='a5', inventory=inventory, created_by=u) + host = Host.objects.create(name='a6', inventory=inventory, created_by=u) def test_license_writer(self): @@ -28,10 +38,31 @@ class LicenseTests(BaseTest): instance_count=500 ) - print writer.get_data() - print writer.get_string() + data = writer.get_data() + + assert data['instance_count'] == 500 + assert data['contact_name'] == 'Michael DeHaan' + assert data['contact_email'] == 'michael@ansibleworks.com' + assert data['license_date'] == 25000 + assert data['license_key'] == "11bae31f31c6a6cdcb483a278cdbe98bd8ac5761acd7163a50090b0f098b3a13" + + strdata = writer.get_string() + strdata_loaded = json.loads(strdata) + assert strdata_loaded == data + + reader = LicenseReader() + + vdata = reader.from_string(strdata) + + print vdata + assert vdata['available_instances'] == 500 + assert vdata['current_instances'] == 6 + assert vdata['free_instances'] == 494 + assert vdata['date_warning'] == True + assert vdata['date_expired'] == True + assert vdata['license_date'] == 25000 + assert vdata['time_remaining'] < 0 + assert vdata['valid_key'] == True + assert vdata['compliant'] == False - assert 2 == 4 - def test_license_reader(self): - assert 2 == 3