tests.py 文件源码

def test_vote_count(self):
        connection = ConnectionFactory()
        url = reverse('connection-detail', kwargs={"pk":connection.id})
        # Cast 5 votes
        VoteFactory.create_batch(size=5, connection=connection)
        response = self.client.get(url).json()
        self.assertEqual(response['vote_count'], 5)