Module osbot_utils.testing.Patch_Print
Expand source code
from unittest.mock import patch, MagicMock, _patch
from osbot_utils.utils.Dev import pprint
from osbot_utils.base_classes.Kwargs_To_Self import Kwargs_To_Self
class Patch_Print(Kwargs_To_Self):
enabled : bool = True
expected_calls : list
mocked_print : MagicMock
patched_print : _patch
print_calls : bool
def __init__(self, **kwargs):
super().__init__(**kwargs)
def __enter__(self):
if self.enabled:
self.patched_print = patch('builtins.print')
self.mocked_print = self.patched_print.start()
return self
def __exit__(self, exc_type, exc_val, exc_tb):
if self.enabled:
self.patched_print.stop()
if self.print_calls:
pprint(self.calls())
#print(self.mocked_print.call_args_list)
if self.expected_calls:
assert self.calls() == self.expected_calls
def call_args_list(self):
if self.mocked_print:
return self.mocked_print.call_args_list
return []
def calls(self):
calls_data = []
if self.mocked_print:
for call in self.mocked_print.call_args_list:
if len(call.args) == 0 and call.kwargs == {}:
call_data = ''
elif len(call.args) == 1 and call.kwargs == {}:
call_data = call.args[0]
else:
call_data = (call.args, call.kwargs)
calls_data.append(call_data)
return calls_data
Classes
class Patch_Print (**kwargs)
-
A mixin class to strictly assign keyword arguments to pre-defined instance attributes during initialization.
This base class provides an init method that assigns values from keyword arguments to instance attributes. If an attribute with the same name as a key from the kwargs is defined in the class, it will be set to the value from kwargs. If the key does not match any predefined attribute names, an exception is raised.
This behavior enforces strict control over the attributes of instances, ensuring that only predefined attributes can be set at the time of instantiation and avoids silent attribute creation which can lead to bugs in the code.
Usage
class MyConfigurableClass(Kwargs_To_Self): attribute1 = 'default_value' attribute2 = True attribute3 : str attribute4 : list attribute4 : int = 42
# Other methods can be added here
Correctly override default values by passing keyword arguments
instance = MyConfigurableClass(attribute1='new_value', attribute2=False)
This will raise an exception as 'attribute3' is not predefined
instance = MyConfigurableClass(attribute3='invalid_attribute')
this will also assign the default value to any variable that has a type defined. In the example above the default values (mapped by default__kwargs and locals) will be: attribute1 = 'default_value' attribute2 = True attribute3 = '' # default value of str attribute4 = [] # default value of list attribute4 = 42 # defined value in the class
Note
It is important that all attributes which may be set at instantiation are predefined in the class. Failure to do so will result in an exception being raised.
Methods
init(**kwargs): The initializer that handles the assignment of keyword arguments to instance attributes. It enforces strict attribute assignment rules, only allowing attributes that are already defined in the class to be set.
Initialize an instance of the derived class, strictly assigning provided keyword arguments to corresponding instance attributes.
Parameters
**kwargs: Variable length keyword arguments.
Raises
Exception
- If a key from kwargs does not correspond to any attribute pre-defined in the class, an exception is raised to prevent setting an undefined attribute.
Expand source code
class Patch_Print(Kwargs_To_Self): enabled : bool = True expected_calls : list mocked_print : MagicMock patched_print : _patch print_calls : bool def __init__(self, **kwargs): super().__init__(**kwargs) def __enter__(self): if self.enabled: self.patched_print = patch('builtins.print') self.mocked_print = self.patched_print.start() return self def __exit__(self, exc_type, exc_val, exc_tb): if self.enabled: self.patched_print.stop() if self.print_calls: pprint(self.calls()) #print(self.mocked_print.call_args_list) if self.expected_calls: assert self.calls() == self.expected_calls def call_args_list(self): if self.mocked_print: return self.mocked_print.call_args_list return [] def calls(self): calls_data = [] if self.mocked_print: for call in self.mocked_print.call_args_list: if len(call.args) == 0 and call.kwargs == {}: call_data = '' elif len(call.args) == 1 and call.kwargs == {}: call_data = call.args[0] else: call_data = (call.args, call.kwargs) calls_data.append(call_data) return calls_data
Ancestors
Class variables
var enabled : bool
var expected_calls : list
var mocked_print : unittest.mock.MagicMock
var patched_print : unittest.mock._patch
var print_calls : bool
Methods
def call_args_list(self)
-
Expand source code
def call_args_list(self): if self.mocked_print: return self.mocked_print.call_args_list return []
def calls(self)
-
Expand source code
def calls(self): calls_data = [] if self.mocked_print: for call in self.mocked_print.call_args_list: if len(call.args) == 0 and call.kwargs == {}: call_data = '' elif len(call.args) == 1 and call.kwargs == {}: call_data = call.args[0] else: call_data = (call.args, call.kwargs) calls_data.append(call_data) return calls_data
Inherited members