Update tests/mgbench/datasets.py

Co-authored-by: Jure Bajic <jure.bajic@memgraph.com>
This commit is contained in:
Jeremy B 2022-10-31 12:56:02 +01:00 committed by GitHub
parent 2898120eeb
commit bae8c084b1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -341,8 +341,8 @@ class AccessControl(Dataset):
def _get_random_uuid(self, type):
assert type in ["File", "Permission", "Identity"]
first_uuid = Dataset.get_size(self)["uuid_ranges"][type]["first_uuid"]
last_uuid = Dataset.get_size(self)["uuid_ranges"][type]["last_uuid"]
first_uuid = self.get_size()["uuid_ranges"][type]["first_uuid"]
last_uuid = self.get_size()["uuid_ranges"][type]["last_uuid"]
random_value = random.randint(first_uuid, last_uuid)
return random_value