xref: /aosp_15_r20/prebuilts/sdk/update_prebuilts/test_buildserver.py (revision 344a7f5ef16c479e7a7f54ee6567a9d112f9e72b)
1*344a7f5eSAndroid Build Coastguard Worker#!/usr/bin/python3
2*344a7f5eSAndroid Build Coastguard Worker
3*344a7f5eSAndroid Build Coastguard Workerimport buildserver
4*344a7f5eSAndroid Build Coastguard Worker
5*344a7f5eSAndroid Build Coastguard Worker
6*344a7f5eSAndroid Build Coastguard Workerdef test_parse_build_id():
7*344a7f5eSAndroid Build Coastguard Worker    build_id_presubmit = buildserver.parse_build_id("12345")
8*344a7f5eSAndroid Build Coastguard Worker    assert build_id_presubmit.url_id == "12345"
9*344a7f5eSAndroid Build Coastguard Worker    assert build_id_presubmit.fs_id == "12345"
10*344a7f5eSAndroid Build Coastguard Worker
11*344a7f5eSAndroid Build Coastguard Worker
12*344a7f5eSAndroid Build Coastguard Workerdef test_parse_build_id_presubmit():
13*344a7f5eSAndroid Build Coastguard Worker    build_id_presubmit = buildserver.parse_build_id("P12345")
14*344a7f5eSAndroid Build Coastguard Worker    assert build_id_presubmit.url_id == "P12345"
15*344a7f5eSAndroid Build Coastguard Worker    assert build_id_presubmit.fs_id == "0"
16*344a7f5eSAndroid Build Coastguard Worker
17*344a7f5eSAndroid Build Coastguard Worker
18*344a7f5eSAndroid Build Coastguard Workerdef test_parse_build_id_invalid():
19*344a7f5eSAndroid Build Coastguard Worker    build_id_presubmit = buildserver.parse_build_id("PABCDE")
20*344a7f5eSAndroid Build Coastguard Worker    assert build_id_presubmit is None
21*344a7f5eSAndroid Build Coastguard Worker
22*344a7f5eSAndroid Build Coastguard Worker
23*344a7f5eSAndroid Build Coastguard Workerif __name__ == "__main__":
24*344a7f5eSAndroid Build Coastguard Worker    test_parse_build_id()
25*344a7f5eSAndroid Build Coastguard Worker    test_parse_build_id_presubmit()
26*344a7f5eSAndroid Build Coastguard Worker    test_parse_build_id_invalid()
27