def test_prepare_mock_side_effect_zero_dots(self) -> None:
'''tormnet.fixtures._prepare_mock(ContextStub, 'symbol', side_effect = range(2))'''
fixtures._prepare_mock(self.context, 'symbol', side_effect=range(2))
self.assertEqual(self.context.mocked_symbol(), 0)
self.assertEqual(self.context.mocked_symbol(), 1)
self.assertRaises(StopIteration, self.context.mocked_symbol)
def test_prepare_mock_return_value_many_dots(self) -> None:
'''tormnet.fixtures._prepare_mock(ContextStub, 'symbol.sub.a.b.c', return_value = 'a')'''
fixtures._prepare_mock(self.context,
'symbol.sub.a.b.c',
return_value='a')
self.assertEqual(self.context.mocked_symbol.sub.a.b.c(), 'a')
def test_prepare_mock_side_effect_zero_dots(self) -> None:
'''tormnet.fixtures._prepare_mock(ContextStub, 'symbol', side_effect = range(2))'''
fixtures._prepare_mock(self.context, 'symbol', side_effect = range(2))
self.assertEqual(self.context.mocked_symbol(), 0)
self.assertEqual(self.context.mocked_symbol(), 1)
self.assertRaises(StopIteration, self.context.mocked_symbol)
def test_prepare_mock_return_value_many_dots_all_levels(self) -> None:
'''tormnet.fixtures._prepare_mock(ContextStub, 'symbol.Sub.a.b.c', return_value = 'a')'''
class ContextStub(contexts.TestContext):
mocked_symbol_sub_a_b_c = unittest.mock.MagicMock(
name='ContextStub.mocked_symbol_sub_a_b_c')
c = ContextStub()
fixtures._prepare_mock(c, 'symbol.Sub.a.b.c', return_value='a')
self.assertEqual(c.mocked_symbol_sub_a_b_c(), 'a')
def test_prepare_mock_return_value_many_dots_all_levels(self) -> None:
'''tormnet.fixtures._prepare_mock(ContextStub, 'symbol.Sub.a.b.c', return_value = 'a')'''
class ContextStub(contexts.TestContext):
mocked_symbol_sub_a_b_c = unittest.mock.MagicMock(name = 'ContextStub.mocked_symbol_sub_a_b_c')
c = ContextStub()
fixtures._prepare_mock(c, 'symbol.Sub.a.b.c', return_value = 'a')
self.assertEqual(c.mocked_symbol_sub_a_b_c(), 'a')
def test_prepare_mock_empty_call_list(self) -> None:
'''torment.fixtures._prepare_mock(ContextStub, 'symbol'): mock has empty call list'''
fixtures._prepare_mock(self.context, 'symbol')
self.assertEqual(len(self.context.mocked_symbol.mock_calls), 0)
def test_prepare_mock_return_value_many_dots(self) -> None:
'''tormnet.fixtures._prepare_mock(ContextStub, 'symbol.sub.a.b.c', return_value = 'a')'''
fixtures._prepare_mock(self.context, 'symbol.sub.a.b.c', return_value = 'a')
self.assertEqual(self.context.mocked_symbol.sub.a.b.c(), 'a')
def test_prepare_mock_empty_call_list(self) -> None:
'''torment.fixtures._prepare_mock(ContextStub, 'symbol'): mock has empty call list'''
fixtures._prepare_mock(self.context, 'symbol')
self.assertEqual(len(self.context.mocked_symbol.mock_calls), 0)
