#!/usr/bin/env python
# ClusterShell.CLI.Nodeset test suite
# Written by S. Thiell


"""Unit test for CLI/Nodeset.py"""

import random
import sys
import unittest

from TLib import *
from ClusterShell.CLI.Nodeset import main

from ClusterShell.NodeUtils import GroupResolverConfig
from ClusterShell.NodeSet import std_group_resolver, set_std_group_resolver


class CLINodesetTestBase(unittest.TestCase):
    """Base unit test class for testing CLI/Nodeset.py"""

    def _nodeset_t(self, args, input, expected_stdout, expected_rc=0,
                   expected_stderr=None):
        CLI_main(self, main, [ 'nodeset' ] + args, input, expected_stdout,
                 expected_rc, expected_stderr)


class CLINodesetTest(CLINodesetTestBase):
    """Unit test class for testing CLI/Nodeset.py"""

    def _battery_count(self, args):
        self._nodeset_t(args + ["--count", ""], None, "0\n")
        self._nodeset_t(args + ["--count", "foo"], None, "1\n")
        self._nodeset_t(args + ["--count", "foo", "bar"], None, "2\n")
        self._nodeset_t(args + ["--count", "foo", "foo"], None, "1\n")
        self._nodeset_t(args + ["--count", "foo", "foo", "bar"], None, "2\n")
        self._nodeset_t(args + ["--count", "foo[0]"], None, "1\n")
        self._nodeset_t(args + ["--count", "foo[2]"], None, "1\n")
        self._nodeset_t(args + ["--count", "foo[1,2]"], None, "2\n")
        self._nodeset_t(args + ["--count", "foo[1-2]"], None, "2\n")
        self._nodeset_t(args + ["--count", "foo[1,2]", "foo[1-2]"], None, "2\n")
        self._nodeset_t(args + ["--count", "foo[1-200,245-394]"], None, "350\n")
        self._nodeset_t(args + ["--count", "foo[395-442]", "foo[1-200,245-394]"], None, "398\n")
        self._nodeset_t(args + ["--count", "foo[395-442]", "foo", "foo[1-200,245-394]"], None, "399\n")
        self._nodeset_t(args + ["--count", "foo[395-442]", "foo", "foo[0-200,245-394]"], None, "400\n")
        self._nodeset_t(args + ["--count", "foo[395-442]", "bar3,bar24", "foo[1-200,245-394]"], None, "400\n")
        # from stdin
        self._nodeset_t(args + ["--count"], "\n", "0\n")
        self._nodeset_t(args + ["--count"], "foo\n", "1\n")
        self._nodeset_t(args + ["--count"], "foo\nbar\n", "2\n")
        self._nodeset_t(args + ["--count"], "foo\nfoo\n", "1\n")
        self._nodeset_t(args + ["--count"], "foo\nfoo\nbar\n", "2\n")
        self._nodeset_t(args + ["--count"], "foo[0]\n", "1\n")
        self._nodeset_t(args + ["--count"], "foo[2]\n", "1\n")
        self._nodeset_t(args + ["--count"], "foo[1,2]\n", "2\n")
        self._nodeset_t(args + ["--count"], "foo[1-2]\n", "2\n")
        self._nodeset_t(args + ["--count"], "foo[1,2]\nfoo[1-2]\n", "2\n")
        self._nodeset_t(args + ["--count"], "foo[1-200,245-394]\n", "350\n")
        self._nodeset_t(args + ["--count"], "foo[395-442]\nfoo[1-200,245-394]\n", "398\n")
        self._nodeset_t(args + ["--count"], "foo[395-442]\nfoo\nfoo[1-200,245-394]\n", "399\n")
        self._nodeset_t(args + ["--count"], "foo[395-442]\nfoo\nfoo[0-200,245-394]\n", "400\n")
        self._nodeset_t(args + ["--count"], "foo[395-442]\nbar3,bar24\nfoo[1-200,245-394]\n", "400\n")

    def test_001_count(self):
        """test nodeset --count"""
        self._battery_count([])
        self._battery_count(["--autostep=1"])
        self._battery_count(["--autostep=2"])
        self._battery_count(["--autostep=5"])
        self._battery_count(["--autostep=auto"])
        self._battery_count(["--autostep=0%"])
        self._battery_count(["--autostep=50%"])
        self._battery_count(["--autostep=100%"])

    def test_002_count_intersection(self):
        """test nodeset --count --intersection"""
        self._nodeset_t(["--count", "foo", "--intersection", "bar"], None, "0\n")
        self._nodeset_t(["--count", "foo", "--intersection", "foo"], None, "1\n")
        self._nodeset_t(["--count", "foo", "--intersection", "foo", "-i", "bar"], None, "0\n")
        self._nodeset_t(["--count", "foo[0]", "--intersection", "foo0"], None, "1\n")
        self._nodeset_t(["--count", "foo[2]", "--intersection", "foo"], None, "0\n")
        self._nodeset_t(["--count", "foo[1,2]", "--intersection", "foo[1-2]"], None, "2\n")
        self._nodeset_t(["--count", "foo[395-442]", "--intersection", "foo[1-200,245-394]"], None, "0\n")
        self._nodeset_t(["--count", "foo[395-442]", "--intersection", "foo", "-i", "foo[1-200,245-394]"], None, "0\n")
        self._nodeset_t(["--count", "foo[395-442]", "-i", "foo", "-i", "foo[0-200,245-394]"], None, "0\n")
        self._nodeset_t(["--count", "foo[395-442]", "--intersection", "bar3,bar24", "-i", "foo[1-200,245-394]"], None, "0\n")

    def test_003_count_intersection_stdin(self):
        """test nodeset --count --intersection (stdin)"""
        self._nodeset_t(["--count", "--intersection", "bar"], "foo\n", "0\n")
        self._nodeset_t(["--count", "--intersection", "foo"], "foo\n", "1\n")
        self._nodeset_t(["--count", "--intersection", "foo", "-i", "bar"], "foo\n", "0\n")
        self._nodeset_t(["--count", "--intersection", "foo0"], "foo[0]\n", "1\n")
        self._nodeset_t(["--count", "--intersection", "foo"], "foo[2]\n", "0\n")
        self._nodeset_t(["--count", "--intersection", "foo[1-2]"], "foo[1,2]\n", "2\n")
        self._nodeset_t(["--count", "--intersection", "foo[1-200,245-394]"], "foo[395-442]\n", "0\n")
        self._nodeset_t(["--count", "--intersection", "foo", "-i", "foo[1-200,245-394]"], "foo[395-442]\n", "0\n")
        self._nodeset_t(["--count", "-i", "foo", "-i", "foo[0-200,245-394]"], "foo[395-442]\n", "0\n")
        self._nodeset_t(["--count", "--intersection", "bar3,bar24", "-i", "foo[1-200,245-394]"], "foo[395-442]\n", "0\n")

    def _battery_fold(self, args):
        self._nodeset_t(args + ["--fold", ""], None, "\n")
        self._nodeset_t(args + ["--fold", "foo"], None, "foo\n")
        self._nodeset_t(args + ["--fold", "foo", "bar"], None, "bar,foo\n")
        self._nodeset_t(args + ["--fold", "foo", "foo"], None, "foo\n")
        self._nodeset_t(args + ["--fold", "foo", "foo", "bar"], None, "bar,foo\n")
        self._nodeset_t(args + ["--fold", "foo[0]"], None, "foo0\n")
        self._nodeset_t(args + ["--fold", "foo[2]"], None, "foo2\n")
        self._nodeset_t(args + ["--fold", "foo[1,2]"], None, "foo[1-2]\n")
        self._nodeset_t(args + ["--fold", "foo[1-2]"], None, "foo[1-2]\n")
        self._nodeset_t(args + ["--fold", "foo[1,2]", "foo[1-2]"], None, "foo[1-2]\n")
        self._nodeset_t(args + ["--fold", "foo[1-200,245-394]"], None, "foo[1-200,245-394]\n")
        self._nodeset_t(args + ["--fold", "foo[395-442]", "foo[1-200,245-394]"], None, "foo[1-200,245-442]\n")
        self._nodeset_t(args + ["--fold", "foo[395-442]", "foo", "foo[1-200,245-394]"], None, "foo,foo[1-200,245-442]\n")
        self._nodeset_t(args + ["--fold", "foo[395-442]", "foo", "foo[0-200,245-394]"], None, "foo,foo[0-200,245-442]\n")
        self._nodeset_t(args + ["--fold", "foo[395-442]", "bar3,bar24", "foo[1-200,245-394]"], None, "bar[3,24],foo[1-200,245-442]\n")
        # stdin
        self._nodeset_t(args + ["--fold"], "\n", "\n")
        self._nodeset_t(args + ["--fold"], "foo\n", "foo\n")
        self._nodeset_t(args + ["--fold"], "foo\nbar\n", "bar,foo\n")
        self._nodeset_t(args + ["--fold"], "foo\nfoo\n", "foo\n")
        self._nodeset_t(args + ["--fold"], "foo\nfoo\nbar\n", "bar,foo\n")
        self._nodeset_t(args + ["--fold"], "foo[0]\n", "foo0\n")
        self._nodeset_t(args + ["--fold"], "foo[2]\n", "foo2\n")
        self._nodeset_t(args + ["--fold"], "foo[1,2]\n", "foo[1-2]\n")
        self._nodeset_t(args + ["--fold"], "foo[1-2]\n", "foo[1-2]\n")
        self._nodeset_t(args + ["--fold"], "foo[1,2]\nfoo[1-2]\n", "foo[1-2]\n")
        self._nodeset_t(args + ["--fold"], "foo[1-200,245-394]\n", "foo[1-200,245-394]\n")
        self._nodeset_t(args + ["--fold"], "foo[395-442]\nfoo[1-200,245-394]\n", "foo[1-200,245-442]\n")
        self._nodeset_t(args + ["--fold"], "foo[395-442]\nfoo\nfoo[1-200,245-394]\n", "foo,foo[1-200,245-442]\n")
        self._nodeset_t(args + ["--fold"], "foo[395-442]\nfoo\nfoo[0-200,245-394]\n", "foo,foo[0-200,245-442]\n")
        self._nodeset_t(args + ["--fold"], "foo[395-442]\nbar3,bar24\nfoo[1-200,245-394]\n", "bar[3,24],foo[1-200,245-442]\n")

    def test_004_fold(self):
        """test nodeset --fold"""
        self._battery_fold([])
        self._battery_fold(["--autostep=3"])
        # --autostep=auto (1.7)
        self._battery_fold(["--autostep=auto"])
        self._battery_count(["--autostep=0%"])
        self._battery_count(["--autostep=50%"])
        self._battery_count(["--autostep=100%"])

    def test_005_fold_autostep(self):
        """test nodeset --fold --autostep=X"""
        self._nodeset_t(["--autostep=2", "-f", "foo0", "foo2", "foo4", "foo6"], None, "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=2", "-f", "foo4", "foo2", "foo0", "foo6"], None, "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=3", "-f", "foo0", "foo2", "foo4", "foo6"], None, "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=4", "-f", "foo0", "foo2", "foo4", "foo6"], None, "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=5", "-f", "foo0", "foo2", "foo4", "foo6"], None, "foo[0,2,4,6]\n")
        self._nodeset_t(["--autostep=auto", "-f", "foo0", "foo2", "foo4", "foo6"], None, "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=auto", "-f", "foo4", "foo2", "foo0", "foo6"], None, "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=auto", "-f", "foo4", "foo2", "foo0", "foo2", "foo6"], None, "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=auto", "-f", "foo4", "foo2", "foo0", "foo5", "foo6"], None, "foo[0,2,4-6]\n")
        self._nodeset_t(["--autostep=auto", "-f", "foo4", "foo2", "foo0", "foo9", "foo6"], None, "foo[0,2,4,6,9]\n")
        self._nodeset_t(["--autostep=75%", "-f", "foo0", "foo2", "foo4", "foo6"], None, "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=75%", "-f", "foo4", "foo2", "foo0", "foo6"], None, "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=80%", "-f", "foo4", "foo2", "foo0", "foo2", "foo6"], None, "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=80%", "-f", "foo4", "foo2", "foo0", "foo5", "foo6"], None, "foo[0,2,4-6]\n")
        self._nodeset_t(["--autostep=80%", "-f", "foo4", "foo2", "foo0", "foo9", "foo6"], None, "foo[0-6/2,9]\n")
        self._nodeset_t(["--autostep=81%", "-f", "foo4", "foo2", "foo0", "foo9", "foo6"], None, "foo[0,2,4,6,9]\n")
        self._nodeset_t(["--autostep=100%", "-f", "foo4", "foo2", "foo0", "foo9", "foo6"], None, "foo[0,2,4,6,9]\n")

    def test_006_expand(self):
        """test nodeset --expand"""
        self._nodeset_t(["--expand", ""], None, "\n")
        self._nodeset_t(["--expand", "foo"], None, "foo\n")
        self._nodeset_t(["--expand", "foo", "bar"], None, "bar foo\n")
        self._nodeset_t(["--expand", "foo", "foo"], None, "foo\n")
        self._nodeset_t(["--expand", "foo[0]"], None, "foo0\n")
        self._nodeset_t(["--expand", "foo[2]"], None, "foo2\n")
        self._nodeset_t(["--expand", "foo[1,2]"], None, "foo1 foo2\n")
        self._nodeset_t(["--expand", "foo[1-2]"], None, "foo1 foo2\n")
        self._nodeset_t(["--expand", "foo[1-2],bar"], None, "bar foo1 foo2\n")

    def test_007_expand_stdin(self):
        """test nodeset --expand (stdin)"""
        self._nodeset_t(["--expand"], "\n", "\n")
        self._nodeset_t(["--expand"], "foo\n", "foo\n")
        self._nodeset_t(["--expand"], "foo\nbar\n", "bar foo\n")
        self._nodeset_t(["--expand"], "foo\nfoo\n", "foo\n")
        self._nodeset_t(["--expand"], "foo[0]\n", "foo0\n")
        self._nodeset_t(["--expand"], "foo[2]\n", "foo2\n")
        self._nodeset_t(["--expand"], "foo[1,2]\n", "foo1 foo2\n")
        self._nodeset_t(["--expand"], "foo[1-2]\n", "foo1 foo2\n")
        self._nodeset_t(["--expand"], "foo[1-2],bar\n", "bar foo1 foo2\n")

    def test_008_expand_separator(self):
        """test nodeset --expand -S"""
        self._nodeset_t(["--expand", "-S", ":", "foo"], None, "foo\n")
        self._nodeset_t(["--expand", "-S", ":", "foo", "bar"], None, "bar:foo\n")
        self._nodeset_t(["--expand", "--separator", ":", "foo", "bar"], None, "bar:foo\n")
        self._nodeset_t(["--expand", "--separator=:", "foo", "bar"], None, "bar:foo\n")
        self._nodeset_t(["--expand", "-S", ":", "foo", "foo"], None, "foo\n")
        self._nodeset_t(["--expand", "-S", ":", "foo[0]"], None, "foo0\n")
        self._nodeset_t(["--expand", "-S", ":", "foo[2]"], None, "foo2\n")
        self._nodeset_t(["--expand", "-S", ":", "foo[1,2]"], None, "foo1:foo2\n")
        self._nodeset_t(["--expand", "-S", ":", "foo[1-2]"], None, "foo1:foo2\n")
        self._nodeset_t(["--expand", "-S", " ", "foo[1-2]"], None, "foo1 foo2\n")
        self._nodeset_t(["--expand", "-S", ",", "foo[1-2],bar"], None, "bar,foo1,foo2\n")
        self._nodeset_t(["--expand", "-S", "uuu", "foo[1-2],bar"], None, "baruuufoo1uuufoo2\n")
        self._nodeset_t(["--expand", "-S", "\\n", "foo[1-2]"], None, "foo1\nfoo2\n")

    def test_009_fold_xor(self):
        """test nodeset --fold --xor"""
        self._nodeset_t(["--fold", "foo", "-X", "bar"], None, "bar,foo\n")
        self._nodeset_t(["--fold", "foo", "-X", "foo"], None, "\n")
        self._nodeset_t(["--fold", "foo[1,2]", "-X", "foo[1-2]"], None, "\n")
        self._nodeset_t(["--fold", "foo[1-10]", "-X", "foo[5-15]"], None, "foo[1-4,11-15]\n")
        self._nodeset_t(["--fold", "foo[395-442]", "-X", "foo[1-200,245-394]"], None, "foo[1-200,245-442]\n")
        self._nodeset_t(["--fold", "foo[395-442]", "-X", "foo", "-X", "foo[1-200,245-394]"], None, "foo,foo[1-200,245-442]\n")
        self._nodeset_t(["--fold", "foo[395-442]", "-X", "foo", "-X", "foo[0-200,245-394]"], None, "foo,foo[0-200,245-442]\n")
        self._nodeset_t(["--fold", "foo[395-442]", "-X", "bar3,bar24", "-X", "foo[1-200,245-394]"], None, "bar[3,24],foo[1-200,245-442]\n")

    def test_010_fold_xor_stdin(self):
        """test nodeset --fold --xor (stdin)"""
        self._nodeset_t(["--fold", "-X", "bar"], "foo\n", "bar,foo\n")
        self._nodeset_t(["--fold", "-X", "foo"], "foo\n", "\n")
        self._nodeset_t(["--fold", "-X", "foo[1-2]"], "foo[1,2]\n", "\n")
        self._nodeset_t(["--fold", "-X", "foo[5-15]"], "foo[1-10]\n", "foo[1-4,11-15]\n")
        self._nodeset_t(["--fold", "-X", "foo[1-200,245-394]"], "foo[395-442]\n", "foo[1-200,245-442]\n")
        self._nodeset_t(["--fold", "-X", "foo", "-X", "foo[1-200,245-394]"], "foo[395-442]\n", "foo,foo[1-200,245-442]\n")
        self._nodeset_t(["--fold", "-X", "foo", "-X", "foo[0-200,245-394]"], "foo[395-442]\n", "foo,foo[0-200,245-442]\n")
        self._nodeset_t(["--fold", "-X", "bar3,bar24", "-X", "foo[1-200,245-394]"], "foo[395-442]\n", "bar[3,24],foo[1-200,245-442]\n")
        # using stdin for -X
        self._nodeset_t(["-f","foo[2-4]","-X","-"], "foo4 foo5 foo6\n", "foo[2-3,5-6]\n")
        self._nodeset_t(["-f","-X","-","foo[1-6]"], "foo4 foo5 foo6\n", "foo[1-6]\n")

    def test_011_fold_exclude(self):
        """test nodeset --fold --exclude"""
        # Empty result
        self._nodeset_t(["--fold", "foo", "-x", "foo"], None, "\n")
        # With no range
        self._nodeset_t(["--fold", "foo,bar", "-x", "foo"], None, "bar\n")
        # Normal with range
        self._nodeset_t(["--fold", "foo[0-5]", "-x", "foo[0-10]"], None, "\n")
        self._nodeset_t(["--fold", "foo[0-10]", "-x", "foo[0-5]"], None, "foo[6-10]\n")
        # Do no change
        self._nodeset_t(["--fold", "foo[6-10]", "-x", "bar[0-5]"], None, "foo[6-10]\n")
        self._nodeset_t(["--fold", "foo[0-10]", "foo[13-18]", "--exclude", "foo[5-10,15]"], None, "foo[0-4,13-14,16-18]\n")

    def test_012_fold_exclude_stdin(self):
        """test nodeset --fold --exclude (stdin)"""
        # Empty result
        self._nodeset_t(["--fold", "-x", "foo"], "", "\n")
        self._nodeset_t(["--fold", "-x", "foo"], "\n", "\n")
        self._nodeset_t(["--fold", "-x", "foo"], "foo\n", "\n")
        # With no range
        self._nodeset_t(["--fold", "-x", "foo"], "foo,bar\n", "bar\n")
        # Normal with range
        self._nodeset_t(["--fold", "-x", "foo[0-10]"], "foo[0-5]\n", "\n")
        self._nodeset_t(["--fold", "-x", "foo[0-5]"], "foo[0-10]\n", "foo[6-10]\n")
        # Do no change
        self._nodeset_t(["--fold", "-x", "bar[0-5]"], "foo[6-10]\n", "foo[6-10]\n")
        self._nodeset_t(["--fold", "--exclude", "foo[5-10,15]"], "foo[0-10]\nfoo[13-18]\n", "foo[0-4,13-14,16-18]\n")
        # using stdin for -x
        self._nodeset_t(["-f","foo[1-6]","-x","-"], "foo4 foo5 foo6\n", "foo[1-3]\n")
        self._nodeset_t(["-f","-x","-","foo[1-6]"], "foo4 foo5 foo6\n", "foo[1-6]\n")

    def test_013_fold_intersection(self):
        """test nodeset --fold --intersection"""
        # Empty result
        self._nodeset_t(["--fold", "foo", "-i", "foo"], None, "foo\n")
        # With no range
        self._nodeset_t(["--fold", "foo,bar", "--intersection", "foo"], None, "foo\n")
        # Normal with range
        self._nodeset_t(["--fold", "foo[0-5]", "-i", "foo[0-10]"], None, "foo[0-5]\n")
        self._nodeset_t(["--fold", "foo[0-10]", "-i", "foo[0-5]"], None, "foo[0-5]\n")
        self._nodeset_t(["--fold", "foo[6-10]", "-i", "bar[0-5]"], None, "\n")
        self._nodeset_t(["--fold", "foo[0-10]", "foo[13-18]", "-i", "foo[5-10,15]"], None, "foo[5-10,15]\n")
        # numerical bracket folding (#228)
        self._nodeset_t(["--fold", "node123[1-2]", "-i", "node1232"], None, "node1232\n")
        self._nodeset_t(["--fold", "node023[1-2]0", "-i", "node02320"], None, "node02320\n")
        self._nodeset_t(["--fold", "node023[1-2]0-ipmi2", "-i", "node02320-ipmi2"], None, "node02320-ipmi2\n")

    def test_014_fold_intersection_stdin(self):
        """test nodeset --fold --intersection (stdin)"""
        # Empty result
        self._nodeset_t(["--fold", "--intersection", "foo"], "", "\n")
        self._nodeset_t(["--fold", "--intersection", "foo"], "\n", "\n")
        self._nodeset_t(["--fold", "-i", "foo"], "foo\n", "foo\n")
        # With no range
        self._nodeset_t(["--fold", "-i", "foo"], "foo,bar\n", "foo\n")
        # Normal with range
        self._nodeset_t(["--fold", "-i", "foo[0-10]"], "foo[0-5]\n", "foo[0-5]\n")
        self._nodeset_t(["--fold", "-i", "foo[0-5]"], "foo[0-10]\n", "foo[0-5]\n")
        # Do no change
        self._nodeset_t(["--fold", "-i", "bar[0-5]"], "foo[6-10]\n", "\n")
        self._nodeset_t(["--fold", "-i", "foo[5-10,15]"], "foo[0-10]\nfoo[13-18]\n", "foo[5-10,15]\n")
        # using stdin for -i
        self._nodeset_t(["-f","foo[1-6]","-i","-"], "foo4 foo5 foo6\n", "foo[4-6]\n")
        self._nodeset_t(["-f","-i","-","foo[1-6]"], "foo4 foo5 foo6\n", "foo[1-6]\n")
        # numerical bracket folding (#228)
        self._nodeset_t(["--fold", "-i", "node123[1-2]"], "node1232\n", "node1232\n")
        self._nodeset_t(["--fold", "-i", "node023[1-2]0"], "node02320\n", "node02320\n")
        self._nodeset_t(["--fold", "-i", "node023[1-2]0-ipmi2"], "node02320-ipmi2\n", "node02320-ipmi2\n")

    def test_015_rangeset(self):
        """test nodeset --rangeset"""
        self._nodeset_t(["--fold","--rangeset","1,2"], None, "1-2\n")
        self._nodeset_t(["--expand","-R","1-2"], None, "1 2\n")
        self._nodeset_t(["--fold","-R","1-2","-X","2-3"], None, "1,3\n")

    def test_016_rangeset_stdin(self):
        """test nodeset --rangeset (stdin)"""
        self._nodeset_t(["--fold","--rangeset"], "1,2\n", "1-2\n")
        self._nodeset_t(["--expand","-R"], "1-2\n", "1 2\n")
        self._nodeset_t(["--fold","-R","-X","2-3"], "1-2\n", "1,3\n")

    def test_017_stdin(self):
        """test nodeset - (stdin)"""
        self._nodeset_t(["-f","-"], "foo\n", "foo\n")
        self._nodeset_t(["-f","-"], "foo1 foo2 foo3\n", "foo[1-3]\n")
        self._nodeset_t(["--autostep=2", "-f"], "foo0 foo2 foo4 foo6\n", "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=auto", "-f"], "foo0 foo2 foo4 foo6\n", "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=100%", "-f"], "foo0 foo2 foo4 foo6\n", "foo[0-6/2]\n")
        self._nodeset_t(["--autostep=0%", "-f"], "foo0 foo2 foo4 foo6\n", "foo[0-6/2]\n")

    def test_018_split(self):
        """test nodeset --split"""
        self._nodeset_t(["--split=2","-f", "bar"], None, "bar\n")
        self._nodeset_t(["--split", "2","-f", "foo,bar"], None, "bar\nfoo\n")
        self._nodeset_t(["--split", "2","-e", "foo", "bar", "bur", "oof", "gcc"], None, "bar bur foo\ngcc oof\n")
        self._nodeset_t(["--split=2","-f", "foo[2-9]"], None, "foo[2-5]\nfoo[6-9]\n")
        self._nodeset_t(["--split=2","-f", "foo[2-3,7]", "bar9"], None, "bar9,foo2\nfoo[3,7]\n")
        self._nodeset_t(["--split=3","-f", "foo[2-9]"], None, "foo[2-4]\nfoo[5-7]\nfoo[8-9]\n")
        self._nodeset_t(["--split=1","-f", "foo2", "foo3"], None, "foo[2-3]\n")
        self._nodeset_t(["--split=4","-f", "foo[2-3]"], None, "foo2\nfoo3\n")
        self._nodeset_t(["--split=4","-f", "foo3", "foo2"], None, "foo2\nfoo3\n")
        self._nodeset_t(["--split=2","-e", "foo[2-9]"], None, "foo2 foo3 foo4 foo5\nfoo6 foo7 foo8 foo9\n")
        self._nodeset_t(["--split=3","-e", "foo[2-9]"], None, "foo2 foo3 foo4\nfoo5 foo6 foo7\nfoo8 foo9\n")
        self._nodeset_t(["--split=1","-e", "foo3", "foo2"], None, "foo2 foo3\n")
        self._nodeset_t(["--split=4","-e", "foo[2-3]"], None, "foo2\nfoo3\n")
        self._nodeset_t(["--split=4","-e", "foo2", "foo3"], None, "foo2\nfoo3\n")
        self._nodeset_t(["--split=2","-c", "foo2", "foo3"], None, "1\n1\n")

    def test_019_contiguous(self):
        """test nodeset --contiguous"""
        self._nodeset_t(["--contiguous", "-f", "bar"], None, "bar\n")
        self._nodeset_t(["--contiguous", "-f", "foo,bar"], None, "bar\nfoo\n")
        self._nodeset_t(["--contiguous", "-f", "foo", "bar", "bur", "oof", "gcc"], None, "bar\nbur\nfoo\ngcc\noof\n")
        self._nodeset_t(["--contiguous", "-e", "foo", "bar", "bur", "oof", "gcc"], None, "bar\nbur\nfoo\ngcc\noof\n")
        self._nodeset_t(["--contiguous", "-f", "foo2"], None, "foo2\n")
        self._nodeset_t(["--contiguous", "-R", "-f", "2"], None, "2\n")
        self._nodeset_t(["--contiguous", "-f", "foo[2-9]"], None, "foo[2-9]\n")
        self._nodeset_t(["--contiguous", "-f", "foo[2-3,7]", "bar9"], None, "bar9\nfoo[2-3]\nfoo7\n")
        self._nodeset_t(["--contiguous", "-R", "-f", "2-3,7", "9"], None, "2-3\n7\n9\n")
        self._nodeset_t(["--contiguous", "-f", "foo2", "foo3"], None, "foo[2-3]\n")
        self._nodeset_t(["--contiguous", "-f", "foo3", "foo2"], None, "foo[2-3]\n")
        self._nodeset_t(["--contiguous", "-f", "foo3", "foo1"], None, "foo1\nfoo3\n")
        self._nodeset_t(["--contiguous", "-f", "foo[1-5/2]", "foo7"], None, "foo1\nfoo3\nfoo5\nfoo7\n")

    def test_020_slice(self):
        """test nodeset -I/--slice"""
        self._nodeset_t(["--slice=0","-f", "bar"], None, "bar\n")
        self._nodeset_t(["--slice=0","-e", "bar"], None, "bar\n")
        self._nodeset_t(["--slice=1","-f", "bar"], None, "\n")
        self._nodeset_t(["--slice=0-1","-f", "bar"], None, "bar\n")
        self._nodeset_t(["-I0","-f", "bar[34-68,89-90]"], None, "bar34\n")
        self._nodeset_t(["-R", "-I0","-f", "34-68,89-90"], None, "34\n")
        self._nodeset_t(["-I 0","-f", "bar[34-68,89-90]"], None, "bar34\n")
        self._nodeset_t(["-I 0","-e", "bar[34-68,89-90]"], None, "bar34\n")
        self._nodeset_t(["-I 0-3","-f", "bar[34-68,89-90]"], None, "bar[34-37]\n")
        self._nodeset_t(["-I 0-3","-f", "bar[34-68,89-90]", "-x", "bar34"], None, "bar[35-38]\n")
        self._nodeset_t(["-I 0-3","-f", "bar[34-68,89-90]", "-x", "bar35"], None, "bar[34,36-38]\n")
        self._nodeset_t(["-I 0-3","-e", "bar[34-68,89-90]"], None, "bar34 bar35 bar36 bar37\n")
        self._nodeset_t(["-I 3,1,0,2","-f", "bar[34-68,89-90]"], None, "bar[34-37]\n")
        self._nodeset_t(["-I 1,3,7,10,16,20,30,34-35,37","-f", "bar[34-68,89-90]"], None, "bar[35,37,41,44,50,54,64,68,89]\n")
        self._nodeset_t(["-I 8","-f", "bar[34-68,89-90]"], None, "bar42\n")
        self._nodeset_t(["-I 8-100","-f", "bar[34-68,89-90]"], None, "bar[42-68,89-90]\n")
        self._nodeset_t(["-I 0-100","-f", "bar[34-68,89-90]"], None, "bar[34-68,89-90]\n")
        self._nodeset_t(["-I 8-100/2","-f", "bar[34-68,89-90]"], None, "bar[42,44,46,48,50,52,54,56,58,60,62,64,66,68,90]\n")
        self._nodeset_t(["--autostep=2", "-I 8-100/2","-f", "bar[34-68,89-90]"], None, "bar[42-68/2,90]\n")
        self._nodeset_t(["--autostep=93%", "-I 8-100/2","-f", "bar[34-68,89-90]"], None, "bar[42-68/2,90]\n")
        self._nodeset_t(["--autostep=94%", "-I 8-100/2","-f", "bar[34-68,89-90]"], None, "bar[42,44,46,48,50,52,54,56,58,60,62,64,66,68,90]\n")
        self._nodeset_t(["--autostep=auto", "-I 8-100/2","-f", "bar[34-68,89-90]"], None, "bar[42,44,46,48,50,52,54,56,58,60,62,64,66,68,90]\n")
        self._nodeset_t(["--autostep=auto", "-I 8-100/2","-f", "bar[34-68]"], None, "bar[42-68/2]\n")
        self._nodeset_t(["--autostep=100%", "-I 8-100/2","-f", "bar[34-68]"], None, "bar[42-68/2]\n")

    def test_021_slice_stdin(self):
        """test nodeset -I/--slice (stdin)"""
        self._nodeset_t(["--slice=0","-f"], "bar\n", "bar\n")
        self._nodeset_t(["--slice=0","-e"], "bar\n", "bar\n")
        self._nodeset_t(["--slice=1","-f"], "bar\n", "\n")
        self._nodeset_t(["--slice=0-1","-f"], "bar\n", "bar\n")
        self._nodeset_t(["-I0","-f"], "bar[34-68,89-90]\n", "bar34\n")
        self._nodeset_t(["-R", "-I0","-f"], "34-68,89-90\n", "34\n")
        self._nodeset_t(["-I 0","-f"], "bar[34-68,89-90]\n", "bar34\n")
        self._nodeset_t(["-I 0","-e"], "bar[34-68,89-90]\n", "bar34\n")
        self._nodeset_t(["-I 0-3","-f"], "bar[34-68,89-90]\n", "bar[34-37]\n")
        self._nodeset_t(["-I 0-3","-f", "-x", "bar34"], "bar[34-68,89-90]\n", "bar[35-38]\n")
        self._nodeset_t(["-I 0-3","-f", "-x", "bar35"], "bar[34-68,89-90]\n", "bar[34,36-38]\n")
        self._nodeset_t(["-I 0-3","-e"], "bar[34-68,89-90]\n", "bar34 bar35 bar36 bar37\n")
        self._nodeset_t(["-I 3,1,0,2","-f"], "bar[34-68,89-90]\n", "bar[34-37]\n")
        self._nodeset_t(["-I 1,3,7,10,16,20,30,34-35,37","-f"], "bar[34-68,89-90]\n", "bar[35,37,41,44,50,54,64,68,89]\n")
        self._nodeset_t(["-I 8","-f"], "bar[34-68,89-90]\n", "bar42\n")
        self._nodeset_t(["-I 8-100","-f"], "bar[34-68,89-90]\n", "bar[42-68,89-90]\n")
        self._nodeset_t(["-I 0-100","-f"], "bar[34-68,89-90]\n", "bar[34-68,89-90]\n")
        self._nodeset_t(["-I 8-100/2","-f"], "bar[34-68,89-90]\n", "bar[42,44,46,48,50,52,54,56,58,60,62,64,66,68,90]\n")
        self._nodeset_t(["--autostep=2", "-I 8-100/2","-f"], "bar[34-68,89-90]\n", "bar[42-68/2,90]\n")
        self._nodeset_t(["--autostep=93%", "-I 8-100/2","-f"], "bar[34-68,89-90]\n", "bar[42-68/2,90]\n")
        self._nodeset_t(["--autostep=93.33%", "-I 8-100/2","-f"], "bar[34-68,89-90]\n", "bar[42-68/2,90]\n")
        self._nodeset_t(["--autostep=94%", "-I 8-100/2","-f"], "bar[34-68,89-90]\n", "bar[42,44,46,48,50,52,54,56,58,60,62,64,66,68,90]\n")
        self._nodeset_t(["--autostep=auto", "-I 8-100/2","-f"], "bar[34-68,89-90]\n", "bar[42,44,46,48,50,52,54,56,58,60,62,64,66,68,90]\n")
        self._nodeset_t(["--autostep=2", "-I 8-100/2","-f"], "bar[34-68]\n", "bar[42-68/2]\n")

    def test_022_output_format(self):
        """test nodeset -O"""
        self._nodeset_t(["--expand", "--output-format", "/path/%s/", "foo"], None, "/path/foo/\n")
        self._nodeset_t(["--expand", "-O", "/path/%s/", "-S", ":", "foo"], None, "/path/foo/\n")
        self._nodeset_t(["--expand", "-O", "/path/%s/", "foo[2]"], None, "/path/foo2/\n")
        self._nodeset_t(["--expand", "-O", "%s-ib0", "foo[1-4]"], None, "foo1-ib0 foo2-ib0 foo3-ib0 foo4-ib0\n")
        self._nodeset_t(["--expand", "-O", "%s-ib0", "-S", ":", "foo[1-4]"], None, "foo1-ib0:foo2-ib0:foo3-ib0:foo4-ib0\n")
        self._nodeset_t(["--fold", "-O", "%s-ipmi", "foo", "bar"], None, "bar-ipmi,foo-ipmi\n")
        self._nodeset_t(["--fold", "-O", "%s-ib0", "foo1", "foo2"], None, "foo[1-2]-ib0\n")
        self._nodeset_t(["--fold", "-O", "%s-ib0", "foo1", "foo2", "bar1", "bar2"], None, "bar[1-2]-ib0,foo[1-2]-ib0\n")
        self._nodeset_t(["--fold", "-O", "%s-ib0", "--autostep=auto", "foo[1-9/2]"], None, "foo[1-9/2]-ib0\n")
        self._nodeset_t(["--fold", "-O", "%s-ib0", "--autostep=6", "foo[1-9/2]"], None, "foo[1,3,5,7,9]-ib0\n")
        self._nodeset_t(["--fold", "-O", "%s-ib0", "--autostep=5", "foo[1-9/2]"], None, "foo[1-9/2]-ib0\n")
        self._nodeset_t(["--count", "-O", "result-%s", "foo1", "foo2"], None, "result-2\n")
        self._nodeset_t(["--contiguous", "-O", "%s-ipmi", "-f", "foo[2-3,7]", "bar9"], None, "bar9-ipmi\nfoo[2-3]-ipmi\nfoo7-ipmi\n")
        self._nodeset_t(["--split=2", "-O", "%s-ib", "-e", "foo[2-9]"], None, "foo2-ib foo3-ib foo4-ib foo5-ib\nfoo6-ib foo7-ib foo8-ib foo9-ib\n")
        self._nodeset_t(["--split=3", "-O", "hwm-%s", "-f", "foo[2-9]"], None, "hwm-foo[2-4]\nhwm-foo[5-7]\nhwm-foo[8-9]\n")
        self._nodeset_t(["-I0", "-O", "{%s}", "-f", "bar[34-68,89-90]"], None, "{bar34}\n")
        # RangeSet mode (-R)
        self._nodeset_t(["--fold", "-O", "{%s}", "--rangeset", "1,2"], None, "{1-2}\n")
        self._nodeset_t(["--expand", "-O", "{%s}", "-R","1-2"], None, "{1} {2}\n")
        self._nodeset_t(["--fold", "-O", "{%s}", "-R","1-2","-X","2-3"], None, "{1,3}\n")
        self._nodeset_t(["--fold", "-O", "{%s}", "-S", ":", "--rangeset", "1,2"], None, "{1-2}\n")
        self._nodeset_t(["--expand", "-O", "{%s}", "-S", ":", "-R","1-2"], None, "{1}:{2}\n")
        self._nodeset_t(["--fold", "-O", "{%s}", "-S", ":", "-R","1-2","-X","2-3"], None, "{1,3}\n")
        self._nodeset_t(["-R", "-I0", "-O", "{%s}", "-f", "34-68,89-90"], None, "{34}\n")

    def test_023_axis(self):
        """test nodeset folding with --axis"""
        self._nodeset_t(["--axis=0","-f", "bar"], None, "bar\n")
        self._nodeset_t(["--axis=1","-f", "bar"], None, "bar\n")
        self._nodeset_t(["--axis=1","-R","-f", "1,2,3"], None, None, 2,
                         "--axis option is only supported when folding nodeset\n")
        self._nodeset_t(["--axis=1","-e", "bar"], None, None, 2,
                         "--axis option is only supported when folding nodeset\n")

        # 1D and 2D nodeset: fold along axis 0 only
        self._nodeset_t(["--axis=1","-f", "comp-[1-2]-[1-3],login-[1-2]"], None,
                         'comp-[1-2]-1,comp-[1-2]-2,comp-[1-2]-3,login-[1-2]\n')
        # 1D and 2D nodeset: fold along axis 1 only
        self._nodeset_t(["--axis=2","-f", "comp-[1-2]-[1-3],login-[1-2]"], None,
                         'comp-1-[1-3],comp-2-[1-3],login-1,login-2\n')
        # 1D and 2D nodeset: fold along last axis only
        self._nodeset_t(["--axis=-1","-f", "comp-[1-2]-[1-3],login-[1-2]"], None,
                         'comp-1-[1-3],comp-2-[1-3],login-[1-2]\n')

        # test for a common case
        ndnodes = []
        for ib in range(2):
            for idx in range(500):
                ndnodes.append("node%d-ib%d" % (idx, ib))
        random.shuffle(ndnodes)

        self._nodeset_t(["--axis=1","-f"] + ndnodes, None,
                         "node[0-499]-ib0,node[0-499]-ib1\n")

        exp_result = []
        for idx in range(500):
            exp_result.append("node%d-ib[0-1]" % idx)

        self._nodeset_t(["--axis=2","-f"] + ndnodes, None,
                         ','.join(exp_result) + '\n')

        # 4D test
        ndnodes = ["c-1-2-3-4", "c-2-2-3-4", "c-3-2-3-4", "c-5-5-5-5",
                   "c-5-7-5-5", "c-5-9-5-5", "c-5-11-5-5", "c-9-8-8-08",
                   "c-9-8-8-09"]
        self._nodeset_t(["--axis=1","-f"] + ndnodes, None,
                         "c-5-5-5-5,c-5-7-5-5,c-5-9-5-5,c-5-11-5-5,c-[1-3]-2-3-4,c-9-8-8-08,c-9-8-8-09\n")
        self._nodeset_t(["--axis=2","-f"] + ndnodes, None,
                         "c-5-[5,7,9,11]-5-5,c-1-2-3-4,c-2-2-3-4,c-3-2-3-4,c-9-8-8-08,c-9-8-8-09\n")
        self._nodeset_t(["--axis=3","-f"] + ndnodes, None,
                         "c-5-5-5-5,c-5-7-5-5,c-5-9-5-5,c-5-11-5-5,c-1-2-3-4,c-2-2-3-4,c-3-2-3-4,c-9-8-8-08,c-9-8-8-09\n")
        self._nodeset_t(["--axis=4","-f"] + ndnodes, None,
                         "c-5-5-5-5,c-5-7-5-5,c-5-9-5-5,c-5-11-5-5,c-1-2-3-4,c-2-2-3-4,c-3-2-3-4,c-9-8-8-[08-09]\n")

        self._nodeset_t(["--axis=1-2","-f"] + ndnodes, None,
                         "c-5-[5,7,9,11]-5-5,c-[1-3]-2-3-4,c-9-8-8-08,c-9-8-8-09\n")
        self._nodeset_t(["--axis=2-3","-f"] + ndnodes, None,
                         "c-5-[5,7,9,11]-5-5,c-1-2-3-4,c-2-2-3-4,c-3-2-3-4,c-9-8-8-08,c-9-8-8-09\n")
        self._nodeset_t(["--axis=3-4","-f"] + ndnodes, None,
                         "c-5-5-5-5,c-5-7-5-5,c-5-9-5-5,c-5-11-5-5,c-1-2-3-4,c-2-2-3-4,c-3-2-3-4,c-9-8-8-[08-09]\n")
        self._nodeset_t(["--axis=1-3","-f"] + ndnodes, None,
                         "c-5-[5,7,9,11]-5-5,c-[1-3]-2-3-4,c-9-8-8-08,c-9-8-8-09\n")
        self._nodeset_t(["--axis=2-4","-f"] + ndnodes, None,
                         "c-5-[5,7,9,11]-5-5,c-1-2-3-4,c-2-2-3-4,c-3-2-3-4,c-9-8-8-[08-09]\n")

        self._nodeset_t(["--axis=1-4","-f"] + ndnodes, None,
                         "c-5-[5,7,9,11]-5-5,c-[1-3]-2-3-4,c-9-8-8-[08-09]\n")
        self._nodeset_t(["-f"] + ndnodes, None,
                         "c-5-[5,7,9,11]-5-5,c-[1-3]-2-3-4,c-9-8-8-[08-09]\n")

        # a case where axis and autostep are working
        self._nodeset_t(["--autostep=4", "--axis=1-2","-f"] + ndnodes, None,
                         "c-5-[5-11/2]-5-5,c-[1-3]-2-3-4,c-9-8-8-08,c-9-8-8-09\n")

    def test_024_axis_stdin(self):
        """test nodeset folding with --axis (stdin)"""
        self._nodeset_t(["--axis=0","-f"], "bar\n", "bar\n")
        self._nodeset_t(["--axis=1","-f"], "bar\n", "bar\n")
        self._nodeset_t(["--axis=1","-R","-f"], "1,2,3", None, 2,
                         "--axis option is only supported when folding nodeset\n")
        self._nodeset_t(["--axis=1","-e"], "bar\n", None, 2,
                         "--axis option is only supported when folding nodeset\n")

        # 1D and 2D nodeset: fold along axis 0 only
        self._nodeset_t(["--axis=1","-f"], "comp-[1-2]-[1-3],login-[1-2]\n",
                         'comp-[1-2]-1,comp-[1-2]-2,comp-[1-2]-3,login-[1-2]\n')
        # 1D and 2D nodeset: fold along axis 1 only
        self._nodeset_t(["--axis=2","-f"], "comp-[1-2]-[1-3],login-[1-2]\n",
                         'comp-1-[1-3],comp-2-[1-3],login-1,login-2\n')
        # 1D and 2D nodeset: fold along last axis only
        self._nodeset_t(["--axis=-1","-f"], "comp-[1-2]-[1-3],login-[1-2]\n",
                         'comp-1-[1-3],comp-2-[1-3],login-[1-2]\n')

        # test for a common case
        ndnodes = []
        for ib in range(2):
            for idx in range(500):
                ndnodes.append("node%d-ib%d" % (idx, ib))
        random.shuffle(ndnodes)

        self._nodeset_t(["--axis=1","-f"], '\n'.join(ndnodes) + '\n',
                         "node[0-499]-ib0,node[0-499]-ib1\n")

        exp_result = []
        for idx in range(500):
            exp_result.append("node%d-ib[0-1]" % idx)

        self._nodeset_t(["--axis=2","-f"], '\n'.join(ndnodes) + '\n',
                         ','.join(exp_result) + '\n')

    def test_025_pick(self):
        """test nodeset --pick"""
        for num in range(1, 100):
            self._nodeset_t(["--count", "--pick", str(num), "foo[1-100]"],
                            None, "%s\n" % num)
            self._nodeset_t(["--count", "--pick", str(num), "-R", "1-100"],
                            None, "%s\n" % num)



