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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
from sqlalchemy import Table, Column, MetaData
from elixir import *
from elixir.ext.list import acts_as_list
class TestActsAsList(object):
def teardown(self):
cleanup_all(True)
def test_acts_as_list(self):
class ToDo(Entity):
subject = Field(String(128))
owner = ManyToOne('Person')
def qualify(self):
return ToDo.owner_id == self.owner_id
acts_as_list(qualifier=qualify, column_name='position')
def __repr__(self):
return '<%d:%s>' % (self.position, self.subject)
class Person(Entity):
name = Field(String(64))
todos = OneToMany('ToDo', order_by='position')
metadata.bind = 'sqlite:///'
setup_all(True)
# create a person
# you must create and commit this _before_ you attach todo's to it
# because of the way that the plugin is implemented
p = Person(name='Jonathan')
session.commit(); session.clear()
# add three todos, in the reverse order that we want them
p = Person.get(1)
p.todos.append(ToDo(subject='Three'))
p.todos.append(ToDo(subject='Two'))
p.todos.append(ToDo(subject='One'))
session.commit(); session.clear()
# move the first item lower
p = Person.get(1)
p.todos[0].move_lower()
session.commit(); session.clear()
# validate it worked
p = Person.get(1)
assert p.todos[0].subject == 'Two'
assert p.todos[1].subject == 'Three'
assert p.todos[2].subject == 'One'
# move the last item to the top to put things in correct order
p.todos[2].move_to_top()
session.commit(); session.clear()
# validate it worked
p = Person.get(1)
assert p.todos[0].subject == 'One'
assert p.todos[1].subject == 'Two'
assert p.todos[2].subject == 'Three'
# lets shuffle them again for the sake of testing move_to_bottom
p.todos[2].move_to_top()
session.commit(); session.clear()
p = Person.get(1)
assert p.todos[0].subject == 'Three'
assert p.todos[1].subject == 'One'
assert p.todos[2].subject == 'Two'
p.todos[1].move_to_bottom()
session.commit(); session.clear()
p = Person.get(1)
assert p.todos[0].subject == 'Three'
assert p.todos[1].subject == 'Two'
assert p.todos[2].subject == 'One'
# lets move everything back to how it should be
p = Person.get(1)
p.todos[0].move_to(3)
p.todos[2].move_to(1)
session.commit(); session.clear()
# validate it worked
p = Person.get(1)
assert p.todos[0].subject == 'One'
assert p.todos[0].position == 1
assert p.todos[1].subject == 'Two'
assert p.todos[1].position == 2
assert p.todos[2].subject == 'Three'
assert p.todos[2].position == 3
# delete the second todo list item
p.todos[1].delete()
session.commit(); session.clear()
# validate that the deletion worked, and sequence numebers
# were properly managed
p = Person.get(1)
assert p.todos[0].subject == 'One'
assert p.todos[0].position == 1
assert p.todos[1].subject == 'Three'
assert p.todos[1].position == 2
def test_acts_as_list_autoload(self):
# Make autoload test fixture
meta = MetaData('sqlite:///')
preloaded_table = Table('preloaded', meta,
Column('name', String(32), primary_key=True),
Column('position', Integer))
meta.create_all()
class Preloaded(Entity):
using_options(tablename='preloaded', autoload=True)
acts_as_list()
metadata.bind = meta.bind
setup_all()
i = Preloaded(name='Foo')
session.commit()
assert i.name == 'Foo'
assert i.position == 1
j = Preloaded(name='Bar')
session.commit()
assert j.name == 'Bar'
assert j.position == 2
|
