|
3 | 3 |
|
4 | 4 | from dataclass_csv import DataclassReader, CsvValueError
|
5 | 5 |
|
6 |
| -from .mocks import User, DataclassWithBooleanValue, DataclassWithBooleanValueNoneDefault |
| 6 | +from .mocks import ( |
| 7 | + User, |
| 8 | + DataclassWithBooleanValue, |
| 9 | + DataclassWithBooleanValueNoneDefault, |
| 10 | + UserWithInitFalse, |
| 11 | + UserWithInitFalseAndDefaultValue, |
| 12 | +) |
7 | 13 |
|
8 | 14 |
|
9 | 15 | def test_reader_with_non_dataclass(create_csv):
|
@@ -114,19 +120,41 @@ def test_parse_bool_value_false(create_csv):
|
114 | 120 | def test_parse_bool_value_invalid(create_csv):
|
115 | 121 | csv_file = create_csv({'boolValue': 'notValidBoolean'})
|
116 | 122 | with csv_file.open() as f:
|
117 |
| - try: |
| 123 | + with pytest.raises(CsvValueError): |
118 | 124 | reader = DataclassReader(f, DataclassWithBooleanValue)
|
119 | 125 | list(reader)
|
120 |
| - assert False # Should not be able to successfully parse |
121 |
| - except CsvValueError: |
122 |
| - pass |
123 | 126 |
|
124 | 127 |
|
125 | 128 | def test_parse_bool_value_none_default(create_csv):
|
126 |
| - """Verify that blank/null values are parsed as None for optional fields""" |
127 | 129 | csv_file = create_csv({'boolValue': ''})
|
128 | 130 | with csv_file.open() as f:
|
129 | 131 | reader = DataclassReader(f, DataclassWithBooleanValueNoneDefault)
|
130 | 132 | items = list(reader)
|
131 | 133 | dataclass_instance = items[0]
|
132 | 134 | assert dataclass_instance.boolValue is None
|
| 135 | + |
| 136 | + |
| 137 | +def test_skip_dataclass_field_when_init_is_false(create_csv): |
| 138 | + csv_file = create_csv({'firstname': 'User1', 'lastname': 'TestUser'}) |
| 139 | + with csv_file.open() as f: |
| 140 | + reader = DataclassReader(f, UserWithInitFalse) |
| 141 | + items = list(reader) |
| 142 | + |
| 143 | + |
| 144 | +def test_try_to_access_not_initialized_prop_raise_attr_error(create_csv): |
| 145 | + csv_file = create_csv({'firstname': 'User1', 'lastname': 'TestUser'}) |
| 146 | + with csv_file.open() as f: |
| 147 | + reader = DataclassReader(f, UserWithInitFalse) |
| 148 | + items = list(reader) |
| 149 | + with pytest.raises(AttributeError): |
| 150 | + user = items[0] |
| 151 | + user_age = user.age |
| 152 | + |
| 153 | + |
| 154 | +def test_try_to_access_not_initialized_prop_with_default_value(create_csv): |
| 155 | + csv_file = create_csv({'firstname': 'User1', 'lastname': 'TestUser'}) |
| 156 | + with csv_file.open() as f: |
| 157 | + reader = DataclassReader(f, UserWithInitFalseAndDefaultValue) |
| 158 | + items = list(reader) |
| 159 | + user = items[0] |
| 160 | + assert user.age == 0 |
0 commit comments