55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
|
import unittest
|
||
|
|
import rocksdb
|
||
|
|
|
||
|
|
class TestFilterPolicy(rocksdb.interfaces.FilterPolicy):
|
||
|
|
def create_filter(self, keys):
|
||
|
|
return 'nix'
|
||
|
|
|
||
|
|
def key_may_match(self, key, fil):
|
||
|
|
return True
|
||
|
|
|
||
|
|
def name(self):
|
||
|
|
return 'testfilter'
|
||
|
|
|
||
|
|
class TestMergeOperator(rocksdb.interfaces.MergeOperator):
|
||
|
|
def full_merge(self, *args, **kwargs):
|
||
|
|
return (False, None)
|
||
|
|
|
||
|
|
def partial_merge(self, *args, **kwargs):
|
||
|
|
return (False, None)
|
||
|
|
|
||
|
|
def name(self):
|
||
|
|
return 'testmergeop'
|
||
|
|
|
||
|
|
class TestOptions(unittest.TestCase):
|
||
|
|
def test_simple(self):
|
||
|
|
opts = rocksdb.Options()
|
||
|
|
self.assertEqual(False, opts.paranoid_checks)
|
||
|
|
opts.paranoid_checks = True
|
||
|
|
self.assertEqual(True, opts.paranoid_checks)
|
||
|
|
|
||
|
|
self.assertIsNone(opts.filter_policy)
|
||
|
|
ob = TestFilterPolicy()
|
||
|
|
opts.filter_policy = ob
|
||
|
|
self.assertEqual(opts.filter_policy, ob)
|
||
|
|
|
||
|
|
self.assertIsNone(opts.merge_operator)
|
||
|
|
ob = TestMergeOperator()
|
||
|
|
opts.merge_operator = ob
|
||
|
|
self.assertEqual(opts.merge_operator, ob)
|
||
|
|
|
||
|
|
self.assertIsInstance(
|
||
|
|
opts.comparator,
|
||
|
|
rocksdb.BytewiseComparator)
|
||
|
|
|
||
|
|
self.assertEqual('snappy_compression', opts.compression)
|
||
|
|
opts.compression = rocksdb.CompressionType.no_compression
|
||
|
|
self.assertEqual('no_compression', opts.compression)
|
||
|
|
|
||
|
|
self.assertEqual(opts.block_size, 4096)
|
||
|
|
|
||
|
|
self.assertIsNone(opts.block_cache)
|
||
|
|
ob = rocksdb.LRUCache(100)
|
||
|
|
opts.block_cache = ob
|
||
|
|
self.assertEqual(ob, opts.block_cache)
|