class CLINodesetGroupResolverTest1(CLINodesetTestBase):
    """Unit test class for testing CLI/Nodeset.py with custom Group Resolver"""

    def setUp(self):
        # Special tests that require a default group source set
        f = make_temp_file("""
[Main]
default: local

[local]
map: echo example[1-100]
all: echo example[1-1000]
list: echo foo bar moo
        """)
        set_std_group_resolver(GroupResolverConfig(f.name))

    def tearDown(self):
        set_std_group_resolver(None)

    def test_022_list(self):
        """test nodeset --list"""
        self._nodeset_t(["--list"], None, "@bar\n@foo\n@moo\n")
        self._nodeset_t(["-ll"], None, "@bar example[1-100]\n@foo example[1-100]\n@moo example[1-100]\n")
        self._nodeset_t(["-lll"], None, "@bar example[1-100] 100\n@foo example[1-100] 100\n@moo example[1-100] 100\n")
        self._nodeset_t(["-l", "example[4,95]", "example5"], None, "@moo\n@bar\n@foo\n")
        self._nodeset_t(["-ll", "example[4,95]", "example5"], None, "@moo example[4-5,95]\n@bar example[4-5,95]\n@foo example[4-5,95]\n")
        self._nodeset_t(["-lll", "example[4,95]", "example5"], None, "@moo example[4-5,95] 3/100\n@bar example[4-5,95] 3/100\n@foo example[4-5,95] 3/100\n")
        # test empty result
        self._nodeset_t(["-l", "foo[3-70]", "bar6"], None, "")
        # more arg-mixed tests
        self._nodeset_t(["-a", "-l"], None, "@moo\n@bar\n@foo\n")
        self._nodeset_t(["-a", "-l", "-x example[1-100]"], None, "")
        self._nodeset_t(["-a", "-l", "-x example[1-40]"], None, "@moo\n@bar\n@foo\n")
        self._nodeset_t(["-l", "-x example3"], None, "") # no -a, remove from nothing
        self._nodeset_t(["-l", "-i example3"], None, "") # no -a, intersect from nothing
        self._nodeset_t(["-l", "-X example3"], None, "@moo\n@bar\n@foo\n") # no -a, xor from nothing
        self._nodeset_t(["-l", "-", "-i example3"], "example[3,500]\n", "@moo\n@bar\n@foo\n")

    def test_023_list_all(self):
        """test nodeset --list-all"""
        self._nodeset_t(["--list-all"], None, "@bar\n@foo\n@moo\n")
        self._nodeset_t(["-L"], None, "@bar\n@foo\n@moo\n")
        self._nodeset_t(["-LL"], None, "@bar example[1-100]\n@foo example[1-100]\n@moo example[1-100]\n")
        self._nodeset_t(["-LLL"], None, "@bar example[1-100] 100\n@foo example[1-100] 100\n@moo example[1-100] 100\n")

