"""Tests for build helper."""

import pytest
from accounting.bin import build_helper


@pytest.mark.parametrize('cmd, expected_command_name', [
    ('reserve_payouts.dynamodb_clean', 'dynamodb_clean'),
    ('reserve_payouts.sqs_clean', 'sqs_clean')
])
def test_get_command_name(cmd, expected_command_name):
    """Test get command name, parametrized."""
    actual_result = build_helper.get_command_name(cmd)
    assert expected_command_name == actual_result


@pytest.mark.parametrize('cmd, expected_package_name', [
    ('reserve_payouts.dynamodb_clean', 'reserve_payouts'),
    ('reserve_payouts.sqs_clean', 'reserve_payouts')
])
def test_get_package_name(cmd, expected_package_name):
    """Test get package name, parametrized."""
    actual_result = build_helper.get_package_name(cmd)
    assert expected_package_name == actual_result