|
2 | 2 |
|
3 | 3 |
|
4 | 4 | def test_data_crud_operations(clean_database): |
5 | | - data = Data.create(geode_object="test_object", input_file="test.txt") |
| 5 | + data = Data.create( |
| 6 | + geode_object="test_object", input_file="test.txt", additional_files=[] |
| 7 | + ) |
| 8 | + print("id", data.id, flush=True) |
6 | 9 | assert data.id is not None |
| 10 | + assert isinstance(data.id, str) |
7 | 11 |
|
8 | 12 | retrieved = Data.get(data.id) |
9 | 13 | assert retrieved is not None |
| 14 | + assert isinstance(retrieved, Data) |
10 | 15 | assert retrieved.geode_object == "test_object" |
| 16 | + assert retrieved.input_file == "test.txt" |
| 17 | + assert retrieved.id == data.id |
11 | 18 | non_existent = Data.get("fake_id") |
12 | 19 | assert non_existent is None |
13 | 20 |
|
14 | 21 |
|
15 | 22 | def test_data_with_additional_files(clean_database): |
16 | 23 | files = ["file1.txt", "file2.txt"] |
17 | 24 | data = Data.create(geode_object="test_files", additional_files=files) |
| 25 | + assert data.id is not None |
| 26 | + assert isinstance(data.id, str) |
18 | 27 |
|
19 | 28 | retrieved = Data.get(data.id) |
20 | 29 | assert retrieved is not None |
| 30 | + assert isinstance(retrieved, Data) |
21 | 31 | assert retrieved.additional_files == files |
| 32 | + assert retrieved.geode_object == "test_files" |
0 commit comments