mirror of
https://github.com/google/bumble.git
synced 2026-07-19 15:00:48 +00:00
more windows compat fixes
This commit is contained in:
+54
-56
@@ -89,93 +89,91 @@ JSON3 = """
|
|||||||
def temporary_file():
|
def temporary_file():
|
||||||
file = tempfile.NamedTemporaryFile(delete=False)
|
file = tempfile.NamedTemporaryFile(delete=False)
|
||||||
file.close()
|
file.close()
|
||||||
temp_file = pathlib.Path(file.name)
|
yield file.name
|
||||||
yield temp_file
|
pathlib.Path(file.name).unlink()
|
||||||
temp_file.unlink()
|
|
||||||
|
|
||||||
|
|
||||||
# -----------------------------------------------------------------------------
|
# -----------------------------------------------------------------------------
|
||||||
async def test_basic(temporary_file):
|
async def test_basic(temporary_file):
|
||||||
with temporary_file.open(mode='w', encoding='utf-8') as file:
|
with open(temporary_file, mode='w', encoding='utf-8') as file:
|
||||||
keystore = JsonKeyStore('my_namespace', file.name)
|
|
||||||
file.write("{}")
|
file.write("{}")
|
||||||
file.flush()
|
file.flush()
|
||||||
|
|
||||||
keys = await keystore.get_all()
|
keystore = JsonKeyStore('my_namespace', temporary_file)
|
||||||
assert len(keys) == 0
|
|
||||||
|
|
||||||
keys = PairingKeys()
|
keys = await keystore.get_all()
|
||||||
await keystore.update('foo', keys)
|
assert len(keys) == 0
|
||||||
foo = await keystore.get('foo')
|
|
||||||
assert foo is not None
|
|
||||||
assert foo.ltk is None
|
|
||||||
ltk = bytes([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15])
|
|
||||||
keys.ltk = PairingKeys.Key(ltk)
|
|
||||||
await keystore.update('foo', keys)
|
|
||||||
foo = await keystore.get('foo')
|
|
||||||
assert foo is not None
|
|
||||||
assert foo.ltk is not None
|
|
||||||
assert foo.ltk.value == ltk
|
|
||||||
|
|
||||||
file.flush()
|
keys = PairingKeys()
|
||||||
with open(file.name, "r", encoding="utf-8") as json_file:
|
await keystore.update('foo', keys)
|
||||||
json_data = json.load(json_file)
|
foo = await keystore.get('foo')
|
||||||
assert 'my_namespace' in json_data
|
assert foo is not None
|
||||||
assert 'foo' in json_data['my_namespace']
|
assert foo.ltk is None
|
||||||
assert 'ltk' in json_data['my_namespace']['foo']
|
ltk = bytes([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15])
|
||||||
|
keys.ltk = PairingKeys.Key(ltk)
|
||||||
|
await keystore.update('foo', keys)
|
||||||
|
foo = await keystore.get('foo')
|
||||||
|
assert foo is not None
|
||||||
|
assert foo.ltk is not None
|
||||||
|
assert foo.ltk.value == ltk
|
||||||
|
|
||||||
|
with open(file.name, "r", encoding="utf-8") as json_file:
|
||||||
|
json_data = json.load(json_file)
|
||||||
|
assert 'my_namespace' in json_data
|
||||||
|
assert 'foo' in json_data['my_namespace']
|
||||||
|
assert 'ltk' in json_data['my_namespace']['foo']
|
||||||
|
|
||||||
|
|
||||||
# -----------------------------------------------------------------------------
|
# -----------------------------------------------------------------------------
|
||||||
async def test_parsing(temporary_file):
|
async def test_parsing(temporary_file):
|
||||||
with temporary_file.open(mode='w', encoding='utf-8') as file:
|
with open(temporary_file, mode='w', encoding='utf-8') as file:
|
||||||
keystore = JsonKeyStore('my_namespace', file.name)
|
|
||||||
file.write(JSON1)
|
file.write(JSON1)
|
||||||
file.flush()
|
file.flush()
|
||||||
|
|
||||||
foo = await keystore.get('14:7D:DA:4E:53:A8/P')
|
keystore = JsonKeyStore('my_namespace', file.name)
|
||||||
assert foo is not None
|
foo = await keystore.get('14:7D:DA:4E:53:A8/P')
|
||||||
assert foo.ltk.value == bytes.fromhex('d1897ee10016eb1a08e4e037fd54c683')
|
assert foo is not None
|
||||||
|
assert foo.ltk.value == bytes.fromhex('d1897ee10016eb1a08e4e037fd54c683')
|
||||||
|
|
||||||
|
|
||||||
# -----------------------------------------------------------------------------
|
# -----------------------------------------------------------------------------
|
||||||
async def test_default_namespace(temporary_file):
|
async def test_default_namespace(temporary_file):
|
||||||
with temporary_file.open(mode='w', encoding='utf-8') as file:
|
with open(temporary_file, mode='w', encoding='utf-8') as file:
|
||||||
keystore = JsonKeyStore(None, file.name)
|
|
||||||
file.write(JSON1)
|
file.write(JSON1)
|
||||||
file.flush()
|
file.flush()
|
||||||
|
|
||||||
all_keys = await keystore.get_all()
|
keystore = JsonKeyStore(None, file.name)
|
||||||
assert len(all_keys) == 1
|
all_keys = await keystore.get_all()
|
||||||
name, keys = all_keys[0]
|
assert len(all_keys) == 1
|
||||||
assert name == '14:7D:DA:4E:53:A8/P'
|
name, keys = all_keys[0]
|
||||||
assert keys.irk.value == bytes.fromhex('e7b2543b206e4e46b44f9e51dad22bd1')
|
assert name == '14:7D:DA:4E:53:A8/P'
|
||||||
|
assert keys.irk.value == bytes.fromhex('e7b2543b206e4e46b44f9e51dad22bd1')
|
||||||
|
|
||||||
with temporary_file.open(mode='w', encoding='utf-8') as file:
|
with open(temporary_file, mode='w', encoding='utf-8') as file:
|
||||||
keystore = JsonKeyStore(None, file.name)
|
|
||||||
file.write(JSON2)
|
file.write(JSON2)
|
||||||
file.flush()
|
file.flush()
|
||||||
|
|
||||||
keys = PairingKeys()
|
keystore = JsonKeyStore(None, file.name)
|
||||||
ltk = bytes([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15])
|
keys = PairingKeys()
|
||||||
keys.ltk = PairingKeys.Key(ltk)
|
ltk = bytes([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15])
|
||||||
await keystore.update('foo', keys)
|
keys.ltk = PairingKeys.Key(ltk)
|
||||||
file.flush()
|
await keystore.update('foo', keys)
|
||||||
with open(file.name, "r", encoding="utf-8") as json_file:
|
with open(file.name, "r", encoding="utf-8") as json_file:
|
||||||
json_data = json.load(json_file)
|
json_data = json.load(json_file)
|
||||||
assert '__DEFAULT__' in json_data
|
assert '__DEFAULT__' in json_data
|
||||||
assert 'foo' in json_data['__DEFAULT__']
|
assert 'foo' in json_data['__DEFAULT__']
|
||||||
assert 'ltk' in json_data['__DEFAULT__']['foo']
|
assert 'ltk' in json_data['__DEFAULT__']['foo']
|
||||||
|
|
||||||
with temporary_file.open(mode='w', encoding='utf-8') as file:
|
with open(temporary_file, mode='w', encoding='utf-8') as file:
|
||||||
keystore = JsonKeyStore(None, file.name)
|
|
||||||
file.write(JSON3)
|
file.write(JSON3)
|
||||||
file.flush()
|
file.flush()
|
||||||
|
|
||||||
all_keys = await keystore.get_all()
|
keystore = JsonKeyStore(None, file.name)
|
||||||
assert len(all_keys) == 1
|
all_keys = await keystore.get_all()
|
||||||
name, keys = all_keys[0]
|
assert len(all_keys) == 1
|
||||||
assert name == '14:7D:DA:4E:53:A8/P'
|
name, keys = all_keys[0]
|
||||||
assert keys.irk.value == bytes.fromhex('e7b2543b206e4e46b44f9e51dad22bd1')
|
assert name == '14:7D:DA:4E:53:A8/P'
|
||||||
|
assert keys.irk.value == bytes.fromhex('e7b2543b206e4e46b44f9e51dad22bd1')
|
||||||
|
|
||||||
|
|
||||||
# -----------------------------------------------------------------------------
|
# -----------------------------------------------------------------------------
|
||||||
|
|||||||
Reference in New Issue
Block a user