mirror of
https://github.com/Thor77/TeamspeakStats.git
synced 2025-12-14 02:01:37 -05:00
bc547228b8
This adds support for a more expressive (albeit more verbose) IdentMap structure. It makes it easier to annotate the structure with additional data (such as names to associate with the IDs), to assist with maintaining the IdentMap.
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
import pytest
|
|
|
|
from tsstats.client import Client, Clients
|
|
from tsstats.utils import transform_pretty_identmap
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def identmap_clients():
|
|
clients = Clients({
|
|
'1': '2',
|
|
'5': '2',
|
|
'UID1': 'UID2',
|
|
'UID5': 'UID2'
|
|
})
|
|
cl = Client('2', 'Client2')
|
|
uidcl = Client('UID2', 'Client2++')
|
|
clients += cl
|
|
clients += uidcl
|
|
return (clients, cl, uidcl)
|
|
|
|
|
|
def test_ident_map(identmap_clients):
|
|
clients, cl, uidcl = identmap_clients
|
|
assert clients['1'] == cl
|
|
assert clients['5'] == cl
|
|
assert clients['UID1'] == uidcl
|
|
assert clients['UID5'] == uidcl
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def pretty_identmap():
|
|
return [
|
|
{'primary_id': '1', 'alternate_ids': ['3', '6']},
|
|
{'primary_id': '4', 'alternate_ids': ['9', '42']}
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize('test_input,expected', [
|
|
(
|
|
[
|
|
{'primary_id': '1', 'alternate_ids': ['3', '6']},
|
|
{'primary_id': '4', 'alternate_ids': ['9', '42', '23']}
|
|
],
|
|
(('3', '1'), ('6', '1'), ('9', '4'), ('42', '4'), ('23', '4'))
|
|
),
|
|
(
|
|
[
|
|
{'name': 'Friend 1', 'primary_id': '2', 'alternate_ids': ['4']},
|
|
{'name': 'Friend 3', 'primary_id': '8', 'alternate_ids': ['9', '14']}
|
|
],
|
|
(('4', '2'), ('9', '8'), ('14', '8'))
|
|
)
|
|
])
|
|
def test_flip_pretty_identmap(test_input, expected):
|
|
transformed_identmap = transform_pretty_identmap(test_input)
|
|
for alternate, primary in expected:
|
|
assert transformed_identmap[alternate] == primary
|