def test_batch_add_with_message():
"""Test making a batch with two messages"""
name1 = helper.user.name()
user_key1 = helper.user.key()
user_id1 = user_key1.public_key
message1 = rbac.user.make(user_id=user_id1, name=name1)
name2 = helper.user.name()
user_key2 = helper.user.key()
user_id2 = user_key2.public_key
message2 = rbac.user.make(user_id=user_id2, name=name2)
batch = rbac.user.batch(
signer_user_id=user_id1, signer_keypair=user_key1, message=message1
)
batch = rbac.user.batch(
signer_user_id=user_id2, signer_keypair=user_key2, message=message2, batch=batch
)
messages = batcher.unmake(batch)
assert len(messages) == 2
assert messages[0].user_id == user_id1
assert messages[0].name == name1
assert messages[1].user_id == user_id2
assert messages[1].name == name2
def test_batch_list_add_with_kargs():
"""Test making a batch list with two messages using kargs"""
name1 = helper.user.name()
user_key1 = helper.user.key()
user_id1 = user_key1.public_key
name2 = helper.user.name()
user_key2 = helper.user.key()
user_id2 = user_key2.public_key
batch_list = rbac.user.batch_list(
signer_user_id=user_id1, signer_keypair=user_key1, user_id=user_id1, name=name1
)
batch_list = rbac.user.batch_list(
signer_user_id=user_id2,
signer_keypair=user_key2,
user_id=user_id2,
name=name2,
batch_list=batch_list,
)
messages = batcher.unmake(batch_list)
assert len(messages) == 2
assert messages[0].user_id == user_id1
assert messages[0].name == name1
assert messages[1].user_id == user_id2
assert messages[1].name == name2
def test_batch_list_add_with_message():
"""Test making a batch list with two messages"""
name1 = helper.user.name()
user_key1 = helper.user.key()
user_id1 = user_key1.public_key
message1 = User().make(next_id=user_id1, name=name1)
name2 = helper.user.name()
user_key2 = helper.user.key()
user_id2 = user_key2.public_key
message2 = User().make(next_id=user_id2, name=name2)
batch_list = User().batch_list(
signer_user_id=user_id2, signer_keypair=user_key1, message=message1
)
batch_list = User().batch_list(
signer_user_id=user_id2,
signer_keypair=user_key2,
message=message2,
batch_list=batch_list,
)
messages = unmake(batch_list)
assert len(messages) == 2
assert messages[0].next_id == user_id1
assert messages[0].name == name1
assert messages[1].next_id == user_id2
assert messages[1].name == name2
def test_batch_add_with_kargs():
"""Test making a batch with two messages using kargs"""
name1 = helper.user.name()
user_key1 = helper.user.key()
user_id1 = user_key1.public_key
name2 = helper.user.name()
user_key2 = helper.user.key()
user_id2 = user_key2.public_key
batch = User().batch(
signer_user_id=user_id1, signer_keypair=user_key1, next_id=user_id1, name=name1
)
batch = User().batch(
signer_user_id=user_id2,
signer_keypair=user_key2,
next_id=user_id2,
name=name2,
batch=batch,
)
messages = unmake(batch)
assert len(messages) == 2
assert messages[0].next_id == user_id1
assert messages[0].name == name1
assert messages[1].next_id == user_id2
assert messages[1].name == name2
def test_unmake(self):
"""Test the unmake batch function with a single message"""
message, message_type, inputs, outputs, signer_keypair = self.get_test_inputs()
payload = batcher.make_payload(
message=message,
message_type=message_type,
inputs=inputs,
outputs=outputs,
signer_user_id=message.user_id,
signer_public_key=signer_keypair.public_key,
)
transaction, batch, batch_list, batch_request = batcher.make(
payload=payload, signer_keypair=signer_keypair
)
messages = batcher.unmake(
batch_object=payload, signer_public_key=signer_keypair.public_key
)
self.assertEqual(len(messages), 1)
self.assertEqualMessage(message, messages[0])
messages = batcher.unmake(
batch_object=transaction, signer_public_key=signer_keypair.public_key
)
self.assertEqual(len(messages), 1)
self.assertEqualMessage(message, messages[0])
messages = batcher.unmake(
batch_object=batch, signer_public_key=signer_keypair.public_key
)
self.assertEqual(len(messages), 1)
self.assertEqualMessage(message, messages[0])
messages = batcher.unmake(
batch_object=batch_list, signer_public_key=signer_keypair.public_key
)
self.assertEqual(len(messages), 1)
self.assertEqualMessage(message, messages[0])
messages = batcher.unmake(
batch_object=batch_request, signer_public_key=signer_keypair.public_key
)
self.assertEqual(len(messages), 1)
self.assertEqualMessage(message, messages[0])
def test_unmake_payload(self):
"""Test the unmake batch function"""
message, message_type, inputs, outputs, signer = self.get_test_inputs()
payload = batcher.make_payload(
message=message, message_type=message_type, inputs=inputs, outputs=outputs
)
messages = batcher.unmake(
batch_object=payload, signer_public_key=signer.public_key
)
self.assertEqual(len(messages), 1)
self.assertEqualMessage(message, messages[0])
def test_batch_with_message():
"""Test making a batch with a message"""
name = helper.user.name()
user_key = helper.user.key()
user_id = user_key.public_key
message = rbac.user.make(user_id=user_id, name=name)
batch = rbac.user.batch(signer_keypair=user_key, message=message)
messages = batcher.unmake(batch)
assert len(messages) == 1
assert messages[0].user_id == user_id
assert messages[0].name == name
def test_batch_with_kargs():
"""Test making a batch with key arguments"""
name = helper.user.name()
user_key = helper.user.key()
user_id = user_key.public_key
batch = rbac.user.batch(
signer_user_id=user_id, signer_keypair=user_key, user_id=user_id, name=name
)
messages = batcher.unmake(batch)
assert len(messages) == 1
assert messages[0].user_id == user_id
assert messages[0].name == name
def test_batch_list_with_kargs():
"""Test making a batch list with key arguments"""
name = helper.user.name()
user_key = helper.user.key()
next_id = user_key.public_key
batch_list = User().batch_list(
signer_user_id=next_id, signer_keypair=user_key, next_id=next_id, name=name
)
messages = unmake(batch_list)
assert len(messages) == 1
assert messages[0].next_id == next_id
assert messages[0].name == name
def test_batch_with_message():
"""Test making a batch with a message"""
name = helper.user.name()
user_key = helper.user.key()
next_id = user_key.public_key
message = User().make(next_id=next_id, name=name)
batch = User().batch(
signer_user_id=next_id, signer_keypair=user_key, message=message
)
messages = unmake(batch)
assert len(messages) == 1
assert messages[0].next_id == next_id
assert messages[0].name == name
def test_batch():
"""Test creating a batch"""
user_id = helper.user.id()
name = helper.user.name()
signer_keypair = helper.user.key()
batch = rbac.user.imports.batch(
signer_user_id=user_id,
signer_keypair=signer_keypair,
user_id=user_id,
name=name,
)
messages = batcher.unmake(batch)
assert isinstance(messages, list)
assert len(messages) == 1
assert messages[0].user_id == user_id
assert messages[0].name == name
