class TestMemberServicesIntegration(unittest.TestCase):
"""Unit test for the MemberService type
"""
layer = INTEGRATION_TESTING
def setUp(self):
self.portal = self.layer['portal']
self.services = self.portal._getOb('products_and_services')
self.intids = queryUtility(IIntIds, context=self.portal)
self.dao = MemberServicesDataAccess(self.portal)
setRoles(self.portal, TEST_USER_ID, ['Member'])
self.session = SESSION()
def tearDown(self):
count = self.session.query(MemberService).filter(
MemberService.memberid == TEST_USER_ID
).delete()
print 'Deleted %s' % count
transaction.commit()
def test_get_all_memberservices(self):
db_ms = self.session.query(MemberService).all()
all_ms = self.dao.get_all_memberservices()
self.assertEquals(len(db_ms), len(all_ms))
self.assertEquals(all_ms, db_ms)
def test_get_memberservices(self):
service = self.services.objectValues()[0]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms1 = self.dao.add_memberservice(**ms_args)
self.session.flush()
ms1_db = self.get_memberservice(ms1.id)
service_uids = [self.intids.getId(service),]
memberid = TEST_USER_ID
memberservices = \
self.dao.get_memberservices(service_uids, memberid)
self.assertEquals(memberservices, [ms1_db])
def test_get_memberservices_by_subject(self):
maths_services = \
[s for s in self.services.objectValues() if 'maths' in s.subject]
for count in range(0,3):
service = maths_services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
self.dao.add_memberservice(**ms_args)
self.session.flush()
science_services = \
[s for s in self.services.objectValues() if 'science' in s.subject]
for count in range(0,3):
service = science_services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
self.dao.add_memberservice(**ms_args)
self.session.flush()
memberservices = \
self.dao.get_memberservices_by_subject(TEST_USER_ID, 'maths')
self.assertEqual(len(memberservices), 3)
memberservices = \
self.dao.get_memberservices_by_subject(TEST_USER_ID, 'science')
self.assertEqual(len(memberservices), 3)
def test_get_memberservices_by_grade(self):
grade_10_services = \
[s for s in self.services.objectValues() if s.grade == 'grade-10']
for count in range(0,3):
service = grade_10_services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
self.dao.add_memberservice(**ms_args)
self.session.flush()
memberservices = \
self.dao.get_memberservices_by_grade(TEST_USER_ID, 'grade-10')
self.assertEqual(len(memberservices), 3)
def test_get_memberservices_by_subject_and_grade(self):
grade_10_maths_services = \
[s for s in self.services.objectValues() if s.grade == 'grade-10']
grade_10_maths_services = \
[s for s in grade_10_maths_services if 'maths' in s.subject]
for count in range(0,3):
service = grade_10_maths_services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
self.dao.add_memberservice(**ms_args)
self.session.flush()
memberservices = \
self.dao.get_memberservices_by_subject_and_grade(TEST_USER_ID,
'maths',
'grade-10')
self.assertEqual(len(memberservices), 3)
memberservices = \
self.dao.get_memberservices_by_subject_and_grade(TEST_USER_ID,
'science',
'grade-10')
self.assertEqual(len(memberservices), 0)
def test_get_active_memberservices(self):
for count in range(0,3):
service = self.services.objectValues()[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
self.dao.add_memberservice(**ms_args)
self.session.flush()
now = datetime.now().date()
td = timedelta(1)
yesterday = now - td
for count in range(3,6):
service = self.services.objectValues()[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = yesterday
self.dao.add_memberservice(**ms_args)
self.session.flush()
active_memberservices = self.dao.get_active_memberservices(TEST_USER_ID)
self.assertEquals(len(active_memberservices), 3)
for ms in active_memberservices:
self.assertEquals(ms.expiry_date, now)
def test_get_active_memberservices_by_subject(self):
now = datetime.now().date()
td = timedelta(1)
yesterday = now - td
services = \
[s for s in self.services.objectValues() if 'maths' in s.subject]
for count in range(0,3):
service = services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = now
self.dao.add_memberservice(**ms_args)
self.session.flush()
services = \
[s for s in self.services.objectValues() if 'science' in s.subject]
for count in range(0,3):
service = services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = yesterday
self.dao.add_memberservice(**ms_args)
self.session.flush()
memberservices = \
self.dao.get_active_memberservices_by_subject(TEST_USER_ID, 'maths')
self.assertEqual(len(memberservices), 3)
for ms in memberservices:
self.assertEquals(ms.expiry_date, now)
def test_get_active_memberservices_by_grade(self):
now = datetime.now().date()
td = timedelta(1)
yesterday = now - td
services = \
[s for s in self.services.objectValues() if s.grade == 'grade-10']
for count in range(0,3):
service = services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = now
self.dao.add_memberservice(**ms_args)
self.session.flush()
services = \
[s for s in self.services.objectValues() if s.grade == 'grade-11']
for count in range(0,3):
service = services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = yesterday
self.dao.add_memberservice(**ms_args)
self.session.flush()
memberservices = \
self.dao.get_active_memberservices_by_grade(TEST_USER_ID, 'grade-10')
self.assertEqual(len(memberservices), 3)
for ms in memberservices:
self.assertEquals(ms.expiry_date, now)
def test_get_active_memberservices_by_subject_and_grade(self):
now = datetime.now().date()
td = timedelta(1)
yesterday = now - td
grade_10_maths_services = \
[s for s in self.services.objectValues() if \
s.grade == 'grade-10' and 'maths' in s.subject]
for count in range(0,3):
service = grade_10_maths_services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = now
self.dao.add_memberservice(**ms_args)
self.session.flush()
grade_10_science_services = \
[s for s in self.services.objectValues() if \
s.grade == 'grade-10' and 'science' in s.subject]
for count in range(0,3):
service = grade_10_science_services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = yesterday
self.dao.add_memberservice(**ms_args)
self.session.flush()
memberservices = \
self.dao.get_memberservices_by_subject_and_grade(TEST_USER_ID,
'maths',
'grade-10')
self.assertEqual(len(memberservices), 3)
for ms in memberservices:
self.assertEquals(ms.expiry_date, now)
def test_get_expired_memberservices(self):
now = datetime.now().date()
for count in range(0,3):
service = self.services.objectValues()[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = now
self.dao.add_memberservice(**ms_args)
self.session.flush()
td = timedelta(1)
yesterday = now - td
for count in range(3,6):
service = self.services.objectValues()[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = yesterday
self.dao.add_memberservice(**ms_args)
self.session.flush()
expired_memberservices = self.dao.get_expired_memberservices(TEST_USER_ID)
self.assertEquals(len(expired_memberservices), 3)
for ms in expired_memberservices:
self.assertEquals(ms.expiry_date, yesterday)
def test_get_expired_memberservices_by_subject(self):
now = datetime.now().date()
td = timedelta(1)
yesterday = now - td
services = \
[s for s in self.services.objectValues() if 'maths' in s.subject]
for count in range(0,3):
service = services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = now
self.dao.add_memberservice(**ms_args)
self.session.flush()
services = \
[s for s in self.services.objectValues() if 'science' in s.subject]
for count in range(0,3):
service = services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = yesterday
self.dao.add_memberservice(**ms_args)
self.session.flush()
memberservices = \
self.dao.get_expired_memberservices_by_subject(TEST_USER_ID, 'science')
self.assertEqual(len(memberservices), 3)
for ms in memberservices:
self.assertEquals(ms.expiry_date, yesterday)
def test_get_expired_memberservices_by_grade(self):
now = datetime.now().date()
td = timedelta(1)
yesterday = now - td
services = \
[s for s in self.services.objectValues() if s.grade == 'grade-10']
for count in range(0,3):
service = services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = now
self.dao.add_memberservice(**ms_args)
self.session.flush()
services = \
[s for s in self.services.objectValues() if s.grade == 'grade-11']
for count in range(0,3):
service = services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = yesterday
self.dao.add_memberservice(**ms_args)
self.session.flush()
memberservices = \
self.dao.get_expired_memberservices_by_grade(TEST_USER_ID,
'grade-11')
self.assertEqual(len(memberservices), 3)
for ms in memberservices:
self.assertEquals(ms.expiry_date, yesterday)
def test_get_expired_memberservices_by_subject_and_grade(self):
now = datetime.now().date()
td = timedelta(1)
yesterday = now - td
grade_10_maths_services = \
[s for s in self.services.objectValues() if \
s.grade == 'grade-10' and 'maths' in s.subject]
for count in range(0,3):
service = grade_10_maths_services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = now
self.dao.add_memberservice(**ms_args)
self.session.flush()
grade_10_science_services = \
[s for s in self.services.objectValues() if \
s.grade == 'grade-10' and 'science' in s.subject]
for count in range(0,3):
service = grade_10_science_services[count]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms_args['expiry_date'] = yesterday
self.dao.add_memberservice(**ms_args)
self.session.flush()
memberservices = \
self.dao.get_expired_memberservices_by_subject_and_grade(
TEST_USER_ID, 'science', 'grade-10')
self.assertEqual(len(memberservices), 3)
for ms in memberservices:
self.assertEquals(ms.expiry_date, yesterday)
def test_get_memberservice_by_primary_key(self):
service = self.services.objectValues()[0]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms1 = self.dao.add_memberservice(**ms_args)
self.session.flush()
ms1_db = self.dao.get_memberservice_by_primary_key(ms1.id)
self.assertEquals(ms1_db, self.get_memberservice(ms1.id))
def test_add_memberservice(self):
service = self.services.objectValues()[0]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms1 = self.dao.add_memberservice(**ms_args)
self.session.flush()
ms1_db = self.get_memberservice(ms1.id)
self.failUnless(IMemberService.providedBy(ms1_db))
def test_adding_duplicate_memberservices(self):
service = self.services.objectValues()[0]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms1 = self.dao.add_memberservice(**ms_args)
ms2 = self.dao.add_memberservice(**ms_args)
self.session.flush()
ms1_db = self.get_memberservice(ms1.id)
ms2_db = self.get_memberservice(ms2.id)
self.failUnless(IMemberService.providedBy(ms1_db))
def test_update_memberservice(self):
service = self.services.objectValues()[0]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms1 = self.dao.add_memberservice(**ms_args)
self.session.flush()
ms1_db = self.get_memberservice(ms1.id)
ms1_db.title = 'new title'
self.dao.update_memberservice(ms1_db)
self.session.flush()
ms1_db = self.get_memberservice(ms1.id)
self.assertEquals(ms1_db.title, 'new title')
def test_delete_memberservice(self):
service = self.services.objectValues()[0]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms1 = self.dao.add_memberservice(**ms_args)
self.session.flush()
ms1_db = self.get_memberservice(ms1.id)
self.dao.delete_memberservice(ms1_db)
ms1_db = self.get_memberservice(ms1.id)
self.assertEquals(ms1_db, None)
def test_related_service(self):
service = self.services.objectValues()[0]
ms_args = self.get_ms_args(service, TEST_USER_ID)
ms1 = self.dao.add_memberservice(**ms_args)
self.session.flush()
ms1_db = self.get_memberservice(ms1.id)
self.failUnless(self.dao.related_service(ms1_db), service)
def test_fti(self):
fti = queryUtility(IDexterityFTI, name='emas.app.memberservice')
self.assertEquals(None, fti)
def test_schema(self):
fti = queryUtility(IDexterityFTI, name='emas.app.memberservice')
schema = fti.lookupSchema()
self.assertEquals(None, schema)
def test_factory(self):
fti = queryUtility(IDexterityFTI, name='emas.app.memberservice')
factory = fti.factory
self.assertEquals(factory, None)
def get_memberservice(self, memberservice_id):
memberservices = self.session.query(MemberService).filter_by(
id = memberservice_id).all()
return memberservices and memberservices[0] or None
def create_services(self):
subjects = ['maths', 'science']
grades = ['10', '11', '12']
for subject in subjects:
for count, grade in grades:
props = {'service_type' : 'subscription',
'grade' : grade,
'subject' : subject,
'price' : '111'}
s_id = self.services.invokeFactory('emas.app.service',
'%s-%s' % (subject, grade),
**props)
service = self.services._getOb(s_id)
service.subject = props['subject']
def get_ms_args(self, service, memberid):
ms_args = {
'memberid': memberid,
'title': '%s for %s' % (service.title, memberid),
'related_service_id': self.intids.getId(service),
'expiry_date': datetime.now(),
}
return ms_args
class List_Services(grok.View):
"""
Custom view for mobile practice service
"""
grok.context(Interface)
grok.require('zope2.View')
grok.name('list-services')
def __init__(self, context, request):
super(List_Services, self).__init__(context, request)
self.pps = self.context.restrictedTraverse('@@plone_portal_state')
self.portal = self.pps.portal()
self.portal_url = self.pps.portal_url()
self.member = self.pps.member()
self.memberservices = []
self.isMxit = self.context.restrictedTraverse('@@mobile_tool').isMXit()
self.navroot = self.pps.navigation_root()
self.subject = self.getSubject(self.navroot)
self.services = self._getServices(self.portal, self.subject)
self.dao = MemberServicesDataAccess(self.context)
memberid = self.member.getId()
if memberid:
ids = practice_service_intids(self.context)
self.memberservices = \
self.dao.get_memberservices_by_subject(memberid, self.subject)
def getSubject(self, navroot):
return navroot.getId()
def craftPaidLink(self, navroot, service):
access_path = service.access_path
link = ['%s' % service.Title(),
'%s/%s' % (navroot.absolute_url(), access_path)]
return link
def craftNotPaidLink(self, navroot, service):
if self.isMxit:
link = ['%s' % service.Title(),
'%s/@@mxitpaymentrequest?productId=%s' % (
navroot.absolute_url(), service.getId())]
else:
link = ['%s' % service.Title(),
'%s/@@purchase?productId=%s' % (
navroot.absolute_url(), service.getId())]
return link
def getURLs(self):
urls = {
'paid': [],
'notpaid': []
}
if self.isMxit:
paid, notpaid = self.getMXitServices(
self.services, self.memberservices)
else:
paid, notpaid = self.getWebAndMobileServices(
self.services, self.memberservices)
urls['paid'] = \
[self.craftPaidLink(self.navroot, s) for s in paid]
urls['notpaid'] = \
[self.craftNotPaidLink(self.navroot, s) for s in notpaid]
return urls
def _getServices(self, portal, subject):
products_and_services = portal.products_and_services
services = products_and_services.getFolderContents(
full_objects=True,
contentFilter={'portal_type': 'emas.app.service',
'subject': subject})
return services
def getMXitServices(self, services, memberservices):
""" Probably good idea to @memoize this.
"""
paid = []
notpaid = []
related_services = \
[s.related_service(self.context) for s in memberservices]
for service in services:
# we are only interested in the mxit services now
if MXIT_SERVICE in service.channels:
if service in related_services:
paid.append(service)
else:
notpaid.append(service)
return paid, notpaid
def getWebAndMobileServices(self, services, memberservices):
""" Probably good idea to @memoize this too.
"""
paid = []
notpaid = []
related_services = \
[s.related_service(self.context) for s in memberservices]
for service in services:
if MXIT_SERVICE not in service.channels:
if service in related_services:
paid.append(service)
else:
notpaid.append(service)
return paid, notpaid
def isPracticeService(self):
"""
"""
path = self.context.getPhysicalPath()
return PRACTICE_URL.split('/')[-1] in path
