update inventory tests to use real host counts.

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