1+ """
2+ Azure Blob Storage test fixtures that load from JSON and XML files
3+ """
4+
5+ import pytest
6+ import io
7+ import json
8+ import os
9+ from pathlib import Path
10+ from waterbutler .core import streams
11+ from waterbutler .providers .azureblobstorage .provider import AzureBlobStorageProvider
12+
13+
14+ # Get the fixtures directory path
15+ FIXTURES_DIR = Path (__file__ ).parent / 'fixtures'
16+
17+
18+ def load_fixture (filename ):
19+ """Load a fixture file from the fixtures directory"""
20+ filepath = FIXTURES_DIR / filename
21+ if filename .endswith ('.xml' ):
22+ with open (filepath , 'r' , encoding = 'utf-8' ) as f :
23+ return f .read ()
24+ else :
25+ with open (filepath , 'r' , encoding = 'utf-8' ) as f :
26+ return json .load (f )
27+
28+
29+ # ============== Authentication Fixtures (JSON) ==============
30+
31+ @pytest .fixture
32+ def auth ():
33+ """Auth information from the user"""
34+ return load_fixture ('auth.json' )
35+
36+
37+ @pytest .fixture
38+ def credentials ():
39+ """OAuth credentials from Azure Entra ID"""
40+ return load_fixture ('credentials.json' )
41+
42+
43+ @pytest .fixture
44+ def settings ():
45+ """Provider settings for Azure Blob Storage"""
46+ return load_fixture ('settings.json' )
47+
48+
49+ # ============== Response Fixtures (Mixed JSON/XML) ==============
50+
51+ @pytest .fixture
52+ def blob_properties_headers ():
53+ """Standard blob properties headers (JSON format for headers)"""
54+ return load_fixture ('blob_properties_headers.json' )
55+
56+
57+ # XML Response Fixtures - Direct from Azure API format
58+ @pytest .fixture
59+ def blob_list_xml ():
60+ """Standard blob list XML response"""
61+ return load_fixture ('blob_list_response.xml' )
62+
63+
64+ @pytest .fixture
65+ def blob_list_with_versions_response ():
66+ """Blob list with version information XML response"""
67+ return load_fixture ('blob_list_with_versions.xml' )
68+
69+
70+ @pytest .fixture
71+ def empty_list_xml ():
72+ """Empty folder list XML response"""
73+ return load_fixture ('empty_list_response.xml' )
74+
75+
76+ @pytest .fixture
77+ def root_list_xml ():
78+ """Root container list XML response"""
79+ return load_fixture ('root_list_response.xml' )
80+
81+
82+ @pytest .fixture
83+ def special_characters_blobs ():
84+ """Blobs with special characters in names XML response"""
85+ return load_fixture ('special_characters_blobs.xml' )
86+
87+
88+ # Error Response Fixtures
89+ @pytest .fixture
90+ def error_authentication_failed_xml ():
91+ """Authentication failed error XML response"""
92+ return load_fixture ('error_authentication_failed.xml' )
93+
94+
95+ @pytest .fixture
96+ def error_authorization_failure_xml ():
97+ """Authorization failure error XML response"""
98+ return load_fixture ('error_authorization_failure.xml' )
99+
100+
101+ @pytest .fixture
102+ def error_not_found_xml ():
103+ """Blob not found error XML response"""
104+ return load_fixture ('error_not_found.xml' )
105+
106+
107+ @pytest .fixture
108+ def error_internal_error_xml ():
109+ """Internal server error XML response"""
110+ return load_fixture ('error_internal_error.xml' )
111+
112+
113+ @pytest .fixture
114+ def error_response_xml ():
115+ """Error response XML generator"""
116+ def _error_xml (error_type ):
117+ try :
118+ return load_fixture (f'error_{ error_type } )
119+ except FileNotFoundError :
120+ # Default error XML
121+ return '''<?xml version="1.0" encoding="utf-8"?>
122+ <Error>
123+ <Code>UnknownError</Code>
124+ <Message>Unknown error occurred</Message>
125+ </Error>'''
126+
127+ return _error_xml
128+
129+
130+ # ============== Provider Fixture ==============
131+
132+ @pytest .fixture
133+ def provider (auth , credentials , settings ):
134+ return AzureBlobStorageProvider (auth , credentials , settings )
135+
136+
137+ # ============== Stream Fixtures ==============
138+
139+ @pytest .fixture
140+ def file_content ():
141+ """Basic file content for testing"""
142+ return b'SLEEP IS FOR THE WEAK GO SERVE STREAMS'
143+
144+
145+ @pytest .fixture
146+ def large_file_content ():
147+ """Large file content for multipart upload testing"""
148+ # 10 MB of data
149+ return b'x' * (10 * 1024 * 1024 )
150+
151+
152+ @pytest .fixture
153+ def file_like (file_content ):
154+ """File-like object"""
155+ return io .BytesIO (file_content )
156+
157+
158+ @pytest .fixture
159+ def large_file_like (large_file_content ):
160+ """Large file-like object"""
161+ return io .BytesIO (large_file_content )
162+
163+
164+ @pytest .fixture
165+ def file_stream (file_like ):
166+ """File stream for upload testing"""
167+ return streams .FileStreamReader (file_like )
168+
169+
170+ @pytest .fixture
171+ def large_file_stream (large_file_like ):
172+ """Large file stream for multipart upload testing"""
173+ return streams .FileStreamReader (large_file_like )
174+
175+
176+ # ============== Folder Creation Fixtures ==============
177+
178+ @pytest .fixture
179+ def empty_folder_list_xml ():
180+ """Empty folder list response for checking if folder exists"""
181+ return load_fixture ('empty_folder_check.xml' )
182+
183+
184+ @pytest .fixture
185+ def folder_validation_response_xml ():
186+ """XML response for folder validation (folder exists)"""
187+ return load_fixture ('folder_validation_response.xml' )
188+
189+
190+ @pytest .fixture
191+ def folder_not_found_response_xml ():
192+ """XML response for folder validation (folder does not exist)"""
193+ return load_fixture ('folder_not_found_response.xml' )
194+
195+
196+ @pytest .fixture
197+ def folder_exists_xml ():
198+ """XML response indicating folder already has content"""
199+ return load_fixture ('folder_exists.xml' )
200+
201+
202+ @pytest .fixture
203+ def folder_placeholder_headers ():
204+ """Standard headers for folder placeholder creation"""
205+ return load_fixture ('folder_placeholder_headers.json' )
206+
207+
208+ @pytest .fixture
209+ def create_folder_test_data ():
210+ """Test data for various folder creation scenarios"""
211+ return {
212+ 'simple_folder' : {
213+ 'path' : '/newfolder/' ,
214+ 'name' : 'newfolder' ,
215+ 'placeholder' : 'newfolder/.osfkeep'
216+ },
217+ 'nested_folder' : {
218+ 'path' : '/parent/child/' ,
219+ 'name' : 'child' ,
220+ 'placeholder' : 'parent/child/.osfkeep'
221+ },
222+ 'special_chars_folder' : {
223+ 'path' : '/folder with spaces/' ,
224+ 'name' : 'folder with spaces' ,
225+ 'placeholder' : 'folder%20with%20spaces/.osfkeep'
226+ }
227+ }
228+
229+
230+ # ============== Helper Functions ==============
231+
232+ @pytest .fixture
233+ def build_error_response ():
234+ """Build a custom error response for testing"""
235+ def _builder (code , message , status = 400 , auth_detail = None ):
236+ xml_body = f'''<?xml version="1.0" encoding="utf-8"?>
237+ <Error>
238+ <Code>{ code }
239+ <Message>{ message }
240+ { f"<AuthenticationErrorDetail>{ auth_detail } if auth_detail else "" }
241+ </Error>'''
242+
243+ return {
244+ 'status' : status ,
245+ 'body' : xml_body ,
246+ 'headers' : {'Content-Type' : 'application/xml' }
247+ }
248+
249+ return _builder
0 commit comments