Add som units tests (split some code to made some part testable)

Sources/ResgenSwift/Images/Generator/ImageExtensionGenerator.swift

@@ -9,38 +9,20 @@ import ToolCore
 import Foundation
 
 class ImageExtensionGenerator {
-  
-    // MARK: - Extension files
-    
-    static func writeStringsFiles(images: [ParsedImage], staticVar: Bool, inputFilename: String, extensionName: String, extensionFilePath: String) {
-        // Get header/footer
-        let extensionHeader = Self.getHeader(inputFilename: inputFilename, extensionClassname: extensionName)
-        let extensionFooter = Self.getFooter()
-        
-        // Create content
-        let extensionContent: String = {
-            var content = ""
-            images.forEach { img in
-                if staticVar {
-                    content += "\n\(img.getStaticImageProperty())"
-                } else {
-                    content += "\n\(img.getImageProperty())"
-                }
-                content += "\n "
-            }
-            return content
-        }()
-        
-        // Generate extension
-        Self.generateExtensionFile(extensionFilePath: extensionFilePath, extensionHeader, extensionContent, extensionFooter)
-    }
     
     // MARK: - pragm
     
-    private static func generateExtensionFile(extensionFilePath: String, _ args: String...) {
-        // Create extension content
-        let extensionContent = args.joined(separator: "\n")
-
+    static func generateExtensionFile(images: [ParsedImage],
+                                      staticVar: Bool,
+                                      inputFilename: String,
+                                      extensionName: String,
+                                      extensionFilePath: String) {
+        // Create extension conten1t
+        let extensionContent = Self.getExtensionContent(images: images,
+                                                        staticVar: staticVar,
+                                                        extensionName: extensionName,
+                                                        inputFilename: inputFilename)
+        
         // Write content
         let extensionFilePathURL = URL(fileURLWithPath: extensionFilePath)
         do {
@@ -52,9 +34,22 @@ class ImageExtensionGenerator {
         }
     }
     
+    // MARK: - Extension content
+    
+    static func getExtensionContent(images: [ParsedImage], staticVar: Bool, extensionName: String, inputFilename: String) -> String {
+        [
+            Self.getHeader(inputFilename: inputFilename, extensionClassname: extensionName),
+            Self.getProperties(images: images, staticVar: staticVar),
+            Self.getFooter()
+        ]
+            .joined(separator: "\n")
+    }
+    
+    // MARK: - Extension part
+    
     private static func getHeader(inputFilename: String, extensionClassname: String) -> String {
         """
-        // Generated by ResgenSwift.Imagium \(ResgenSwiftVersion)
+        // Generated by ResgenSwift.\(Images.toolName) \(ResgenSwiftVersion)
         // Images from \(inputFilename)
         
         import UIKit
@@ -63,13 +58,20 @@ class ImageExtensionGenerator {
         """
     }
     
+    private static func getProperties(images: [ParsedImage], staticVar: Bool) -> String {
+        if staticVar {
+            return images
+                .map { "\n\($0.getStaticImageProperty())" }
+                .joined(separator: "\n")
+        }
+        return images
+            .map { "\n\($0.getImageProperty())" }
+            .joined(separator: "\n")
+    }
+    
     private static func getFooter() -> String {
         """
         }
         """
     }
 }
-
-//@objc var onboarding_foreground3: UIImage {
-//        return UIImage(named: "onboarding_foreground3")!
-//    }

Sources/ResgenSwift/Images/Images.swift

@@ -52,11 +52,11 @@ struct Images: ParsableCommand {
                                            xcassetsPath: options.xcassetsPath)
         
         // Generate extension
-        ImageExtensionGenerator.writeStringsFiles(images: imagesToGenerate,
-                                                  staticVar: options.staticMembers,
-                                                  inputFilename: options.inputFilenameWithoutExt,
-                                                  extensionName: options.extensionName,
-                                                  extensionFilePath: options.extensionFilePath)
+        ImageExtensionGenerator.generateExtensionFile(images: imagesToGenerate,
+                                                      staticVar: options.staticMembers,
+                                                      inputFilename: options.inputFilenameWithoutExt,
+                                                      extensionName: options.extensionName,
+                                                      extensionFilePath: options.extensionFilePath)
         
         
         print("[\(Self.toolName)] Images generated")

Sources/ResgenSwift/Images/Parser/ImageFileParser.swift

@@ -13,10 +13,13 @@ class ImageFileParser {
         let inputFileContent = try! String(contentsOfFile: inputFile, encoding: .utf8)
         let stringsByLines = inputFileContent.components(separatedBy: .newlines)
         
+        return Self.parseLines(stringsByLines, platform: platform)
+    }
+    
+    static func parseLines(_ lines: [String], platform: PlatormTag) -> [ParsedImage] {
         var imagesToGenerate = [ParsedImage]()
         
-        // Parse file
-        stringsByLines.forEach {
+        lines.forEach {
             guard $0.removeLeadingTrailingWhitespace().isEmpty == false, $0.first != "#" else {
                 return
             }