test_auth.py

import os
import pytest
import argparse
from shelly_ota.auth import AuthProvider
from shelly_ota.shelly import ShellyDevice


@pytest.fixture
def empty_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('--auth_env')
    parser.add_argument('--auth_file')
    args = parser.parse_args()
    return args

@pytest.fixture
def simple_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('--auth_env', default=None)
    parser.add_argument('--auth_file', default=None)
    os.environ["TEST_AUTH"] = "user:pass"
    args = parser.parse_args(args=["--auth_env", "TEST_AUTH"])
    return args

@pytest.fixture
def multi_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('--auth_env', default=None)
    parser.add_argument('--auth_file', default=None)
    os.environ["TEST_AUTH"] = "user:pass"
    args = parser.parse_args(args=["--auth_file", "auth_file.sample"])
    return args

@pytest.fixture
def no_auth_provider(empty_args):
    return AuthProvider(empty_args)

@pytest.fixture
def simple_auth_provider(simple_args):
    return AuthProvider(simple_args)

@pytest.fixture
def multi_auth_provider(multi_args):
    return AuthProvider(multi_args)

@pytest.fixture
def dummy_device1():
    return ShellyDevice(
        '0.0.0.0',
        {'type': 'DUMMY', 'mac': '00:00:00:00:00:00'}
        )

@pytest.fixture
def dummy_device2():
    return ShellyDevice(
        '1.1.1.1',
        {'type': 'DUMMY', 'mac': '01:01:01:01:01:01'}
        )

@pytest.fixture
def dummy_device3():
    return ShellyDevice(
        '2.2.2.2',
        {'type': 'SHPLG-S', 'mac': '02:02:02:02:02:02'}
        )


def test_no_auth(no_auth_provider, dummy_device1):
    assert no_auth_provider is not None
    assert no_auth_provider.get_auth_for(None) is None
    assert no_auth_provider.get_auth_for(dummy_device1) is None

def test_simple_auth(simple_auth_provider, dummy_device1):
    assert simple_auth_provider.get_auth_for(dummy_device1) == ('user', 'pass')

def test_multi_auth1(multi_auth_provider, dummy_device1):
    assert multi_auth_provider.get_auth_for(dummy_device1) == ('test1_user', 'test1_pass')

def test_multi_auth2(multi_auth_provider, dummy_device2):
    assert multi_auth_provider.get_auth_for(dummy_device2) == ('test2_user', 'test2_pass')

def test_multi_auth3(multi_auth_provider, dummy_device3):
    assert multi_auth_provider.get_auth_for(dummy_device3) == ('test3_user', 'test3_pass')