# Copyright (c) 2020, Nordic Semiconductor ASA

import itertools

import os
import sys

from west.app.main import BUILTIN_COMMAND_GROUPS
from conftest import cmd

def test_builtin_help_and_dash_h(west_init_tmpdir):
    # Test "west help" and "west -h" are the same for built-in
    # functionality.

    h1out = cmd('help')
    h2out = cmd('-h')
    assert h1out == h2out

    for cls in itertools.chain(*BUILTIN_COMMAND_GROUPS.values()):
        c = cls()
        h1out = cmd(f'help {c.name}')
        h2out = cmd(f'{c.name} -h')
        assert h1out == h2out

def test_extension_help_and_dash_h(west_init_tmpdir):
    # Test "west help <command>" and "west <command> -h" for extension
    # commands (west_init_tmpdir has a command with one).

    cmd('update')
    ext1out = cmd('help test-extension')
    ext2out = cmd('test-extension -h')

    expected = EXTENSION_EXPECTED
    if sys.platform == 'win32':
        # Manage gratuitous incompatibilities:
        #
        # - multiline python strings are \n separated even on windows
        # - the windows command help output gets an extra newline
        expected = os.linesep.join(expected.splitlines()) + os.linesep

    assert ext1out == ext2out
    assert ext1out == expected


EXTENSION_EXPECTED = '''\
usage: west test-extension [-h]

optional arguments:
  -h, --help  show this help message and exit
'''