def execute(self): # Check if username is unique, otherwise barf member_exists = len(list(Member.objects.raw({'_id': self.username}))) != 0 if member_exists: return # instantiate & save new member m = Member(username=self.username, slack_id=self.slack_id) m.save()
def test_persistence(self): # Arrange Config().connect_to_db() m = Member(TestMember.test_username) # Act m.save() # Assert retrieved_members = Member.objects.raw( {'_id': TestMember.test_username}) self.assertEqual(1, retrieved_members.count()) self.assertEqual(m, retrieved_members[0]) # Tear down m.delete()