class CLINodesetGroupResolverTest2(CLINodesetTestBase):
    """Unit test class for testing CLI/Nodeset.py with custom Group Resolver"""

    def setUp(self):
        # Special tests that require a default group source set
        f = make_temp_file("""
[Main]
default: test

[test]
map: echo example[1-100]
all: echo @foo,@bar,@moo
list: echo foo bar moo

[other]
map: echo nova[030-489]
all: echo @baz,@qux,@norf
list: echo baz qux norf
        """)
        set_std_group_resolver(GroupResolverConfig(f.name))

    def tearDown(self):
        set_std_group_resolver(None)

    def test_024_groups(self):
        self._nodeset_t(["--split=2","-r", "unknown2", "unknown3"], None, "unknown2\nunknown3\n")
        self._nodeset_t(["-f", "-a"], None, "example[1-100]\n")
        self._nodeset_t(["-f", "@moo"], None, "example[1-100]\n")
        self._nodeset_t(["-f", "@moo", "@bar"], None, "example[1-100]\n")
        self._nodeset_t(["-e", "-a"], None, ' '.join(["example%d" % i for i in range(1, 101)]) + '\n')
        self._nodeset_t(["-c", "-a"], None, "100\n")
        self._nodeset_t(["-r", "-a"], None, "@bar\n")
        self._nodeset_t(["--split=2","-r", "unknown2", "unknown3"], None, "unknown2\nunknown3\n")

    # We need to split following unit tests in order to reset group
    # source in setUp/tearDown...

    def test_025_groups(self):
        self._nodeset_t(["-s", "test", "-c", "-a", "-d"], None, "100\n")

    def test_026_groups(self):
        self._nodeset_t(["-s", "test", "-r", "-a"], None, "@test:bar\n")

    def test_027_groups(self):
        self._nodeset_t(["-s", "test", "-G", "-r", "-a"], None, "@bar\n")

    def test_028_groups(self):
        self._nodeset_t(["-s", "test", "--groupsources"], None, "test (default)\nother\n")

    def test_029_groups(self):
        self._nodeset_t(["-s", "test", "-q", "--groupsources"], None, "test\nother\n")

    def test_030_groups(self):
        self._nodeset_t(["-f", "-a", "-"], "example101\n", "example[1-101]\n")
        self._nodeset_t(["-f", "-a", "-"], "example102 example101\n", "example[1-102]\n")

    # Check default group source switching...

    def test_031_groups(self):
        self._nodeset_t(["-s", "other", "-c", "-a", "-d"], None, "460\n")
        self._nodeset_t(["-s", "test", "-c", "-a", "-d"], None, "100\n")

    def test_032_groups(self):
        self._nodeset_t(["-s", "other", "-r", "-a"], None, "@other:baz\n")
        self._nodeset_t(["-s", "test", "-r", "-a"], None, "@test:bar\n")

    def test_033_groups(self):
        self._nodeset_t(["-s", "other", "-G", "-r", "-a"], None, "@baz\n")
        self._nodeset_t(["-s", "test", "-G", "-r", "-a"], None, "@bar\n")

    def test_034_groups(self):
        self._nodeset_t(["--groupsources"], None, "test (default)\nother\n")

    def test_035_groups(self):
        self._nodeset_t(["-s", "other", "--groupsources"], None, "other (default)\ntest\n")

    def test_036_groups(self):
        self._nodeset_t(["--groupsources"], None, "test (default)\nother\n")

    def test_037_groups_output_format(self):
        self._nodeset_t(["-r", "-O", "{%s}", "-a"], None, "{@bar}\n")

    def test_038_groups_output_format(self):
        self._nodeset_t(["-O", "{%s}", "-s", "other", "-r", "-a"], None, "{@other:baz}\n")

    def test_039_list_all(self):
        """test nodeset --list-all (multi sources)"""
        self._nodeset_t(["--list-all"], None,
                        "@bar\n@foo\n@moo\n@other:baz\n@other:norf\n@other:qux\n")
        self._nodeset_t(["--list-all", "-G"], None,
                        "@bar\n@foo\n@moo\n@baz\n@norf\n@qux\n")
        self._nodeset_t(["-GL"], None,
                        "@bar\n@foo\n@moo\n@baz\n@norf\n@qux\n")
        self._nodeset_t(["--list-all", "-s", "other"], None,
                        "@other:baz\n@other:norf\n@other:qux\n@test:bar\n@test:foo\n@test:moo\n")
        self._nodeset_t(["--list-all", "-G", "-s", "other"], None,
                        "@baz\n@norf\n@qux\n@bar\n@foo\n@moo\n") # 'other' source first

