|
6 | 6 | from unittest import mock
|
7 | 7 |
|
8 | 8 | from click.testing import Result
|
| 9 | +from vdk.internal.core.errors import ResolvableByActual |
9 | 10 | from vdk.internal.core.errors import UserCodeError
|
10 | 11 | from vdk.plugin.csv import csv_plugin
|
11 | 12 | from vdk.plugin.sqlite import sqlite_plugin
|
@@ -150,32 +151,32 @@ def test_export_csv_with_already_existing_file(tmpdir):
|
150 | 151 | cli_assert_equal(1, result)
|
151 | 152 |
|
152 | 153 |
|
153 |
| -# def test_csv_export_with_nonexistent_table(tmpdir): |
154 |
| -# db_dir = str(tmpdir) + "vdk-sqlite.db" |
155 |
| -# with mock.patch.dict( |
156 |
| -# os.environ, |
157 |
| -# { |
158 |
| -# "VDK_DB_DEFAULT_TYPE": "SQLITE", |
159 |
| -# "VDK_SQLITE_FILE": db_dir, |
160 |
| -# }, |
161 |
| -# ): |
162 |
| -# runner = CliEntryBasedTestRunner(sqlite_plugin, csv_plugin) |
163 |
| -# drop_table(runner, "test_table") |
164 |
| -# result = runner.invoke( |
165 |
| -# [ |
166 |
| -# "export-csv", |
167 |
| -# "--query", |
168 |
| -# "SELECT * FROM test_table", |
169 |
| -# "--file", |
170 |
| -# "result3.csv", |
171 |
| -# ] |
172 |
| -# ) |
173 |
| -# assert isinstance(result.exception, OperationalError) |
174 |
| -# assert hasattr(result.exception, "_vdk_resolvable_actual") |
175 |
| -# assert ( |
176 |
| -# getattr(result.exception, "_vdk_resolvable_actual") |
177 |
| -# == ResolvableByActual.PLATFORM |
178 |
| -# ) |
| 154 | +def test_csv_export_with_nonexistent_table(tmpdir): |
| 155 | + db_dir = str(tmpdir) + "vdk-sqlite.db" |
| 156 | + with mock.patch.dict( |
| 157 | + os.environ, |
| 158 | + { |
| 159 | + "VDK_DB_DEFAULT_TYPE": "SQLITE", |
| 160 | + "VDK_SQLITE_FILE": db_dir, |
| 161 | + }, |
| 162 | + ): |
| 163 | + runner = CliEntryBasedTestRunner(sqlite_plugin, csv_plugin) |
| 164 | + drop_table(runner, "test_table") |
| 165 | + result = runner.invoke( |
| 166 | + [ |
| 167 | + "export-csv", |
| 168 | + "--query", |
| 169 | + "SELECT * FROM test_table", |
| 170 | + "--file", |
| 171 | + "result3.csv", |
| 172 | + ] |
| 173 | + ) |
| 174 | + assert isinstance(result.exception, OperationalError) |
| 175 | + assert hasattr(result.exception, "_vdk_resolvable_actual") |
| 176 | + assert ( |
| 177 | + getattr(result.exception, "_vdk_resolvable_actual") |
| 178 | + == ResolvableByActual.PLATFORM |
| 179 | + ) |
179 | 180 |
|
180 | 181 |
|
181 | 182 | def test_csv_export_with_no_data(tmpdir):
|
|
0 commit comments