Skip to content

Commit 82b6594

Browse files
committed
[feature/PI-605-bulk_etl_transform] fix up asids
1 parent 4e6a069 commit 82b6594

File tree

2 files changed

+9
-17
lines changed

2 files changed

+9
-17
lines changed

src/layers/domain/repository/device_repository/tests/v1/conftest.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -37,9 +37,7 @@ def device_with_asid() -> Device:
3737
device = product.create_device(name="Device-1")
3838
device.add_tag(foo="bar", abc="123")
3939
device.add_tag(bar="foo")
40-
device.add_key(
41-
key_value="ABC:1234567890", key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID
42-
)
40+
device.add_key(key_value="1234567890", key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID)
4341
device.add_key(key_value="P.WWW-CCC", key_type=DeviceKeyType.PRODUCT_ID)
4442
return device
4543

@@ -52,8 +50,8 @@ def device_with_mhs_id() -> Device:
5250
device = product.create_device(name="Device-2")
5351
device.add_key(key_value="P.WWW-YYY", key_type=DeviceKeyType.PRODUCT_ID)
5452
device.add_key(
55-
key_value="ABC:DEF-444:4444444444",
56-
key_type=DeviceKeyType.MESSAGE_HANDLING_SYSTEM_ID,
53+
key_value="123456",
54+
key_type=DeviceKeyType.CPA_ID,
5755
)
5856
device.add_tag(abc="123")
5957
device.add_tag(bar="foo")

src/layers/domain/repository/device_repository/tests/v1/test_device_repository_keys_v1.py

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def test__device_repository__add_two_keys(device: Device, repository: DeviceRepo
1717
id=device.id,
1818
)
1919
second_device.add_key(
20-
key_value="ABC:1234567890", key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID
20+
key_value="1234567890", key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID
2121
)
2222
repository.write(second_device)
2323
time.sleep(1)
@@ -28,29 +28,23 @@ def test__device_repository__add_two_keys(device: Device, repository: DeviceRepo
2828
id=device.id,
2929
).keys == [
3030
DeviceKey(key_type=DeviceKeyType.PRODUCT_ID, key_value="P.WWW-XXX"),
31-
DeviceKey(
32-
key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID, key_value="ABC:1234567890"
33-
),
31+
DeviceKey(key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID, key_value="1234567890"),
3432
]
3533
assert repository.read(
3634
product_team_id=device.product_team_id,
3735
product_id=device.product_id,
3836
id="P.WWW-XXX",
3937
).keys == [
4038
DeviceKey(key_type=DeviceKeyType.PRODUCT_ID, key_value="P.WWW-XXX"),
41-
DeviceKey(
42-
key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID, key_value="ABC:1234567890"
43-
),
39+
DeviceKey(key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID, key_value="1234567890"),
4440
]
4541
assert repository.read(
4642
product_team_id=device.product_team_id,
4743
product_id=device.product_id,
48-
id="ABC:1234567890",
44+
id="1234567890",
4945
).keys == [
5046
DeviceKey(key_type=DeviceKeyType.PRODUCT_ID, key_value="P.WWW-XXX"),
51-
DeviceKey(
52-
key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID, key_value="ABC:1234567890"
53-
),
47+
DeviceKey(key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID, key_value="1234567890"),
5448
]
5549

5650

@@ -69,7 +63,7 @@ def test__device_repository__delete_key(
6963
id=device_with_asid.id,
7064
)
7165
intermediate_device.delete_key(
72-
key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID, key_value="ABC:1234567890"
66+
key_type=DeviceKeyType.ACCREDITED_SYSTEM_ID, key_value="1234567890"
7367
)
7468
repository.write(intermediate_device)
7569
time.sleep(1)

0 commit comments

Comments
 (0)