コード例 #1
0
    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()
コード例 #2
0
    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()