# SPDX-FileCopyrightText: 2024 Marco Ricci <m@the13thletter.info>
#
# SPDX-License-Identifier: MIT

"""Test passphrase generation via derivepassphrase.Vault."""

import pytest

import derivepassphrase
import sequin

Vault = derivepassphrase.Vault
phrase = b'She cells C shells bye the sea shoars'

@pytest.mark.parametrize('service,expected', [
    (b'google', rb': 4TVH#5:aZl8LueOT\{'),
    ('twitter', rb"[ (HN_N:lI&<ro=)3'g9"),
])
def test_200_basic_configuration(service, expected):
    assert Vault(phrase=phrase).generate(service) == expected

def test_201_phrase_dependence():
    assert (
        Vault(phrase=(phrase + b'X')).generate('google') ==
        b'n+oIz6sL>K*lTEWYRO%7'
    )

def test_202_reproducibility_and_bytes_service_name():
    assert (
        Vault(phrase=phrase).generate(b'google') ==
        Vault(phrase=phrase).generate('google')
    )

def test_203_reproducibility_and_bytearray_service_name():
    assert (
        Vault(phrase=phrase).generate(b'google') ==
        Vault(phrase=phrase).generate(bytearray(b'google'))
    )

def test_210_nonstandard_length():
    assert Vault(phrase=phrase, length=4).generate('google') == b'xDFu'

def test_211_repetition_limit():
    assert (
        Vault(phrase=b'', length=24, symbol=0, number=0,
              repeat=1).generate('asd') ==
        b'IVTDzACftqopUXqDHPkuCIhV'
    )

def test_212_without_symbols():
    assert (
        Vault(phrase=phrase, symbol=0).generate('google') ==
        b'XZ4wRe0bZCazbljCaMqR'
    )

def test_213_too_many_symbols():
    with pytest.raises(ValueError,
                       match='requested passphrase length too short'):
        Vault(phrase=phrase, symbol=100)

def test_214_no_numbers():
    assert (
        Vault(phrase=phrase, number=0).generate('google') ==
        b'_*$TVH.%^aZl(LUeOT?>'
    )

def test_214_no_lowercase_letters():
    assert (
        Vault(phrase=phrase, lower=0).generate('google') ==
        b':{?)+7~@OA:L]!0E$)(+'
    )

def test_215_at_least_5_digits():
    assert (
        Vault(phrase=phrase, length=8, number=5).generate('songkick') ==
        b'i0908.7['
    )

def test_216_lots_of_spaces():
    assert (
        Vault(phrase=phrase, space=12).generate('songkick') ==
        b' c   6 Bq  % 5fR    '
    )

def test_217_no_viable_characters():
    with pytest.raises(ValueError,
                       match='no allowed characters left'):
        Vault(phrase=phrase, lower=0, upper=0, number=0,
              space=0, dash=0, symbol=0)

def test_218_all_character_classes():
    assert (
        Vault(phrase=phrase, lower=2, upper=2, number=1,
              space=3, dash=2, symbol=1).generate('google') ==
        b': : fv_wqt>a-4w1S  R'
    )

def test_219_only_numbers_and_very_high_repetition_limit():
    generated = Vault(phrase=b'', length=40, lower=0, upper=0, space=0,
                      dash=0, symbol=0, repeat=4).generate('abcdef')
    assert b'0000' not in generated
    assert b'1111' not in generated
    assert b'2222' not in generated
    assert b'3333' not in generated
    assert b'4444' not in generated
    assert b'5555' not in generated
    assert b'6666' not in generated
    assert b'7777' not in generated
    assert b'8888' not in generated
    assert b'9999' not in generated

def test_220_very_limited_character_set():
    generated = Vault(phrase=b'', length=24, lower=0, upper=0,
                      space=0, symbol=0).generate('testing')
    assert b'763252593304946694588866' == generated

def test_300_character_set_subtraction():
    assert Vault._subtract(b'be', b'abcdef') == bytearray(b'acdf')

def test_301_character_set_subtraction_duplicate():
    with pytest.raises(ValueError, match='duplicate characters'):
        Vault._subtract(b'abcdef', b'aabbccddeeff')
    with pytest.raises(ValueError, match='duplicate characters'):
        Vault._subtract(b'aabbccddeeff', b'abcdef')