def test_register_item(self, mock_item_map): """Test registering a HoudiniEventItem.""" mock_map = {} mock_item_map.return_value = mock_map mock_item = MagicMock(spec=ht.events.item.HoudiniEventItem) event = ht.events.event.HoudiniEvent(None) event.register_item(mock_item) self.assertEqual(mock_map, {mock_item.priority: [mock_item]})
def test_register_item(self, init_event, mocker): """Test registering a HoudiniEventItem.""" mock_item_map = mocker.patch.object(ht.events.event.HoudiniEvent, "item_map", new_callable=mocker.PropertyMock) mock_map = {} mock_item_map.return_value = mock_map # Mock the item via the reference in the module so that the isinstance() # test is more reliable vs mocking ht.events.item.HoudiniEventItem mock_item = mocker.MagicMock(spec=ht.events.event.HoudiniEventItem) event = init_event() event.register_item(mock_item) assert mock_map == {mock_item.priority: [mock_item]}
def test_register_item__non_item(self): """Test registering an object which is not a HoudiniEventItem.""" event = ht.events.event.HoudiniEvent(None) with self.assertRaises(TypeError): event.register_item(None)
def test_register_item__non_item(self, init_event): """Test registering an object which is not a HoudiniEventItem.""" event = init_event() with pytest.raises(TypeError): event.register_item(None)