Neels Hofmeyr | 3531a19 | 2017-03-28 14:30:28 +0200 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
| 2 | |
| 3 | import _prep |
| 4 | |
| 5 | import sys |
| 6 | import os |
| 7 | import io |
| 8 | import pprint |
| 9 | import copy |
| 10 | |
Pau Espin Pedrol | e8bbcbf | 2020-04-10 19:51:31 +0200 | [diff] [blame] | 11 | from osmo_gsm_tester.core import config, log, schema |
Neels Hofmeyr | 3531a19 | 2017-03-28 14:30:28 +0200 | [diff] [blame] | 12 | |
| 13 | example_config_file = 'test.cfg' |
Pau Espin Pedrol | ab1904a | 2020-05-06 18:35:26 +0200 | [diff] [blame] | 14 | example_config = os.path.join(_prep.script_dir, example_config_file) |
Neels Hofmeyr | 3531a19 | 2017-03-28 14:30:28 +0200 | [diff] [blame] | 15 | cfg = config.read(example_config) |
| 16 | |
Neels Hofmeyr | 2694a9d | 2017-04-27 19:48:09 +0200 | [diff] [blame] | 17 | pprint.pprint(cfg, width=81) |
Neels Hofmeyr | 3531a19 | 2017-03-28 14:30:28 +0200 | [diff] [blame] | 18 | |
| 19 | test_schema = { |
| 20 | 'modems[].dbus_path': schema.STR, |
| 21 | 'modems[].msisdn': schema.STR, |
| 22 | 'modems[].imsi': schema.IMSI, |
| 23 | 'modems[].ki': schema.STR, |
| 24 | 'bts[].name' : schema.STR, |
| 25 | 'bts[].type' : schema.STR, |
| 26 | 'bts[].addr' : schema.STR, |
| 27 | 'bts[].trx[].timeslots[]' : schema.STR, |
| 28 | 'bts[].trx[].band' : schema.BAND, |
| 29 | 'a_dict.foo' : schema.INT, |
| 30 | 'addr[]' : schema.IPV4, |
| 31 | 'hwaddr[]' : schema.HWADDR, |
| 32 | 'imsi[]' : schema.IMSI, |
| 33 | 'ki[]' : schema.KI, |
| 34 | } |
| 35 | |
| 36 | def val(which): |
| 37 | try: |
| 38 | schema.validate(which, test_schema) |
| 39 | print('Validation: OK') |
| 40 | except ValueError: |
| 41 | log.log_exn() |
| 42 | print('Validation: Error') |
| 43 | |
| 44 | print('- expect validation success:') |
| 45 | val(cfg) |
| 46 | |
| 47 | print('- unknown item:') |
| 48 | c = copy.deepcopy(cfg) |
| 49 | c['bts'][0]['unknown_item'] = 'no' |
| 50 | val(c) |
| 51 | |
| 52 | print('- wrong type modems[].imsi:') |
| 53 | c = copy.deepcopy(cfg) |
| 54 | c['modems'][0]['imsi'] = {'no':'no'} |
| 55 | val(c) |
| 56 | |
| 57 | print('- invalid key with space:') |
| 58 | c = copy.deepcopy(cfg) |
| 59 | c['modems'][0]['imsi '] = '12345' |
| 60 | val(c) |
| 61 | |
| 62 | print('- list instead of dict:') |
| 63 | c = copy.deepcopy(cfg) |
| 64 | c['a_dict'] = [ 1, 2, 3 ] |
| 65 | val(c) |
| 66 | |
| 67 | print('- unknown band:') |
| 68 | c = copy.deepcopy(cfg) |
| 69 | c['bts'][0]['trx'][0]['band'] = 'what' |
| 70 | val(c) |
| 71 | |
| 72 | print('- invalid v4 addrs:') |
| 73 | c = copy.deepcopy(cfg) |
| 74 | c['addr'][3] = '1.2.3' |
| 75 | val(c) |
| 76 | c['addr'][3] = '1.2.3 .4' |
| 77 | val(c) |
| 78 | c['addr'][3] = '91.2.3' |
| 79 | val(c) |
| 80 | c['addr'][3] = 'go away' |
| 81 | val(c) |
| 82 | c['addr'][3] = '' |
| 83 | val(c) |
| 84 | c['addr'][3] = None |
| 85 | val(c) |
| 86 | |
| 87 | print('- invalid hw addrs:') |
| 88 | c = copy.deepcopy(cfg) |
| 89 | c['hwaddr'][3] = '1.2.3' |
| 90 | val(c) |
| 91 | c['hwaddr'][3] = '0b:0c:0d:0e:0f:0g' |
| 92 | val(c) |
| 93 | c['hwaddr'][3] = '0b:0c:0d:0e : 0f:0f' |
| 94 | val(c) |
| 95 | c['hwaddr'][3] = 'go away' |
| 96 | val(c) |
| 97 | c['hwaddr'][3] = '' |
| 98 | val(c) |
| 99 | c['hwaddr'][3] = None |
| 100 | val(c) |
| 101 | |
| 102 | print('- invalid imsis:') |
| 103 | c = copy.deepcopy(cfg) |
| 104 | c['imsi'][2] = '99999999x9' |
| 105 | val(c) |
| 106 | c['imsi'][2] = '123 456 789 123' |
| 107 | val(c) |
| 108 | c['imsi'][2] = 'go away' |
| 109 | val(c) |
| 110 | c['imsi'][2] = '' |
| 111 | val(c) |
| 112 | c['imsi'][2] = None |
| 113 | val(c) |
| 114 | |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 115 | print('- Combine dicts:') |
| 116 | a = {'times': '2'} |
| 117 | b = {'type': 'osmo-bts-trx'} |
| 118 | res = {'times': '2', 'type': 'osmo-bts-trx'} |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 119 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 120 | assert a == res |
| 121 | |
| 122 | print('- Combine dicts 2:') |
| 123 | a = {'times': '1', 'label': 'foo', 'type': 'osmo-bts-trx'} |
| 124 | b = {'type': 'osmo-bts-trx'} |
| 125 | res = {'times': '1', 'label': 'foo', 'type': 'osmo-bts-trx'} |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 126 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 127 | assert a == res |
| 128 | |
| 129 | print('- Combine lists:') |
| 130 | a = { 'a_list': ['x', 'y', 'z'] } |
| 131 | b = { 'a_list': ['y'] } |
| 132 | res = {'a_list': ['x', 'y', 'z']} |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 133 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 134 | assert a == res |
| 135 | |
| 136 | print('- Combine lists 2:') |
| 137 | a = { 'a_list': ['x'] } |
| 138 | b = { 'a_list': ['w', 'u', 'x', 'y', 'z'] } |
| 139 | res = {'a_list': ['x', 'w', 'u', 'y', 'z']} |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 140 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 141 | assert a == res |
| 142 | |
| 143 | print('- Combine lists 3:') |
| 144 | a = { 'a_list': ['x', 3] } |
| 145 | b = { 'a_list': ['y', 'z'] } |
| 146 | try: |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 147 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 148 | except ValueError: |
| 149 | print("ValueError expected") |
| 150 | |
| 151 | print('- Combine lists 4:') |
| 152 | a = { 'a_list': [2, 3] } |
| 153 | b = { 'a_list': ['y', 'z'] } |
| 154 | try: |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 155 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 156 | except ValueError: |
| 157 | print("ValueError expected") |
| 158 | |
| 159 | print('- Combine lists 5:') |
| 160 | a = { 'a_list': [{}, {}] } |
| 161 | b = { 'a_list': ['y', 'z'] } |
| 162 | try: |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 163 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 164 | except ValueError: |
| 165 | print("ValueError expected") |
| 166 | |
| 167 | print('- Combine lists 6:') |
| 168 | a = { 'a_list': [{}, {}] } |
| 169 | b = { 'a_list': [{}] } |
| 170 | res = {'a_list': [{}, {}]} |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 171 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 172 | assert a == res |
| 173 | |
| 174 | print('- Combine lists 7:') |
| 175 | a = { 'times': '1', 'label': 'foo', 'trx': [{'nominal power': '10'}, {'nominal power': '12'}] } |
| 176 | b = { 'type': 'osmo-bts-trx', 'trx': [{'nominal power': '10'}, {'nominal power': '12'}] } |
| 177 | res = {'times': '1', 'label': 'foo', 'trx': [{'nominal power': '10'}, {'nominal power': '12'}], 'type': 'osmo-bts-trx'} |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 178 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 179 | assert a == res |
| 180 | |
| 181 | print('- Combine lists 8:') |
| 182 | a = { 'times': '1', 'label': 'foo', 'trx': [{'nominal power': '10'}] } |
| 183 | b = { 'type': 'osmo-bts-trx', 'trx': [{'nominal power': '10'}, {'nominal power': '12'}] } |
| 184 | res = {'times': '1', 'label': 'foo', 'trx': [{'nominal power': '10'}, {'nominal power': '12'}], 'type': 'osmo-bts-trx'} |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 185 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 186 | assert a == res |
| 187 | |
| 188 | print('- Combine lists 9:') |
| 189 | a = { 'times': '1', 'label': 'foo', 'trx': [{'nominal power': '10'}, {'nominal power': '12'}] } |
| 190 | b = { 'type': 'osmo-bts-trx', 'trx': [{'nominal power': '10'}] } |
| 191 | res = {'times': '1', 'label': 'foo', 'trx': [{'nominal power': '10'}, {'nominal power': '12'}], 'type': 'osmo-bts-trx'} |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 192 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 193 | assert a == res |
| 194 | |
| 195 | print('- Combine lists 10:') |
| 196 | a = { 'times': '1', 'label': 'foo', 'trx': [{'nominal power': '10'}, {'nominal power': '12'}] } |
| 197 | b = { 'type': 'osmo-bts-trx', 'trx': [{}, {'nominal power': '12'}] } |
| 198 | res = {'times': '1', 'label': 'foo', 'trx': [{'nominal power': '10'}, {'nominal power': '12'}], 'type': 'osmo-bts-trx'} |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 199 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 200 | assert a == res |
| 201 | |
| 202 | print('- Combine lists 13:') |
| 203 | a = { 'times': '1', 'label': 'foo', 'trx': [{}, {'nominal power': '12'}] } |
| 204 | b = { 'type': 'osmo-bts-trx', 'trx': [{'nominal power': '10'}, {'nominal power': '12'}] } |
| 205 | res = {'times': '1', 'label': 'foo', 'trx': [{'nominal power': '10'}, {'nominal power': '12'}], 'type': 'osmo-bts-trx'} |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 206 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 207 | assert a == res |
| 208 | |
| 209 | print('- Combine lists 14:') |
| 210 | a = { 'times': '1', 'label': 'foo', 'trx': [] } |
| 211 | b = { 'type': 'osmo-bts-trx', 'trx': [] } |
| 212 | res = {'times': '1', 'label': 'foo', 'trx': [], 'type': 'osmo-bts-trx'} |
Pau Espin Pedrol | ea8c3d4 | 2020-05-04 12:05:05 +0200 | [diff] [blame] | 213 | schema.combine(a, b) |
Pau Espin Pedrol | 43737da | 2017-08-28 14:04:07 +0200 | [diff] [blame] | 214 | assert a == res |
| 215 | |
Neels Hofmeyr | 3531a19 | 2017-03-28 14:30:28 +0200 | [diff] [blame] | 216 | # vim: expandtab tabstop=4 shiftwidth=4 |