From 162b0568d2465bf20aac1e97183e1061a8760319 Mon Sep 17 00:00:00 2001
From: kntkymt <kntkc0de@gmail.com>
Date: Sat, 3 Feb 2024 21:37:54 +0900
Subject: [PATCH] Add Test for Optimizer

---
 .../xcshareddata/xcschemes/CCompiler.xcscheme | 10 +++++
 Package.swift                                 |  4 ++
 Sources/Generator/AsmRepresentation.swift     |  6 +--
 .../RemoveImmediatePushPopPathTest.swift      | 39 +++++++++++++++++++
 4 files changed, 56 insertions(+), 3 deletions(-)
 create mode 100644 Tests/OptimizerTest/RemoveImmediatePushPopPathTest.swift

diff --git a/.swiftpm/xcode/xcshareddata/xcschemes/CCompiler.xcscheme b/.swiftpm/xcode/xcshareddata/xcschemes/CCompiler.xcscheme
index f8397a6..44c5f07 100644
--- a/.swiftpm/xcode/xcshareddata/xcschemes/CCompiler.xcscheme
+++ b/.swiftpm/xcode/xcshareddata/xcschemes/CCompiler.xcscheme
@@ -111,6 +111,16 @@
                ReferencedContainer = "container:">
             </BuildableReference>
          </TestableReference>
+         <TestableReference
+            skipped = "NO">
+            <BuildableReference
+               BuildableIdentifier = "primary"
+               BlueprintIdentifier = "OptimizerTest"
+               BuildableName = "OptimizerTest"
+               BlueprintName = "OptimizerTest"
+               ReferencedContainer = "container:">
+            </BuildableReference>
+         </TestableReference>
       </Testables>
    </TestAction>
    <LaunchAction
diff --git a/Package.swift b/Package.swift
index 6a63c96..427fda6 100644
--- a/Package.swift
+++ b/Package.swift
@@ -63,6 +63,10 @@ let package = Package(
             name: "CCompilerCoreTest",
             dependencies: ["CCompilerCore"]
         ),
+        .testTarget(
+            name: "OptimizerTest",
+            dependencies: ["Optimizer"]
+        ),
         .testTarget(
             name: "GeneratorTest",
             dependencies: ["Generator"]
diff --git a/Sources/Generator/AsmRepresentation.swift b/Sources/Generator/AsmRepresentation.swift
index 5e75093..ab25528 100644
--- a/Sources/Generator/AsmRepresentation.swift
+++ b/Sources/Generator/AsmRepresentation.swift
@@ -1,7 +1,7 @@
 
 public enum AsmRepresent {
 
-    public enum Instruction {
+    public enum Instruction: Equatable {
         case mov(dst: Register, src: Register)
         case movi(dst: Register, immediate: Int64)
 
@@ -49,7 +49,7 @@ public enum AsmRepresent {
         case globl(label: String)
         case dataDecl(kind: DataKind, value: String)
 
-        public enum Address {
+        public enum Address: Equatable {
             case register(_ register: Register)
             case distance(_ register: Register, _ diff: Int64)
         }
@@ -79,7 +79,7 @@ public enum AsmRepresent {
         }
     }
 
-    public enum Register: String {
+    public enum Register: String, Equatable {
         case x0
         case x1
         case x2
diff --git a/Tests/OptimizerTest/RemoveImmediatePushPopPathTest.swift b/Tests/OptimizerTest/RemoveImmediatePushPopPathTest.swift
new file mode 100644
index 0000000..cb74ad0
--- /dev/null
+++ b/Tests/OptimizerTest/RemoveImmediatePushPopPathTest.swift
@@ -0,0 +1,39 @@
+import XCTest
+@testable import Optimizer
+import Generator
+
+final class RemoveImmediatePushPopPathTest: XCTestCase {
+
+    func testRemoved() {
+        let instructions: [AsmRepresent.Instruction] = [
+            .movi(dst: .x0, immediate: 10),
+            .push(src: .x0),
+            .pop(dst: .x0),
+            .ret
+        ]
+
+        let result = RemoveImmediatePushPopPath().optimize(instructions: instructions)
+        let expected: [AsmRepresent.Instruction] = [
+            instructions[0],
+            instructions[3]
+        ]
+
+        XCTAssertEqual(result, expected)
+    }
+
+    func testNotRemoved() {
+        let instructions: [AsmRepresent.Instruction] = [
+            .movi(dst: .x0, immediate: 10),
+            .push(src: .x0),
+            .add(dst: .x0, src1: .x1, src2: .x0),
+            .mov(dst: .x1, src: .x0),
+            .pop(dst: .x0),
+            .ret
+        ]
+
+        let result = RemoveImmediatePushPopPath().optimize(instructions: instructions)
+        let expected: [AsmRepresent.Instruction] = instructions
+
+        XCTAssertEqual(result, expected)
+    }
+}