class CLINodesetGroupResolverTest3(CLINodesetTestBase):
    """Unit test class for testing CLI/Nodeset.py with custom Group Resolver

    A case we support: one of the source misses the list upcall.
    """

    def setUp(self):
        # Special tests that require a default group source set
        f = make_temp_file("""
[Main]
default: test

[test]
map: echo example[1-100]
all: echo @foo,@bar,@moo
list: echo foo bar moo

[other]
map: echo nova[030-489]
all: echo @baz,@qux,@norf
list: echo baz qux norf

[pdu]
map: echo pdu-[0-3]-[1-2]
        """)
        set_std_group_resolver(GroupResolverConfig(f.name))

    def tearDown(self):
        set_std_group_resolver(None)

    def test_040_list_all(self):
        """test nodeset --list-all (w/ missing list upcall)"""
        self._nodeset_t(["--list-all"], None,
                        "@bar\n@foo\n@moo\n@other:baz\n@other:norf\n@other:qux\n", 0,
                        "Warning: No list upcall defined for group source pdu\n")
        self._nodeset_t(["-LL"], None,
                        "@bar example[1-100]\n@foo example[1-100]\n@moo example[1-100]\n"
                        "@other:baz nova[030-489]\n@other:norf nova[030-489]\n@other:qux nova[030-489]\n", 0,
                        "Warning: No list upcall defined for group source pdu\n")
        self._nodeset_t(["-LLL"], None,
                        "@bar example[1-100] 100\n@foo example[1-100] 100\n@moo example[1-100] 100\n"
                        "@other:baz nova[030-489] 460\n@other:norf nova[030-489] 460\n@other:qux nova[030-489] 460\n", 0,
                        "Warning: No list upcall defined for group source pdu\n")

    def test_041_list_failure(self):
        """test nodeset --list -s source w/ missing list upcall"""
        self._nodeset_t(["--list", "-s", "pdu"], None, "", 1,
                        'No list upcall defined for group source "pdu"\n')