From d4524c9b13479346b66571fd245763a7059277dd Mon Sep 17 00:00:00 2001 From: Robert Kolner Date: Wed, 20 Feb 2013 23:01:57 +0100 Subject: [PATCH] Fixed key errors in membership test. --- dusken/main/api/membership.py | 2 ++ dusken/main/tests/membership.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/dusken/main/api/membership.py b/dusken/main/api/membership.py index 26dd937..2349cff 100644 --- a/dusken/main/api/membership.py +++ b/dusken/main/api/membership.py @@ -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 diff --git a/dusken/main/tests/membership.py b/dusken/main/tests/membership.py index a9ab578..56d7f49 100644 --- a/dusken/main/tests/membership.py +++ b/dusken/main/tests/membership.py @@ -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'])