forked from loafle/openapi-generator-original
135 lines
4.3 KiB
Python
135 lines
4.3 KiB
Python
#!/usr/bin/env python
|
|
|
|
import sys
|
|
import unittest
|
|
import urllib.request, urllib.error, urllib.parse
|
|
import json
|
|
import random
|
|
|
|
from BaseApiTest import BaseApiTest
|
|
|
|
sys.path = ['./'] + sys.path
|
|
from petstore import *
|
|
from petstore.models import *
|
|
|
|
|
|
class PetApiTest(BaseApiTest):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
# super(PetApiTest, self).setUp()
|
|
cls.randomId = int(90000 * random.random()) + 10000
|
|
|
|
def testPetApis(self):
|
|
url = self.apiUrl + '/pet.json'
|
|
request = urllib.request.urlopen(url)
|
|
encoding = request.headers.get_content_charset()
|
|
if not encoding:
|
|
encoding = 'iso-8859-1'
|
|
response = request.read().decode(encoding)
|
|
doc = json.loads(response)
|
|
assert len(doc['apis']) == 3, 'there should be 3 pet apis'
|
|
|
|
def testPetApisAuthenticated(self):
|
|
url = self.apiUrl + '/pet.json?' + 'api_key=special-key'
|
|
request = urllib.request.urlopen(url)
|
|
encoding = request.headers.get_content_charset()
|
|
if not encoding:
|
|
encoding = 'iso-8859-1'
|
|
response = request.read().decode(encoding)
|
|
doc = json.loads(response)
|
|
|
|
assert len(doc['apis']) == 4, 'there should be 4 pet apis when' + \
|
|
'authenticated'
|
|
|
|
def testGetPetById(self):
|
|
res = self.petApi.getPetById(1)
|
|
assert res, 'null getWord result'
|
|
assert res.id == 1, 'pet id should be 1'
|
|
|
|
def testAddPet(self):
|
|
pet = Pet.Pet()
|
|
|
|
pet.id = self.randomId
|
|
tag1 = Tag.Tag()
|
|
tag1.name = "tag1"
|
|
tag2 = Tag.Tag()
|
|
tag2.name = "some tag"
|
|
pet.tags = [tag1, tag2]
|
|
category = Category.Category()
|
|
category.name = "Cats"
|
|
pet.category = category
|
|
pet.status = "sold"
|
|
pet.name = "Shermie"
|
|
pet.photoUrls = ["http://foo.com/1.jpg", "http://foo.com/1.jpg"]
|
|
self.petApi.addPet(pet)
|
|
|
|
new_pet = self.petApi.getPetById(pet.id)
|
|
|
|
assert new_pet.id == pet.id, 'ids should match'
|
|
assert new_pet.name == pet.name, 'names should match'
|
|
assert(set([tag.name for tag in new_pet.tags]) ==
|
|
set([tag.name for tag in pet.tags])), 'tags should match'
|
|
assert new_pet.status == pet.status, 'status should match'
|
|
assert new_pet.category.name == pet.category.name, 'category should match'
|
|
assert new_pet.photoUrls == pet.photoUrls, 'photoUrls should match'
|
|
|
|
def testUpdatePet(self):
|
|
alpahbet = list('ABCDEFGHIJKLMNOPQRSTUVWXYZ')
|
|
|
|
pet = Pet.Pet()
|
|
pet.id = self.randomId
|
|
|
|
tag1 = Tag.Tag()
|
|
tag1.name = "special-tag"
|
|
tag2 = Tag.Tag()
|
|
random.shuffle(alpahbet)
|
|
tag2.name = ''.join(alpahbet)
|
|
pet.tags = [tag1, tag2]
|
|
category = Category.Category()
|
|
random.shuffle(alpahbet)
|
|
category.name = ''.join(alpahbet)
|
|
pet.category = category
|
|
pet.status = "sold"
|
|
random.shuffle(alpahbet)
|
|
pet.name = ''.join(alpahbet)
|
|
pet.photoUrls = ["http://foo.com/22.jpg", "http://foo.com/55.jpg"]
|
|
self.petApi.updatePet(pet)
|
|
|
|
updated_pet = self.petApi.getPetById(pet.id)
|
|
|
|
assert updated_pet.id == pet.id, 'ids should match'
|
|
assert updated_pet.name == pet.name, 'names should match'
|
|
assert(set([tag.name for tag in updated_pet.tags]) ==
|
|
set([tag.name for tag in pet.tags])), 'tags should match'
|
|
assert updated_pet.status == pet.status, 'status should match'
|
|
assert updated_pet.category.name == pet.category.name, 'category should match'
|
|
assert updated_pet.photoUrls == pet.photoUrls, 'photoUrls should match'
|
|
|
|
def testFindPetsByTags(self):
|
|
pet = Pet.Pet()
|
|
pet.id = self.randomId
|
|
|
|
tag1 = Tag.Tag()
|
|
tag1.name = "special-tag"
|
|
pet.tags = [tag1]
|
|
self.petApi.updatePet(pet)
|
|
|
|
res = self.petApi.findPetsByTags("special-tag")
|
|
assert self.randomId in [pet.id for pet in res], 'must find by tag'
|
|
|
|
def testFindPetsByStatus(self):
|
|
pet = Pet.Pet()
|
|
pet.id = self.randomId
|
|
|
|
tag1 = Tag.Tag()
|
|
tag1.name = "special-tag"
|
|
pet.status = "sold"
|
|
self.petApi.updatePet(pet)
|
|
|
|
res = self.petApi.findPetsByStatus("sold")
|
|
assert self.randomId in [pet.id for pet in res], 'must find by status'
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|