def test_digest(self, sha256): p = '1234' digest = Policy.digest(p) sha256.assert_called_once_with() sha256.return_value.update.assert_called_once_with(p) self.assertEqual(digest, sha256.return_value.hexdigest.return_value)