9
9
10
10
@pytest .mark .asyncio
11
11
async def test_async_main_no_stderr (monkeypatch ):
12
- mock_cli_args = MagicMock (
13
- no_stderr = True , project_root = "." , action = CliAction .version
14
- )
12
+ mock_cli_args = Config (no_stderr = True , project_root = "." , action = CliAction .version )
15
13
monkeypatch .setattr (
16
14
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
17
15
)
@@ -24,9 +22,7 @@ async def test_async_main_no_stderr(monkeypatch):
24
22
25
23
@pytest .mark .asyncio
26
24
async def test_async_main_default_project_root (monkeypatch ):
27
- mock_cli_args = MagicMock (
28
- no_stderr = False , project_root = None , action = CliAction .version
29
- )
25
+ mock_cli_args = Config (no_stderr = False , project_root = None , action = CliAction .version )
30
26
monkeypatch .setattr (
31
27
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
32
28
)
@@ -42,9 +38,7 @@ async def test_async_main_default_project_root(monkeypatch):
42
38
43
39
@pytest .mark .asyncio
44
40
async def test_async_main_ioerror (monkeypatch ):
45
- mock_cli_args = MagicMock (
46
- no_stderr = False , project_root = "." , action = CliAction .version
47
- )
41
+ mock_cli_args = Config (no_stderr = False , project_root = "." , action = CliAction .version )
48
42
monkeypatch .setattr (
49
43
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
50
44
)
@@ -61,7 +55,7 @@ async def test_async_main_ioerror(monkeypatch):
61
55
62
56
@pytest .mark .asyncio
63
57
async def test_async_main_cli_action_check (monkeypatch ):
64
- mock_cli_args = MagicMock (no_stderr = False , project_root = "." , action = CliAction .check )
58
+ mock_cli_args = Config (no_stderr = False , project_root = "." , action = CliAction .check )
65
59
monkeypatch .setattr (
66
60
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
67
61
)
@@ -79,7 +73,7 @@ async def test_async_main_cli_action_check(monkeypatch):
79
73
80
74
@pytest .mark .asyncio
81
75
async def test_async_main_cli_action_init (monkeypatch ):
82
- mock_cli_args = MagicMock (no_stderr = False , project_root = "." , action = CliAction .init )
76
+ mock_cli_args = Config (no_stderr = False , project_root = "." , action = CliAction .init )
83
77
monkeypatch .setattr (
84
78
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
85
79
)
@@ -94,15 +88,15 @@ async def test_async_main_cli_action_init(monkeypatch):
94
88
95
89
@pytest .mark .asyncio
96
90
async def test_async_main_cli_action_chunks (monkeypatch ):
97
- mock_cli_args = MagicMock (
98
- no_stderr = False , project_root = "." , action = CliAction .chunks
99
- )
91
+ mock_cli_args = Config (no_stderr = False , project_root = "." , action = CliAction .chunks )
100
92
monkeypatch .setattr (
101
93
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
102
94
)
103
95
mock_chunks = AsyncMock (return_value = 0 )
104
96
monkeypatch .setattr ("vectorcode.subcommands.chunks" , mock_chunks )
105
- monkeypatch .setattr ("vectorcode.main.get_project_config" , AsyncMock ())
97
+ monkeypatch .setattr (
98
+ "vectorcode.main.get_project_config" , AsyncMock (return_value = Config ())
99
+ )
106
100
monkeypatch .setattr ("vectorcode.common.try_server" , AsyncMock (return_value = True ))
107
101
108
102
return_code = await async_main ()
@@ -112,9 +106,7 @@ async def test_async_main_cli_action_chunks(monkeypatch):
112
106
113
107
@pytest .mark .asyncio
114
108
async def test_async_main_cli_action_version (monkeypatch , capsys ):
115
- mock_cli_args = MagicMock (
116
- no_stderr = False , project_root = "." , action = CliAction .version
117
- )
109
+ mock_cli_args = Config (no_stderr = False , project_root = "." , action = CliAction .version )
118
110
monkeypatch .setattr (
119
111
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
120
112
)
@@ -127,13 +119,15 @@ async def test_async_main_cli_action_version(monkeypatch, capsys):
127
119
128
120
@pytest .mark .asyncio
129
121
async def test_async_main_cli_action_prompts (monkeypatch ):
130
- mock_cli_args = MagicMock (project_root = "." , action = CliAction .prompts )
122
+ mock_cli_args = Config (project_root = "." , action = CliAction .prompts )
131
123
monkeypatch .setattr (
132
124
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
133
125
)
134
126
mock_prompts = MagicMock (return_value = 0 )
135
127
monkeypatch .setattr ("vectorcode.subcommands.prompts" , mock_prompts )
136
- monkeypatch .setattr ("vectorcode.main.get_project_config" , AsyncMock ())
128
+ monkeypatch .setattr (
129
+ "vectorcode.main.get_project_config" , AsyncMock (return_value = Config ())
130
+ )
137
131
138
132
return_code = await async_main ()
139
133
assert return_code == 0
@@ -142,12 +136,12 @@ async def test_async_main_cli_action_prompts(monkeypatch):
142
136
143
137
@pytest .mark .asyncio
144
138
async def test_async_main_cli_action_query (monkeypatch ):
145
- mock_cli_args = MagicMock (no_stderr = False , project_root = "." , action = CliAction .query )
139
+ mock_cli_args = Config (no_stderr = False , project_root = "." , action = CliAction .query )
146
140
monkeypatch .setattr (
147
141
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
148
142
)
149
- mock_final_configs = MagicMock (
150
- host = " test_host" , port = 1234 , action = CliAction .query , pipe = False
143
+ mock_final_configs = Config (
144
+ db_url = "http:// test_host: 1234" , action = CliAction .query , pipe = False
151
145
)
152
146
monkeypatch .setattr (
153
147
"vectorcode.main.get_project_config" ,
@@ -168,7 +162,7 @@ async def test_async_main_cli_action_query(monkeypatch):
168
162
169
163
@pytest .mark .asyncio
170
164
async def test_async_main_cli_action_vectorise (monkeypatch ):
171
- mock_cli_args = MagicMock (
165
+ mock_cli_args = Config (
172
166
no_stderr = False ,
173
167
project_root = "." ,
174
168
action = CliAction .vectorise ,
@@ -177,8 +171,8 @@ async def test_async_main_cli_action_vectorise(monkeypatch):
177
171
monkeypatch .setattr (
178
172
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
179
173
)
180
- mock_final_configs = MagicMock (
181
- host = " test_host" , port = 1234 , action = CliAction .vectorise , include_hidden = True
174
+ mock_final_configs = Config (
175
+ db_url = "http:// test_host: 1234" , action = CliAction .vectorise , include_hidden = True
182
176
)
183
177
monkeypatch .setattr (
184
178
"vectorcode.main.get_project_config" ,
@@ -199,11 +193,11 @@ async def test_async_main_cli_action_vectorise(monkeypatch):
199
193
200
194
@pytest .mark .asyncio
201
195
async def test_async_main_cli_action_drop (monkeypatch ):
202
- mock_cli_args = MagicMock (no_stderr = False , project_root = "." , action = CliAction .drop )
196
+ mock_cli_args = Config (no_stderr = False , project_root = "." , action = CliAction .drop )
203
197
monkeypatch .setattr (
204
198
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
205
199
)
206
- mock_final_configs = MagicMock ( host = " test_host" , port = 1234 , action = CliAction .drop )
200
+ mock_final_configs = Config ( db_url = "http:// test_host: 1234" , action = CliAction .drop )
207
201
monkeypatch .setattr (
208
202
"vectorcode.main.get_project_config" ,
209
203
AsyncMock (
@@ -223,11 +217,11 @@ async def test_async_main_cli_action_drop(monkeypatch):
223
217
224
218
@pytest .mark .asyncio
225
219
async def test_async_main_cli_action_ls (monkeypatch ):
226
- mock_cli_args = MagicMock (no_stderr = False , project_root = "." , action = CliAction .ls )
220
+ mock_cli_args = Config (no_stderr = False , project_root = "." , action = CliAction .ls )
227
221
monkeypatch .setattr (
228
222
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
229
223
)
230
- mock_final_configs = MagicMock ( host = " test_host" , port = 1234 , action = CliAction .ls )
224
+ mock_final_configs = Config ( db_url = "http:// test_host: 1234" , action = CliAction .ls )
231
225
monkeypatch .setattr (
232
226
"vectorcode.main.get_project_config" ,
233
227
AsyncMock (
@@ -259,13 +253,11 @@ async def test_async_main_cli_action_files(monkeypatch):
259
253
260
254
@pytest .mark .asyncio
261
255
async def test_async_main_cli_action_update (monkeypatch ):
262
- mock_cli_args = MagicMock (
263
- no_stderr = False , project_root = "." , action = CliAction .update
264
- )
256
+ mock_cli_args = Config (no_stderr = False , project_root = "." , action = CliAction .update )
265
257
monkeypatch .setattr (
266
258
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
267
259
)
268
- mock_final_configs = MagicMock ( host = " test_host" , port = 1234 , action = CliAction .update )
260
+ mock_final_configs = Config ( db_url = "http:// test_host: 1234" , action = CliAction .update )
269
261
monkeypatch .setattr (
270
262
"vectorcode.main.get_project_config" ,
271
263
AsyncMock (
@@ -285,11 +277,11 @@ async def test_async_main_cli_action_update(monkeypatch):
285
277
286
278
@pytest .mark .asyncio
287
279
async def test_async_main_cli_action_clean (monkeypatch ):
288
- mock_cli_args = MagicMock (no_stderr = False , project_root = "." , action = CliAction .clean )
280
+ mock_cli_args = Config (no_stderr = False , project_root = "." , action = CliAction .clean )
289
281
monkeypatch .setattr (
290
282
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
291
283
)
292
- mock_final_configs = MagicMock ( host = " test_host" , port = 1234 , action = CliAction .clean )
284
+ mock_final_configs = Config ( db_url = "http:// test_host: 1234" , action = CliAction .clean )
293
285
monkeypatch .setattr (
294
286
"vectorcode.main.get_project_config" ,
295
287
AsyncMock (
@@ -309,11 +301,11 @@ async def test_async_main_cli_action_clean(monkeypatch):
309
301
310
302
@pytest .mark .asyncio
311
303
async def test_async_main_exception_handling (monkeypatch ):
312
- mock_cli_args = MagicMock (no_stderr = False , project_root = "." , action = CliAction .query )
304
+ mock_cli_args = Config (no_stderr = False , project_root = "." , action = CliAction .query )
313
305
monkeypatch .setattr (
314
306
"vectorcode.main.parse_cli_args" , AsyncMock (return_value = mock_cli_args )
315
307
)
316
- mock_final_configs = MagicMock ( host = " test_host" , port = 1234 , action = CliAction .query )
308
+ mock_final_configs = Config ( db_url = "http:// test_host: 1234" , action = CliAction .query )
317
309
monkeypatch .setattr (
318
310
"vectorcode.main.get_project_config" ,
319
311
AsyncMock (
0 commit comments