|
20 | 20 | get_source_file_or_blob_service_client_track2, |
21 | 21 | validate_encryption_source, validate_source_uri, |
22 | 22 | validate_encryption_services, as_user_validator, |
23 | | - get_not_none_validator) |
| 23 | + get_not_none_validator, validate_upload_blob) |
24 | 24 |
|
25 | 25 |
|
26 | 26 | class MockCLI(CLI): |
@@ -183,6 +183,19 @@ def test_get_not_none_validator(self): |
183 | 183 | validate_arg(cmd, Namespace(arg=0)) |
184 | 184 | validate_arg(cmd, Namespace(arg=False)) |
185 | 185 |
|
| 186 | + def test_validate_upload_blob(self): |
| 187 | + from azure.cli.core.azclierror import InvalidArgumentValueError |
| 188 | + |
| 189 | + validate_upload_blob(Namespace(file_path=None, data='')) |
| 190 | + validate_upload_blob(Namespace(file_path=None, data='test data')) |
| 191 | + validate_upload_blob(Namespace(file_path='test.txt', data=None)) |
| 192 | + |
| 193 | + with self.assertRaisesRegex(InvalidArgumentValueError, 'please only specify one of --file and --data'): |
| 194 | + validate_upload_blob(Namespace(file_path='test.txt', data='')) |
| 195 | + |
| 196 | + with self.assertRaisesRegex(InvalidArgumentValueError, 'please specify one of --file and --data'): |
| 197 | + validate_upload_blob(Namespace(file_path=None, data=None)) |
| 198 | + |
186 | 199 |
|
187 | 200 | class TestEncryptionValidators(unittest.TestCase): |
188 | 201 | def setUp(self): |
|
0 commit comments