summaryrefslogtreecommitdiff
path: root/tests/test_property.rs
blob: 9782bd25d6e5ca354234c6b04b0a3a6b2bcc90c0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
// Copyright 2020 Tyler Neely
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

mod util;

use pretty_assertions::assert_eq;

use rocksdb::{properties, Options, DB};
use util::DBPath;

#[test]
fn property_test() {
    let n = DBPath::new("_rust_rocksdb_property_test");
    {
        let db = DB::open_default(&n).unwrap();
        let prop_name: &std::ffi::CStr = properties::STATS;
        let value = db.property_value(prop_name).unwrap().unwrap();
        assert!(value.contains("Stats"));
    }

    {
        let db = DB::open_default(&n).unwrap();
        let prop_name: properties::PropertyName = properties::STATS.to_owned();
        let value = db.property_value(&prop_name).unwrap().unwrap();
        assert!(value.contains("Stats"));
    }

    {
        let db = DB::open_default(&n).unwrap();
        let prop_name: String = properties::STATS.to_owned().into_string();
        let value = db.property_value(&prop_name).unwrap().unwrap();
        assert!(value.contains("Stats"));
    }
}

#[test]
fn property_cf_test() {
    let n = DBPath::new("_rust_rocksdb_property_cf_test");
    {
        let opts = Options::default();
        #[cfg(feature = "multi-threaded-cf")]
        let db = DB::open_default(&n).unwrap();
        #[cfg(not(feature = "multi-threaded-cf"))]
        let mut db = DB::open_default(&n).unwrap();
        db.create_cf("cf1", &opts).unwrap();
        let cf = db.cf_handle("cf1").unwrap();
        let value = db
            .property_value_cf(&cf, properties::STATS)
            .unwrap()
            .unwrap();

        assert!(value.contains("Stats"));
    }
}

#[test]
fn property_int_test() {
    let n = DBPath::new("_rust_rocksdb_property_int_test");
    {
        let db = DB::open_default(&n).unwrap();
        let value = db
            .property_int_value(properties::ESTIMATE_LIVE_DATA_SIZE)
            .unwrap();

        assert_eq!(value, Some(0));
    }
}

#[test]
fn property_int_cf_test() {
    let n = DBPath::new("_rust_rocksdb_property_int_cf_test");
    {
        let opts = Options::default();
        #[cfg(feature = "multi-threaded-cf")]
        let db = DB::open_default(&n).unwrap();
        #[cfg(not(feature = "multi-threaded-cf"))]
        let mut db = DB::open_default(&n).unwrap();
        db.create_cf("cf1", &opts).unwrap();
        let cf = db.cf_handle("cf1").unwrap();
        let total_keys = db
            .property_int_value_cf(&cf, properties::ESTIMATE_NUM_KEYS)
            .unwrap();

        assert_eq!(total_keys, Some(0));
    }
}