class Property(Base):
formFieldPrefix = 'property_'
fields = {
'propertyKey': '',
'partnerKey': '',
'propName': '',
'propInfo': PropInfo(),
'address': Location(),
'contactName': Name(),
'contactInfo': Contact(),
'rooms': [],
'reviews': [],
'rating': 0,
'totalBooked': 0
}
def getKey(self):
return self['propertyKey']
def generatePropertyKey(self):
import random
# generate property key
first4 = self['propName']
first4 = first4.replace(' ', '')
first4 = first4[0:4]
first4 = first4.ljust(4, '_')
self['propertyKey'] = first4.upper() + str(random.randint(1000, 9999))
return self['propertyKey']
class PropBooking(Base) :
fields = {
'propertyKey' : '',
'propertyName' : '',
'propertyAddr' : Location(),
'propertyContact' : Contact()
}
class CustBooking(Base) :
fields = {
'customerKey' : '',
'customerName' : Name(),
'customerAddr' : Location(),
'customerContact' : Contact(),
'custParty' : []
}
def setUp(self):
self.testDict['name'] = Name(self.testName)
self.testDict['address'] = Location(self.testLocation)
self.testDict['contact'] = Contact(self.testContact)
self.testDict['otherContact'] = OtherContact(self.testOtherContact)
self.testDict['otherInfo'] = OtherInfo(self.testOtherInfo)
self.testDict['login'] = LoginInfo(self.testLoginInfo)
self.customerFromDict = Customer(self.testDict)
self.customerDefaults = Customer()
self.maxDiff = None
def setUp(self):
self.propertyDefaults = Property()
self.testDict['propInfo'] = PropInfo(self.testPropInfo)
self.testDict['address'] = Location(self.testLocation)
self.testDict['contactName'] = Name(self.testName)
self.testDict['contactInfo'] = Contact(self.testContact)
self.testDict['rooms'] = [
RoomType(self.testRoomType1),
RoomType(self.testRoomType2)
]
self.testDict['reviews'] = [Review(self.testReview)]
self.propertyFromDict = Property(self.testDict)
self.maxDiff = None
class Customer(Base):
fields = {
'customerKey': '',
'name': Name(),
'address': Location(),
'contact': Contact(),
'otherContact': OtherContact(),
'otherInfo': OtherInfo(),
'login': LoginInfo()
}
def getKey(self):
return self['customerKey']
def setUp(self):
custBooking = {
'customerKey':
'00000000',
'customerName':
Name(self.testCustName),
'customerAddr':
Location(self.testCustLocation),
'customerContact':
Contact(self.testCustContact),
'custParty': [
CustParty({
'name': Name(self.testCustPartyName1),
'other': OtherInfo(self.testOtherInfo1)
}),
CustParty({
'name': Name(self.testCustPartyName2),
'other': OtherInfo(self.testOtherInfo2)
})
]
}
propBooking = {
'propertyKey': 'ZZZZ9999',
'propertyName': 'The Slate Resort',
'propertyAddr': Location(self.testPropLocation),
'propertyContact': Contact(self.testPropContact),
}
self.testDict = {
'bookingKey': '20190501AAAA1111',
'customer': custBooking,
'property': propBooking,
'bookingInfo': BookingInfo(self.testBookingInfo),
'rooms': [RoomBooking(self.testRoomBooking)],
'totalPrice': 999.99
}
self.bookingFromDict = Booking(self.testDict)
self.bookingDefaults = Booking()
self.maxDiff = None
def getPropertyFromForms(self):
# get property form
prop_form = self.forms['prop']
# create data dictionary
data = {
'partnerKey': prop_form.cleaned_data.get('prop_partnerKey'),
'propName': prop_form.cleaned_data.get('prop_propName'),
'propInfo': PropInfo(self.getCleanDataByKey('info')),
'address': Location(self.getCleanDataByKey('location')),
'contactName': Name(self.getCleanDataByKey('name')),
'contactInfo': Contact(self.getCleanDataByKey('contact')),
}
# create and return Property instance
return Property(data)
class Partner(Base):
fields = {
'partnerKey': '',
'businessName': '',
'revenueSplit': 0.00,
'acctBalance': 0.00,
'name': Name(),
'address': Location(),
'contact': Contact(),
'otherContact': OtherContact(),
'otherInfo': OtherInfo(),
'login': LoginInfo()
}
def getKey(self):
return self['partnerKey']
class Property(Base):
fields = {
'propertyKey': '',
'partnerKey': '',
'propName': '',
'propInfo': PropInfo(),
'address': Location(),
'contactName': Name(),
'contactInfo': Contact(),
'rooms': [],
'reviews': [],
'rating': 0,
'totalBooked': 0
}
def getKey(self):
return self['propertyKey']
def assemble(self, doc):
prop = Property(doc)
if 'propInfo' in doc:
prop['propInfo'] = PropInfo(doc['propInfo'])
if 'address' in doc:
prop['address'] = Location(doc['address'])
if 'contactName' in doc:
prop['contactName'] = Name(doc['contactName'])
if 'contactInfo' in doc:
prop['contactInfo'] = Contact(doc['contactInfo'])
if 'rooms' in doc:
prop['rooms'] = []
for room in doc['rooms']:
prop['rooms'].append(RoomType(room))
if 'reviews' in doc:
prop['reviews'] = []
for review in doc['reviews']:
prop['reviews'].append(Review(room))
return prop
class TestCustomer(unittest.TestCase):
customerFromDict = None
customerDefaults = None
testDict = {
'customerKey': '00000000',
'name': Name(),
'address': Location(),
'contact': Contact(),
'otherContact': OtherContact(),
'otherInfo': OtherInfo(),
'login': LoginInfo(),
'totalBooked': 100,
'discount': 0.05
}
testLocation = {
'streetAddress': '123 Main Street',
'buildingName': '',
'floor': '',
'roomAptCondoFlat': '',
'city': 'Bedrock',
'stateProvince': 'ZZ',
'locality': 'Unknown',
'country': 'None',
'postalCode': '00000',
'latitude': '+111.111',
'longitude': '-111.111'
}
testName = {
'title': 'Mr',
'first': 'Fred',
'middle': 'Folsom',
'last': 'Flintstone',
'suffix': 'CM'
}
testContact = {
'email': '*****@*****.**',
'phone': '+0-000-000-0000',
'socMedia': {
'google': '*****@*****.**'
}
}
testOtherContact = {
'emails': ['*****@*****.**', '*****@*****.**'],
'phoneNumbers': [{
'home': '000-000-0000'
}, {
'work': '111-111-1111'
}],
'socMedias': [{
'google': '*****@*****.**'
}, {
'skype': 'fflintstone'
}]
}
testOtherInfo = {'gender': 'M', 'dateOfBirth': '0000-01-01'}
testLoginInfo = {
'username': '******',
'oauth2': '*****@*****.**',
'password': '******'
}
testJson = '''{
"customerKey" : "00000000",
"address" : {
"streetAddress" : "123 Main Street",
"buildingName" : "",
"floor" : "",
"roomAptCondoFlat" : "",
"city" : "Bedrock",
"stateProvince" : "ZZ",
"locality" : "Unknown",
"country" : "None",
"postalCode" : "00000",
"latitude" : "+111.111",
"longitude" : "-111.111"
},
"name" : {
"title" : "Mr",
"first" : "Fred",
"middle" : "Folsom",
"last" : "Flintstone",
"suffix" : "CM"
},
"contact" : {
"email" : "*****@*****.**",
"phone" : "+0-000-000-0000",
"socMedia" : {"google" : "*****@*****.**"}
},
"otherContact" : {
"emails" : ["*****@*****.**", "*****@*****.**"],
"phoneNumbers" : [{"home" : "000-000-0000"}, {"work" : "111-111-1111"}],
"socMedias" : [{"google" : "*****@*****.**"}, {"skype" : "fflintstone"}]
},
"otherInfo" : {
"gender" : "M",
"dateOfBirth" : "0000-01-01"
},
"login" : {
"username" : "fred",
"oauth2" : "*****@*****.**",
"password" : "abcdefghijklmnopqrstuvwxyz"
},
"totalBooked" : 100,
"discount" : 0.05
}'''
def setUp(self):
self.testDict['name'] = Name(self.testName)
self.testDict['address'] = Location(self.testLocation)
self.testDict['contact'] = Contact(self.testContact)
self.testDict['otherContact'] = OtherContact(self.testOtherContact)
self.testDict['otherInfo'] = OtherInfo(self.testOtherInfo)
self.testDict['login'] = LoginInfo(self.testLoginInfo)
self.customerFromDict = Customer(self.testDict)
self.customerDefaults = Customer()
self.maxDiff = None
def test_customer_key(self):
expected = '00000000'
actual = self.customerFromDict.getKey()
self.assertEqual(expected, actual)
def test_customer_from_dict(self):
expected = 'Flintstone'
name = self.customerFromDict.get('name')
actual = name.get('last')
self.assertEqual(expected, actual)
def test_customer_from_blank(self):
expected = True
actual = isinstance(self.customerDefaults.get('address'), Location)
self.assertEqual(expected, actual)
def test_customer_from_dict_to_json(self):
expected = json.loads(self.testJson)
actual = json.loads(self.customerFromDict.toJson())
self.assertEqual(expected, actual)
class TestProperty(unittest.TestCase):
propertyFromDict = None
propertyDefaults = None
testDict = {
'propertyKey': 'AAAA1111',
'partnerKey': 'BBBB2222',
'propName': 'Test',
'propInfo': PropInfo(),
'address': Location(),
'contactName': Name(),
'contactInfo': Contact(),
'rooms': [],
'reviews': [],
'rating': 3,
'totalBooked': 100
}
testPropInfo = {
'type': 'hotel',
'chain': 'Accor',
'photos': ['image1.jpg', 'image2.jpg'],
'facilities': ['free parking', 'wifi'],
'description':
'Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.',
'currency': 'GBP',
'taxFee': 0.11
}
testLocation = {
'streetAddress': '123 Main Street',
'buildingName': 'The Gravel Building',
'floor': '22',
'roomAptCondoFlat': '',
'city': 'Bedrock',
'stateProvince': 'ZZ',
'locality': 'Unknown',
'country': 'None',
'postalCode': '00000',
'latitude': '+111.111',
'longitude': '-111.111'
}
testName = {
'title': 'Mr',
'first': 'Fred',
'middle': 'Folsom',
'last': 'Flintstone',
'suffix': 'CM'
}
testContact = {
'email': '*****@*****.**',
'phone': '+0-000-000-0000',
'socMedia': {
'google': '*****@*****.**'
}
}
testRoomType1 = {
'roomTypeKey': 'aa11',
'type': 'Pool Side',
'view': 'Pool',
'description':
'Cupcake ipsum dolor. Sit amet marshmallow topping cheesecake muffin. Halvah croissant candy canes bonbon candy. Apple pie jelly beans topping carrot cake danish tart cake cheesecake.',
'beds': ['single', 'single'],
'numAvailable': 22,
'numBooked': 11,
'price': 44.44
}
testRoomType2 = {
'roomTypeKey': 'bb22',
'type': 'Standard',
'view': 'City',
'description':
'Lorem ipsum dolor sit amet consectetur adipiscing elit, urna consequat felis vehicula class ultricies mollis dictumst, aenean non a in donec nulla.',
'beds': ['king'],
'numAvailable': 12,
'numBooked': 6,
'price': 33.33
}
testReview = {
'customerKey':
'AAAABBBB0000',
'staff':
1,
'cleanliness':
2,
'facilities':
3,
'comfort':
4,
'goodStuff':
"All of the words in Lorem Ipsum have flirted with me - consciously or unconsciously. That's to be expected. Be careful, or I will spill the beans on your placeholder text. I’m the best thing that ever happened to placeholder text. Look at that text! Would anyone use that? Can you imagine that, the text of your next webpage?!",
'badStuff':
'Despite the constant negative ipsum covfefe. Lorem Ispum is a choke artist. It chokes!'
}
testJson = '''{
"propertyKey" : "AAAA1111",
"partnerKey" : "BBBB2222",
"propName" : "Test",
"propInfo" : {
"type" : "hotel",
"chain" : "Accor",
"photos" : ["image1.jpg","image2.jpg"],
"facilities" : ["free parking","wifi"],
"description" : "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.",
"currency" : "GBP",
"taxFee" : 0.11
},
"address" : {
"streetAddress" : "123 Main Street",
"buildingName" : "The Gravel Building",
"floor" : "22",
"roomAptCondoFlat" : "",
"city" : "Bedrock",
"stateProvince" : "ZZ",
"locality" : "Unknown",
"country" : "None",
"postalCode" : "00000",
"latitude" : "+111.111",
"longitude" : "-111.111"
},
"contactName" : {
"title" : "Mr",
"first" : "Fred",
"middle" : "Folsom",
"last" : "Flintstone",
"suffix" : "CM"
},
"contactInfo" : {
"email" : "*****@*****.**",
"phone" : "+0-000-000-0000",
"socMedia" : {"google" : "*****@*****.**"}
},
"rooms" : [
{
"roomTypeKey" : "aa11",
"type" : "Pool Side",
"view" : "Pool",
"description" : "Cupcake ipsum dolor. Sit amet marshmallow topping cheesecake muffin. Halvah croissant candy canes bonbon candy. Apple pie jelly beans topping carrot cake danish tart cake cheesecake.",
"beds" : ["single","single"],
"numAvailable" : 22,
"numBooked" : 11,
"price" : 44.44
},
{
"roomTypeKey" : "bb22",
"type" : "Standard",
"view" : "City",
"description" : "Lorem ipsum dolor sit amet consectetur adipiscing elit, urna consequat felis vehicula class ultricies mollis dictumst, aenean non a in donec nulla.",
"beds" : ["king"],
"numAvailable" : 12,
"numBooked" : 6,
"price" : 33.33
}
],
"reviews" : [
{
"customerKey" : "AAAABBBB0000",
"staff" : 1,
"cleanliness" : 2,
"facilities" : 3,
"comfort" : 4,
"goodStuff" : "All of the words in Lorem Ipsum have flirted with me - consciously or unconsciously. That's to be expected. Be careful, or I will spill the beans on your placeholder text. I’m the best thing that ever happened to placeholder text. Look at that text! Would anyone use that? Can you imagine that, the text of your next webpage?!",
"badStuff" : "Despite the constant negative ipsum covfefe. Lorem Ispum is a choke artist. It chokes!"
}
],
"rating" : 3,
"totalBooked" : 100
}'''
def setUp(self):
self.propertyDefaults = Property()
self.testDict['propInfo'] = PropInfo(self.testPropInfo)
self.testDict['address'] = Location(self.testLocation)
self.testDict['contactName'] = Name(self.testName)
self.testDict['contactInfo'] = Contact(self.testContact)
self.testDict['rooms'] = [
RoomType(self.testRoomType1),
RoomType(self.testRoomType2)
]
self.testDict['reviews'] = [Review(self.testReview)]
self.propertyFromDict = Property(self.testDict)
self.maxDiff = None
def test_property_key(self):
expected = 'AAAA1111'
actual = self.propertyFromDict.getKey()
self.assertEqual(expected, actual)
def test_property_from_dict(self):
expected = 'Flintstone'
name = self.propertyFromDict.get('contactName')
actual = name.get('last')
self.assertEqual(expected, actual)
def test_property_from_dict_book_room(self):
roomList = self.propertyFromDict.get('rooms')
# test booking room type 1
expected = 21
actual = roomList[0].bookOneRoom()
self.assertEqual(expected, actual)
# test booking room type 2
expected = 11
actual = roomList[1].bookOneRoom()
self.assertEqual(expected, actual)
def test_property_from_blank(self):
expected = True
actual = isinstance(self.propertyDefaults.get('propInfo'), PropInfo)
self.assertEqual(expected, actual)
def test_property_from_dict_to_json(self):
expected = json.loads(self.testJson)
actual = json.loads(self.propertyFromDict.toJson())
self.assertEqual(expected, actual)
class TestBooking(unittest.TestCase):
bookingFromDict = None
bookingDefaults = None
testDict = {
'bookingKey': '20190501AAAA1111',
'customer': CustBooking(),
'property': PropBooking(),
'bookingInfo': BookingInfo(),
'rooms': [],
'totalPrice': 999.99
}
testCustBooking = {
'customerKey': '00000000',
'customerName': Name(),
'customerAddr': Location(),
'customerContact': Contact(),
'custParty': []
}
testCustName = {
'title': 'Ms',
'first': 'Wilma',
'middle': 'Slaghoople',
'last': 'Flintstone',
'suffix': ''
}
testCustLocation = {
'streetAddress': '345 Cave Stone Road',
'buildingName': '',
'floor': '',
'roomAptCondoFlat': '',
'city': 'Bedrock',
'stateProvince': 'ZZ',
'locality': 'Unknown',
'country': 'None',
'postalCode': '00000',
'latitude': '+111.1111',
'longitude': '-111.1111'
}
testCustContact = {
'email': '*****@*****.**',
'phone': '+0-000-000-0000',
'socMedia': {
'google': '*****@*****.**'
}
}
testCustParty: [{
'name': Name(),
'other': OtherInfo()
}, {
'name': Name(),
'other': OtherInfo()
}]
testCustPartyName1 = {
'title': 'Mr',
'first': 'Frederick',
'middle': 'Joseph',
'last': 'Flintstone',
'suffix': 'Caveman'
}
testCustPartyName2 = {
'title': 'Baby',
'first': 'Bam',
'middle': 'Bam',
'last': 'Flintstone',
'suffix': ''
}
testOtherInfo1 = {'gender': 'M', 'dateOfBirth': '0000-00-00'}
testOtherInfo2 = {'gender': 'M', 'dateOfBirth': '0030-01-01'}
testPropBooking = {
'propertyKey': 'ZZZZ9999',
'propertyName': 'The Slate Resort',
'propertyAddr': Location(),
'propertyContact': Contact()
}
testPropLocation = {
'streetAddress': '140 Boulder Avenue',
'buildingName': 'The Slate Resort',
'floor': '',
'roomAptCondoFlat': '',
'city': 'Granite Town',
'stateProvince': 'XX',
'locality': 'Unknown',
'country': 'None',
'postalCode': '00001',
'latitude': '+222.2222',
'longitude': '-222.2222'
}
testPropContact = {
'email': '*****@*****.**',
'phone': '+0-000-000-0000',
'socMedia': {
'google': '*****@*****.**'
}
}
testBookingInfo = {
'arrivalDate': '0040-03-15',
'departureDate': '0040-04-01',
'checkoutTime': '12:00',
'refundableUntil': '0040-03-28',
'reservationStatus': 'pending',
'paymentStatus': 'pending'
}
testRoomBooking = {
'roomType': 'poolside',
'roomTypeKey': '0001',
'price': 99.99,
'qty': 1
}
testJson = '''{
"bookingKey" : "20190501AAAA1111",
"customer" : {
"customerKey" : "00000000",
"customerName" : {
"title" : "Ms",
"first" : "Wilma",
"middle" : "Slaghoople",
"last" : "Flintstone",
"suffix" : ""
},
"customerAddr" : {
"streetAddress" : "345 Cave Stone Road",
"buildingName" : "",
"floor" : "",
"roomAptCondoFlat" : "",
"city" : "Bedrock",
"stateProvince" : "ZZ",
"locality" : "Unknown",
"country" : "None",
"postalCode" : "00000",
"latitude" : "+111.1111",
"longitude" : "-111.1111"
},
"customerContact" : {
"email" : "*****@*****.**",
"phone" : "+0-000-000-0000",
"socMedia" : {"google" : "*****@*****.**"}
},
"custParty" : [
{
"name" : {
"title" : "Mr",
"first" : "Frederick",
"middle" : "Joseph",
"last" : "Flintstone",
"suffix" : "Caveman"
},
"other" : {
"gender" : "M",
"dateOfBirth" : "0000-00-00"
}
},
{
"name" : {
"title" : "Baby",
"first" : "Bam",
"middle" : "Bam",
"last" : "Flintstone",
"suffix" : ""
},
"other" : {
"gender" : "M",
"dateOfBirth" : "0030-01-01"
}
}
]
},
"property" : {
"propertyKey" : "ZZZZ9999",
"propertyName" : "The Slate Resort",
"propertyAddr" : {
"streetAddress" : "140 Boulder Avenue",
"buildingName" : "The Slate Resort",
"floor" : "",
"roomAptCondoFlat" : "",
"city" : "Granite Town",
"stateProvince" : "XX",
"locality" : "Unknown",
"country" : "None",
"postalCode" : "00001",
"latitude" : "+222.2222",
"longitude" : "-222.2222"
},
"propertyContact" : {
"email" : "*****@*****.**",
"phone" : "+0-000-000-0000",
"socMedia" : {"google" : "*****@*****.**"}
}
},
"bookingInfo" : {
"arrivalDate" : "0040-03-15",
"departureDate" : "0040-04-01",
"checkoutTime" : "12:00",
"refundableUntil" : "0040-03-28",
"reservationStatus" : "pending",
"paymentStatus" : "pending"
},
"rooms" : [
{
"roomType" : "poolside",
"roomTypeKey" : "0001",
"price" : 99.99,
"qty" : 1
}
],
"totalPrice" : 999.99
}'''
def setUp(self):
custBooking = {
'customerKey':
'00000000',
'customerName':
Name(self.testCustName),
'customerAddr':
Location(self.testCustLocation),
'customerContact':
Contact(self.testCustContact),
'custParty': [
CustParty({
'name': Name(self.testCustPartyName1),
'other': OtherInfo(self.testOtherInfo1)
}),
CustParty({
'name': Name(self.testCustPartyName2),
'other': OtherInfo(self.testOtherInfo2)
})
]
}
propBooking = {
'propertyKey': 'ZZZZ9999',
'propertyName': 'The Slate Resort',
'propertyAddr': Location(self.testPropLocation),
'propertyContact': Contact(self.testPropContact),
}
self.testDict = {
'bookingKey': '20190501AAAA1111',
'customer': custBooking,
'property': propBooking,
'bookingInfo': BookingInfo(self.testBookingInfo),
'rooms': [RoomBooking(self.testRoomBooking)],
'totalPrice': 999.99
}
self.bookingFromDict = Booking(self.testDict)
self.bookingDefaults = Booking()
self.maxDiff = None
def test_booking_key(self):
expected = '20190501AAAA1111'
actual = self.bookingFromDict.getKey()
self.assertEqual(expected, actual)
def test_booking_from_dict(self):
expected = 'Flintstone'
custBook = self.bookingFromDict.get('customer')
custName = custBook.get('customerName')
actual = custName.get('last')
self.assertEqual(expected, actual)
def test_booking_from_blank(self):
expected = True
actual = isinstance(self.bookingDefaults.get('property'), PropBooking)
self.assertEqual(expected, actual)
def test_booking_from_dict_to_json(self):
expected = json.loads(self.testJson)
actual = json.loads(self.bookingFromDict.toJson())
self.assertEqual(expected, actual)
class TestPartner(unittest.TestCase):
partnerFromDict = None
partnerDefaults = None
testDict = {
'partnerKey': '00000000',
'businessName': 'Granite Town Quarrel and Gravel Company',
'revenueSplit': 0.90,
'acctBalance': 9999.99,
'name': Name(),
'address': Location(),
'contact': Contact(),
'otherContact': OtherContact(),
'otherInfo': OtherInfo(),
'login': LoginInfo()
}
testLocation = {
'streetAddress': '140 Boulder Avenue',
'buildingName': 'The Granite Building',
'floor': '22',
'roomAptCondoFlat': '',
'city': 'Granite Town',
'stateProvince': 'ZZ',
'locality': 'Unknown',
'country': 'None',
'postalCode': '00000',
'latitude': '+111.111',
'longitude': '-111.111'
}
testName = {
'title': 'Mr',
'first': 'Rockhead',
'middle': 'Sylvester',
'last': 'Slate',
'suffix': 'Boss'
}
testContact = {
'email': '*****@*****.**',
'phone': '+0-000-000-0000',
'socMedia': {
'google': '*****@*****.**'
}
}
testOtherContact = {
'emails': ['*****@*****.**', '*****@*****.**'],
'phoneNumbers': [{
'home': '000-000-0000'
}, {
'work': '111-111-1111'
}],
'socMedias': [{
'google': '*****@*****.**'
}, {
'skype': 'fslate'
}]
}
testOtherInfo = {'gender': 'M', 'dateOfBirth': '0000-01-01'}
testLoginInfo = {
'username': '******',
'oauth2': '*****@*****.**',
'password': '******'
}
testJson = '''{
"partnerKey" : "00000000",
"businessName" : "Granite Town Quarrel and Gravel Company",
"revenueSplit" : 0.90,
"acctBalance" : 9999.99,
"address" : {
"streetAddress" : "140 Boulder Avenue",
"buildingName" : "The Granite Building",
"floor" : "22",
"roomAptCondoFlat" : "",
"city" : "Granite Town",
"stateProvince" : "ZZ",
"locality" : "Unknown",
"country" : "None",
"postalCode" : "00000",
"latitude" : "+111.111",
"longitude" : "-111.111"
},
"name" : {
"title" : "Mr",
"first" : "Rockhead",
"middle" : "Sylvester",
"last" : "Slate",
"suffix" : "Boss"
},
"contact" : {
"email" : "*****@*****.**",
"phone" : "+0-000-000-0000",
"socMedia" : {"google" : "*****@*****.**"}
},
"otherContact" : {
"emails" : ["*****@*****.**", "*****@*****.**"],
"phoneNumbers" : [{"home" : "000-000-0000"}, {"work" : "111-111-1111"}],
"socMedias" : [{"google" : "*****@*****.**"}, {"skype" : "fslate"}]
},
"otherInfo" : {
"gender" : "M",
"dateOfBirth" : "0000-01-01"
},
"login" : {
"username" : "rockhead",
"oauth2" : "*****@*****.**",
"password" : "abcdefghijklmnopqrstuvwxyz"
}
}'''
def setUp(self):
self.testDict['name'] = Name(self.testName)
self.testDict['address'] = Location(self.testLocation)
self.testDict['contact'] = Contact(self.testContact)
self.testDict['otherContact'] = OtherContact(self.testOtherContact)
self.testDict['otherInfo'] = OtherInfo(self.testOtherInfo)
self.testDict['login'] = LoginInfo(self.testLoginInfo)
self.partnerFromDict = Partner(self.testDict)
self.partnerDefaults = Partner()
self.maxDiff = None
def test_partner_key(self):
expected = '00000000'
actual = self.partnerFromDict.getKey()
self.assertEqual(expected, actual)
def test_partner_from_dict(self):
expected = 'Slate'
name = self.partnerFromDict.get('name')
actual = name.get('last')
self.assertEqual(expected, actual)
def test_partner_from_blank(self):
expected = True
actual = isinstance(self.partnerDefaults.get('address'), Location)
self.assertEqual(expected, actual)
def test_partner_from_dict_to_json(self):
expected = json.loads(self.testJson)
actual = json.loads(self.partnerFromDict.toJson())
self.assertEqual(expected, actual)