49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
from dataclasses import dataclass
|
|
from types import SimpleNamespace
|
|
|
|
|
|
@dataclass
|
|
class FakeItem:
|
|
attributes: dict
|
|
secret: bytes
|
|
secret_content_type: str
|
|
deleted: bool = False
|
|
locked: bool = False
|
|
|
|
def get_secret(self):
|
|
return self.secret
|
|
|
|
def get_attributes(self):
|
|
return self.attributes
|
|
|
|
def is_locked(self):
|
|
return self.locked
|
|
|
|
def delete(self):
|
|
self.deleted = True
|
|
|
|
|
|
class FakeCollection:
|
|
def __init__(self):
|
|
self.session = SimpleNamespace(encrypted=True)
|
|
self.items = []
|
|
self.locked = False
|
|
|
|
def is_locked(self):
|
|
return self.locked
|
|
|
|
def search_items(self, attributes):
|
|
return iter(
|
|
i for i in self.items if not i.deleted and i.attributes == attributes
|
|
)
|
|
|
|
def create_item(
|
|
self, label, attributes, secret, replace=False, content_type="text/plain"
|
|
):
|
|
if replace:
|
|
for item in self.search_items(attributes):
|
|
item.delete()
|
|
item = FakeItem(attributes, secret, content_type)
|
|
self.items.append(item)
|
|
return item
|