import unittest
from onedrived.api import items
from onedrived.store import items_db
from tests import get_data
from tests.factory import drive_factory, db_factory, mock_factory
mock_factory.mock_register()
class TestItemStorage(unittest.TestCase):
record_keymap = {
'item_id': 'id',
'item_name': 'name',
'type': 'type',
'size': 'size',
'e_tag': 'e_tag',
'c_tag': 'c_tag',
}
def setUp(self):
self.all_items = []
self.all_items_data = []
self.drive = drive_factory.get_sample_drive_object()
self.itemdb_mgr = db_factory.get_sample_item_storage_manager()
self.itemdb = self.itemdb_mgr.get_item_storage(self.drive)
for name in [
'image_item.json', 'folder_item.json', 'folder_child_item.json'
]:
self._add_item(name)
import unittest
from onedrived.api import drives
from onedrived.common.drive_config import DriveConfig
from onedrived.store import drives_db
from tests import get_data
from tests.factory import mock_factory
from tests.factory.account_factory import get_sample_personal_account
from tests.factory.drive_factory import get_sample_drive_object
from tests.store.test_account_db import get_sample_account_storage
mock_factory.mock_register()
class TestDriveStorage(unittest.TestCase):
def setUp(self):
self.personal_account = get_sample_personal_account()
self.account_store = get_sample_account_storage(self.personal_account.client, None)
self.account_store.add_account(self.personal_account)
self.drives_store = drives_db.DriveStorage(':memory:', self.account_store)
self.drive_root = self.drives_store.get_drive_root(
self.personal_account.profile.user_id, self.personal_account.TYPE)
def test_get_drive_root(self):
self.assertIsInstance(self.drive_root, drives.DriveRoot)
self.assertIs(self.drive_root.account, self.personal_account)
def test_get_all_drives(self):
drive_conf = dict(DriveConfig.DEFAULT_VALUES)
drive_conf['local_root'] = '/tmp'
drive = drives.DriveObject(self.drive_root, get_data('drive.json'), DriveConfig(drive_conf))