Skip to content

Commit b37bd40

Browse files
committed
Add endswith SKU lookup
1 parent 10f0989 commit b37bd40

File tree

4 files changed

+5
-5
lines changed

4 files changed

+5
-5
lines changed

airos/model_map.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def get_sku_by_devmodel(self, devmodel: str) -> str:
103103
for model_name in MODELS:
104104
lower_model_name = model_name.lower()
105105

106-
if lower_model_name.startswith(lower_devmodel):
106+
if lower_model_name.endswith(lower_devmodel):
107107
if not best_match_is_prefix or len(lower_model_name) == len(
108108
lower_devmodel
109109
):

fixtures/airos_LiteBeam5AC_ap-ptp_30mhz.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
"ptmp": false,
1818
"ptp": true,
1919
"role": "access_point",
20-
"sku": "AMBIGUOUS",
20+
"sku": "LBE-5AC-GEN2",
2121
"station": false
2222
},
2323
"firewall": {

fixtures/airos_LiteBeam5AC_sta-ptp_30mhz.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
"ptmp": false,
1818
"ptp": true,
1919
"role": "station",
20-
"sku": "AMBIGUOUS",
20+
"sku": "LBE-5AC-GEN2",
2121
"station": true
2222
},
2323
"firewall": {

tests/test_model_map.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,10 +38,10 @@ def test_get_sku_by_devmodel_not_found_raises_keyerror(self):
3838
def test_get_sku_by_devmodel_multiple_matches_raises_exception_dynamic(self):
3939
"""Test to raise AirOSMultipleMatchesFoundException when partial match is ambiguous."""
4040
with pytest.raises(AirOSMultipleMatchesFoundException) as excinfo:
41-
self.mapper.get_sku_by_devmodel("Rocket")
41+
self.mapper.get_sku_by_devmodel("LiteBeam")
4242

4343
exception_message = str(excinfo.value)
44-
expected_matches = 7
44+
expected_matches = 4
4545

4646
match = re.search(r"matched multiple \((\d+)\) products", exception_message)
4747
assert match is not None

0 commit comments

Comments
 (0)