objectbox-python/tests/test_model.py at main · objectbox/objectbox-python · GitHub

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
98
99
100
101
102
from objectbox.model import *
from objectbox import *
from objectbox.model.idsync import sync_model
import os
import os.path

def test_reuse_model():
    @Entity()
    class MyEntity:
        id = Id()
        name = String()

    model = Model()
    model.entity(MyEntity)
    model_filepath = "test-model.json"
    if os.path.exists(model_filepath):
        os.remove(model_filepath)
    sync_model(model, model_filepath)

    db1path = "test-db1"
    db2path = "test-db2"
    Store.remove_db_files(db1path)
    Store.remove_db_files(db2path)

    store1 = Store(model=model, directory=db1path)
    store2 = Store(model=model, directory=db2path)

    store1.close()
    store2.close()


def test_reuse_entity():
    @Entity()
    class MyEntity:
        id = Id()
        name = String()

    m1 = Model()
    m1.entity(MyEntity)
    model_filepath = "test-model1.json"
    if os.path.exists(model_filepath):
        os.remove(model_filepath)
    sync_model(m1, model_filepath)

    db1path = "test-db1"
    db2path = "test-db2"
    Store.remove_db_files(db1path)
    Store.remove_db_files(db2path)

    store1 = Store(model=m1, directory=db1path)

    box1 = store1.box(MyEntity)
    box1.put(MyEntity(name="foo"))
    assert box1.count() == 1

    m2 = Model()

    @Entity()
    class MyEntity2:
        id = Id()
        name = String()
        value = Int64()

    m2.entity(MyEntity2)
    m2.entity(MyEntity)
    model_filepath = "test-model2.json"
    if os.path.exists(model_filepath):
        os.remove(model_filepath)
    sync_model(m2, model_filepath)

    store2 = Store(model=m2, directory=db2path)
    box2 = store2.box(MyEntity)
    box2.put(MyEntity(name="bar"))
    box2.put(MyEntity(name="bar"))
    box2.put(MyEntity(name="bar"))
    assert box2.count() == 3

    box1.put(MyEntity(name="foo"))
    box1.put(MyEntity(name="foo"))
    assert box1.count() == 3

    store1.close()
    store2.close()


def test_property_args():
    """ Properties do not take an ID (it was silently ignored; IDs come from the model JSON file), only a UID, which
    is keyword-only (positional values were IDs before). Unknown arguments raise. """
    import pytest
    for property_class in (Id, Bool, String, Int8, Int32, Int64, Float32, Float64, Bytes, Date, DateNano, Flex,
                           Float32Vector, Int32List, SyncClock, SyncPrecedence):
        with pytest.raises(TypeError, match="Properties do not take an id"):
            property_class(id=3)
        with pytest.raises(TypeError):
            property_class(3, 1234)  # Was (id, uid)
        assert property_class(uid=1234).uid == 1234
    with pytest.raises(TypeError, match="Properties do not take an id"):
        Relation("Target", id=3)
    assert Relation("Target", uid=1234).uid == 1234
    with pytest.raises(TypeError, match="Unexpected property argument.*: index_type"):
        String(index_type=2)
    assert Date(float).uid == 0  # The Python type is still positional