Python UnitTest-如何访问subTests消息而不必手动编写它们？

Question

A little bit of context: I am using Python and Selenium for QA Web Automation and I am using the Page Object Model pattern. I am currently in the process of making the logger part of the system, and I want to make the code more efficient, without having to write a lot of dublicate code. I checked the docs for subTests, but can't find anything in particular and this is the reason I am here. So I am wondering if there is a way to access this part of the code (so I don't have to write each logger message each time, which is not very practical):

  class TestScenario(unittest.TestCase):

      .... # set Up class

      def logger(self,info):
          logger.error(f"Error happened at {info}")

      def test_method(self):

          with self.subTest("MESSAGE"):---------------------------------------------                                     
                                                                                   |
              try:                                                                 |
                  ... something                                                    |
              except AssertionError:                                               | 
                  self.logger(#Is there a way to access subTest msg parameter?) <---
                  raise

          .... other subTests which will follow the same pattern

Thank you in advance!

Answer 1

You can do self.logger(self._subtest._message) to get the message.

Beware, you are accessing internals variable of unittest thus this code can be broken in the next release of python without warning.

Answer 2

Possible approach:

class Message:
    def __init__(self, message=''):
        self.message = message
    def __str__(self):
        return self.message

# separate message for every subtest or one message continually modified between subtests
msg = Message('qwe')

...
with self.subTest(msg=msg):
    msg.message = 'asd'  # access or modify
    self.assertTrue(True)
...

Or you may create attribute message in the setUp method of the test and use it to pass message into subTest and access/modify when you need.