def test_get_formatter_for_type(self, mock_formatter):
context = mock.MagicMock()
command = RemoveCommand(context)
command.get_formatter_for_type(TYPE_PUPPET_MODULE)
mock_formatter.assert_called_once_with(TYPE_PUPPET_MODULE)
def setUp(self):
super(RemovePuppetModulesCommand, self).setUp()
self.command = RemoveCommand(self.context)
