lib/assets/SnapshotHelper.swift in snapshot-1.2.1 vs lib/assets/SnapshotHelper.swift in snapshot-1.2.2

@@ -9,53 +9,46 @@
 import Foundation
 import XCTest
 
 var deviceLanguage = ""
 
-func setLanguage(app: XCUIApplication)
-{
+func setLanguage(app: XCUIApplication) {
     Snapshot.setLanguage(app)
 }
 
-func snapshot(name: String, waitForLoadingIndicator: Bool = false)
-{
+func snapshot(name: String, waitForLoadingIndicator: Bool = false) {
     Snapshot.snapshot(name, waitForLoadingIndicator: waitForLoadingIndicator)
 }
 
-
-
-@objc class Snapshot: NSObject
-{
-    class func setLanguage(app: XCUIApplication)
-    {
+class Snapshot: NSObject {
+    
+    class func setLanguage(app: XCUIApplication) {
         let path = "/tmp/language.txt"
-
+        
         do {
             let locale = try NSString(contentsOfFile: path, encoding: NSUTF8StringEncoding) as String
             deviceLanguage = locale.substringToIndex(locale.startIndex.advancedBy(2, limit:locale.endIndex))
-            app.launchArguments += ["-AppleLanguages", "(\(deviceLanguage))", "-AppleLocale", "\"\(locale)\"","-ui_testing"]
+            app.launchArguments += ["-AppleLanguages", "(\(deviceLanguage))", "-AppleLocale", "\"\(locale)\"", "-ui_testing"]
         } catch {
             print("Couldn't detect/set language...")
         }
     }
     
-    class func snapshot(name: String, waitForLoadingIndicator: Bool = false)
-    {
-        if (waitForLoadingIndicator)
-        {
+    class func snapshot(name: String, waitForLoadingIndicator: Bool = false) {
+        if waitForLoadingIndicator {
             waitForLoadingIndicatorToDisappear()
         }
+        
         print("snapshot: \(name)") // more information about this, check out https://github.com/krausefx/snapshot
         
         sleep(1) // Waiting for the animation to be finished (kind of)
         XCUIDevice.sharedDevice().orientation = .Unknown
     }
     
-    class func waitForLoadingIndicatorToDisappear()
-    {
+    class func waitForLoadingIndicatorToDisappear() {
         let query = XCUIApplication().statusBars.childrenMatchingType(.Other).elementBoundByIndex(1).childrenMatchingType(.Other)
         
-        while (query.count > 4) {
+        while query.count > 4 {
             sleep(1)
             print("Number of Elements in Status Bar: \(query.count)... waiting for status bar to disappear")
         }
     }
 }