@@ -56,12 +56,7 @@ def __init__(self):
56
56
57
57
self .put_settings = pretend .call_recorder (lambda * a , ** kw : None )
58
58
self .forcemerge = pretend .call_recorder (lambda * a , ** kw : None )
59
-
60
- def create (self , index , body ):
61
- self .indices [index ] = body
62
-
63
- def delete (self , index ):
64
- self .indices .pop (index , None )
59
+ self .delete = pretend .call_recorder (lambda * a , ** kw : None )
65
60
66
61
def exists_alias (self , name ):
67
62
return name in self .aliases
@@ -138,6 +133,8 @@ def parallel_bulk(client, iterable):
138
133
monkeypatch .setattr (
139
134
warehouse .cli .search .reindex , "parallel_bulk" , parallel_bulk )
140
135
136
+ monkeypatch .setattr (os , "urandom" , lambda n : b"\xcb " * n )
137
+
141
138
result = cli .invoke (reindex , obj = config )
142
139
143
140
assert result .exit_code == - 1
@@ -148,7 +145,9 @@ def parallel_bulk(client, iterable):
148
145
]
149
146
assert sess_obj .rollback .calls == [pretend .call ()]
150
147
assert sess_obj .close .calls == [pretend .call ()]
151
- assert es_client .indices .indices == {}
148
+ assert es_client .indices .delete .calls == [
149
+ pretend .call (index = 'warehouse-cbcbcbcbcb' ),
150
+ ]
152
151
assert es_client .indices .put_settings .calls == []
153
152
assert es_client .indices .forcemerge .calls == []
154
153
@@ -199,7 +198,7 @@ def project_docs(db):
199
198
assert parallel_bulk .calls == [pretend .call (es_client , docs )]
200
199
assert sess_obj .rollback .calls == [pretend .call ()]
201
200
assert sess_obj .close .calls == [pretend .call ()]
202
- assert set ( es_client .indices .indices ) == { "warehouse-cbcbcbcbcb" }
201
+ assert es_client .indices .delete . calls == []
203
202
assert es_client .indices .aliases == {
204
203
"warehouse" : ["warehouse-cbcbcbcbcb" ],
205
204
}
@@ -267,7 +266,9 @@ def project_docs(db):
267
266
assert parallel_bulk .calls == [pretend .call (es_client , docs )]
268
267
assert sess_obj .rollback .calls == [pretend .call ()]
269
268
assert sess_obj .close .calls == [pretend .call ()]
270
- assert set (es_client .indices .indices ) == {"warehouse-cbcbcbcbcb" }
269
+ assert es_client .indices .delete .calls == [
270
+ pretend .call ('warehouse-aaaaaaaaaa' ),
271
+ ]
271
272
assert es_client .indices .aliases == {
272
273
"warehouse" : ["warehouse-cbcbcbcbcb" ],
273
274
}
0 commit comments