Fixed key errors in membership test.

This commit is contained in:
Robert Kolner committed 2013-02-20 23:01:57 +01:00
1 parent c9dd5418d1
commit d4524c9b13
2 files changed
+4 -2

No files matched your search

+2
View File
@@ -26,4 +26,6 @@ class MembershipResource(ModelResource):
def dehydrate(self, bundle):
bundle.data['expires'] = str(bundle.obj.expires().date())
bundle.data['member'] = bundle.obj.member.id
bundle.data['mtype'] = bundle.obj.mtype.id
bundle.data['payment'] = bundle.obj.payment.id
return bundle
+2 -2
View File
@@ -20,10 +20,10 @@ class MembershipTest(ResourceTestCase):
u'resource_uri'
])
self.assertNotEquals(None,membership['id'])
self.assertNotEquals(None,membership['start_type'])
self.assertNotEquals(None,membership['start_date'])
self.assertNotEquals(None,membership['expires'])
self.assertNotEquals(None,membership['mtype'])
self.assertNotEquals(None,membership['membership'])
self.assertNotEquals(None,membership['member'])
self.assertNotEquals(None,membership['created'])
self.assertNotEquals(None,membership['updated'])
self.assertNotEquals(None,membership['resource_uri'])