@@ -8,14 +8,24 @@ import AST
8
8
class MockGeneratorTests : XCTestCase {
9
9
10
10
var trackedLines : Int ?
11
-
11
+
12
12
// The test project is copied to the resources directory build phases
13
13
let testProject = Bundle ( for: MockGeneratorTests . self) . resourcePath! + " /TestProject "
14
14
15
+ override func setUp( ) {
16
+ super. setUp ( )
17
+ formatterFactory = FormatterFactory { DefaultFormatter ( useTabs: false , indentationWidth: 4 ) }
18
+ }
19
+
15
20
func test_generatesSimpleMock( ) {
16
21
assertMockGeneratesExpected ( " SimpleProtocolMock " )
17
22
}
18
23
24
+ func test_generatesSimpleMockWithTabs( ) {
25
+ formatterFactory = FormatterFactory { DefaultFormatter ( useTabs: true , indentationWidth: 4 ) }
26
+ assertMockGeneratesExpected ( " TabSimpleProtocolMock " )
27
+ }
28
+
19
29
func test_deletesMockBody( ) {
20
30
assertMockGeneratesExpected ( " DeleteBodyMock " )
21
31
}
@@ -179,7 +189,7 @@ class MockGeneratorTests: XCTestCase {
179
189
func test_subscriptInClass( ) {
180
190
assertMockGeneratesExpected ( " ClassSubscriptMock " )
181
191
}
182
-
192
+
183
193
func test_overriddenSubscripts( ) {
184
194
assertMockGeneratesExpected ( " AugmentedClassSubscriptMock " )
185
195
}
@@ -215,8 +225,6 @@ class MockGeneratorTests: XCTestCase {
215
225
_ = try Generator (
216
226
projectURL: URL ( fileURLWithPath: testProject) ,
217
227
templateName: " spy " ,
218
- useTabsForIndentation: false ,
219
- indentationWidth: 4 ,
220
228
trackLines: { self . trackedLines = $0 }
221
229
) . execute ( buffer: buffer)
222
230
}
@@ -238,8 +246,6 @@ class MockGeneratorTests: XCTestCase {
238
246
_ = try Generator (
239
247
projectURL: URL ( fileURLWithPath: testProject) ,
240
248
templateName: " spy " ,
241
- useTabsForIndentation: false ,
242
- indentationWidth: 4 ,
243
249
trackLines: { self . trackedLines = $0 }
244
250
) . execute ( buffer: buffer)
245
251
} catch {
@@ -276,7 +282,7 @@ class MockGeneratorTests: XCTestCase {
276
282
let nsError = error as NSError ?
277
283
XCTAssertEqual ( nsError? . localizedDescription, expectedError, line: line)
278
284
}
279
-
285
+
280
286
private func readMock( named fileName: String ) -> ( String , String ) {
281
287
let mock = readFile ( named: fileName + " .swift " )
282
288
let expected = readFile ( named: fileName + " _expected.swift " )
@@ -301,8 +307,6 @@ class MockGeneratorTests: XCTestCase {
301
307
let generator = Generator (
302
308
projectURL: URL ( fileURLWithPath: testProject) ,
303
309
templateName: templateName,
304
- useTabsForIndentation: false ,
305
- indentationWidth: 4 ,
306
310
trackLines: { self . trackedLines = $0 }
307
311
)
308
312
let result = try CommandTestHelper ( ) . execute ( buffer: buffer, command: generator)
0 commit comments