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)