import pytest


@pytest.fixture(scope='module')
def start_cmd():
    ''' broken by default, required override '''
    return None


@pytest.fixture
def running_pihole(docker_persist, slow, persist_webserver, persist_tag, start_cmd):
    ''' Override the running_pihole to run and check for success of a
        pihole-FTL start based `pihole` script command
        
        Individual tests all must override start_cmd'''
    assert docker_persist.dig.run('ping -c 1 test_pihole').rc == 0
    slow(lambda: docker_persist.run('pgrep pihole-FTL').rc == 0)
    slow(lambda: docker_persist.run('pgrep {}'.format(persist_webserver)).rc == 0)
    oldpid = docker_persist.run('pidof pihole-FTL')
    cmd = docker_persist.run('pihole {}'.format(start_cmd))
    slow(lambda: docker_persist.run('pgrep pihole-FTL').rc == 0)
    newpid = docker_persist.run('pidof pihole-FTL')
    for pid in [oldpid, newpid]:
        assert pid != ''
    # ensure a new pid for pihole-FTL appeared due to service restart
    assert oldpid != newpid
    assert cmd.rc == 0
    # Save out cmd result to check different stdout of start/enable/disable
    docker_persist.cmd = cmd
    return docker_persist


@pytest.mark.parametrize('start_cmd,hostname,expected_ip, expected_messages', [
    ('enable',  'pi.hole', '127.0.0.1', ['Blocking already enabled,','nothing to do']),
    ('disable', 'pi.hole', '127.0.0.1', ['Disabling blocking','Pi-hole Disabled']),
])
def test_pihole_enable_disable_command(running_pihole, dig, persist_tag, start_cmd, hostname, expected_ip, expected_messages):
    ''' the start_cmd tests are all built into the running_pihole fixture in this file '''
    dig_cmd = "dig +time=1 +noall +answer {} @test_pihole".format(hostname)
    lookup = running_pihole.dig.run(dig_cmd)
    assert lookup.rc == 0
    lookup_ip = lookup.stdout.split()[4]
    assert lookup_ip == expected_ip

    for part_of_output in expected_messages:
        assert part_of_output in running_pihole.cmd.stdout

@pytest.mark.parametrize('start_cmd,expected_message', [
    ('-up', 'Function not supported in Docker images')
])
def test_pihole_update_command(running_pihole, start_cmd, expected_message):
    assert running_pihole.cmd.stdout.strip() == expected_message