def _get_dict_with_txlog_attrs_from_share(self, share, op_type):
when_last_changed = share.when_last_changed
extra_data = dict(
shared_to=share.shared_to.id if share.shared_to else share.email,
share_id=str(share.id), share_name=share.name,
access_level=share.access,
when_shared=get_epoch_secs(share.when_shared),
when_last_changed=get_epoch_secs(when_last_changed))
return self._get_dict_with_txlog_attrs_from(
share.subtree, op_type, omit_generation=True,
extra_data_dict=extra_data)
def test_extra_data_new_node(self):
"""Check that extra_data_new_node includes all we need."""
f = self.factory.make_file()
f_extra_data = dict(
size=f.content.size, storage_key=unicode(f.content.storage_key),
public_uuid=None, content_hash=f.content_hash,
when_created=get_epoch_secs(f.when_created),
last_modified=get_epoch_secs(f.when_last_modified),
kind=f.kind, volume_path=f.volume.path)
expected = TransactionLog.extra_data_new_node(f)
self.assertEqual(expected, f_extra_data)
def _get_dict_with_txlog_attrs_from_share(self, share, op_type):
when_last_changed = share.when_last_changed
extra_data = dict(
shared_to=share.shared_to.id if share.shared_to else share.email,
share_id=str(share.id),
share_name=share.name,
access_level=share.access,
when_shared=get_epoch_secs(share.when_shared),
when_last_changed=get_epoch_secs(when_last_changed))
return self._get_dict_with_txlog_attrs_from(share.subtree,
op_type,
omit_generation=True,
extra_data_dict=extra_data)
def test_extra_data_new_node(self):
"""Check that extra_data_new_node includes all we need."""
f = self.factory.make_file()
f_extra_data = dict(size=f.content.size,
storage_key=unicode(f.content.storage_key),
public_uuid=None,
content_hash=f.content_hash,
when_created=get_epoch_secs(f.when_created),
last_modified=get_epoch_secs(f.when_last_modified),
kind=f.kind,
volume_path=f.volume.path)
expected = TransactionLog.extra_data_new_node(f)
self.assertEqual(expected, f_extra_data)
def _get_dict_with_txlog_attrs_from_udf(self, udf, op_type):
extra_data = None
if op_type == TransactionLog.OP_UDF_CREATED:
when_created = get_epoch_secs(udf.when_created)
extra_data = dict(when_created=when_created)
return dict(
node_id=None, volume_id=udf.id, owner_id=udf.owner_id,
op_type=op_type, path=udf.path, generation=udf.generation,
mimetype=None, old_path=None, extra_data_dict=extra_data)
def assertBootstrappingPickedUpUDFs(self, user, udfs):
txlogs = TransactionLog.objects.filter(
owner_id=user.id, op_type=TransactionLog.OP_UDF_CREATED)
expected = {}
self.assertEqual(len(udfs), txlogs.count())
for udf in udfs:
udf_txlog = txlogs.get(volume_id=udf.id)
when_created = get_epoch_secs(udf.when_created)
expected = dict(
node_id=None, volume_id=udf.id, generation=udf.generation,
path=udf.path, mimetype=None, owner_id=udf.owner_id,
extra_data_dict=dict(when_created=when_created),
op_type=TransactionLog.OP_UDF_CREATED)
self.assert_txlog_correct(udf_txlog, expected)
def _get_dict_with_txlog_attrs_from_udf(self, udf, op_type):
extra_data = None
if op_type == TransactionLog.OP_UDF_CREATED:
when_created = get_epoch_secs(udf.when_created)
extra_data = dict(when_created=when_created)
return dict(node_id=None,
volume_id=udf.id,
owner_id=udf.owner_id,
op_type=op_type,
path=udf.path,
generation=udf.generation,
mimetype=None,
old_path=None,
extra_data_dict=extra_data)
def assertBootstrappingPickedUpUDFs(self, user, udfs):
txlogs = TransactionLog.objects.filter(
owner_id=user.id, op_type=TransactionLog.OP_UDF_CREATED)
expected = {}
self.assertEqual(len(udfs), txlogs.count())
for udf in udfs:
udf_txlog = txlogs.get(volume_id=udf.id)
when_created = get_epoch_secs(udf.when_created)
expected = dict(node_id=None,
volume_id=udf.id,
generation=udf.generation,
path=udf.path,
mimetype=None,
owner_id=udf.owner_id,
extra_data_dict=dict(when_created=when_created),
op_type=TransactionLog.OP_UDF_CREATED)
self.assert_txlog_correct(udf_txlog, expected)
