Merge pull request #28 from onflow/signature-algorithm-test

Unit tests: Signer, SignatureAlgorithm

src/test/kotlin/com/nftco/flow/sdk/models/SignatureAlgorithmTest.kt

@@ -0,0 +1,31 @@
+package com.nftco.flow.sdk.models
+
+import com.nftco.flow.sdk.SignatureAlgorithm
+import org.junit.jupiter.api.Assertions.assertEquals
+import org.junit.jupiter.api.Test
+
+class SignatureAlgorithmTest {
+    @Test
+    fun `Test fromCode`() {
+        assertEquals(SignatureAlgorithm.ECDSA_P256, SignatureAlgorithm.fromCode(2))
+        assertEquals(SignatureAlgorithm.ECDSA_SECP256k1, SignatureAlgorithm.fromCode(3))
+        assertEquals(SignatureAlgorithm.UNKNOWN, SignatureAlgorithm.fromCode(-1))
+    }
+
+    @Test
+    fun `Test fromCadenceIndex`() {
+        assertEquals(SignatureAlgorithm.ECDSA_P256, SignatureAlgorithm.fromCadenceIndex(1))
+        assertEquals(SignatureAlgorithm.ECDSA_SECP256k1, SignatureAlgorithm.fromCadenceIndex(2))
+        assertEquals(SignatureAlgorithm.UNKNOWN, SignatureAlgorithm.fromCadenceIndex(-1))
+    }
+
+    @Test
+    fun `Test fromCode with invalid code`() {
+        assertEquals(SignatureAlgorithm.UNKNOWN, SignatureAlgorithm.fromCode(0))
+    }
+
+    @Test
+    fun `Test fromCadenceIndex with invalid index`() {
+        assertEquals(SignatureAlgorithm.UNKNOWN, SignatureAlgorithm.fromCadenceIndex(4))
+    }
+}

src/test/kotlin/com/nftco/flow/sdk/models/SignerTest.kt

@@ -0,0 +1,55 @@
+package com.nftco.flow.sdk.models
+
+import com.nftco.flow.sdk.DomainTag
+import org.junit.jupiter.api.Assertions.assertArrayEquals
+import org.junit.jupiter.api.BeforeEach
+import org.junit.jupiter.api.Test
+import com.nftco.flow.sdk.Signer
+import com.nftco.flow.sdk.Hasher
+
+class SignerTest {
+    lateinit var hasher: Hasher
+
+    lateinit var signer: Signer
+
+    @BeforeEach
+    fun setUp() {
+        signer = object : Signer {
+            override val hasher: Hasher
+                get() = this@SignerTest.hasher
+
+            override fun sign(bytes: ByteArray): ByteArray {
+                // Mock sign function to return bytes * 2
+                return bytes + bytes
+            }
+        }
+    }
+
+    @Test
+    fun `Test signWithDomain`() {
+        val inputBytes = byteArrayOf(0x01, 0x02, 0x03)
+        val domainBytes = byteArrayOf(0x04, 0x05)
+        val expectedBytes = signer.sign(domainBytes + inputBytes)
+
+        val actualBytes = signer.signWithDomain(inputBytes, domainBytes)
+        assertArrayEquals(expectedBytes, actualBytes)
+    }
+
+    @Test
+    fun `Test signAsUser`() {
+        val inputBytes = byteArrayOf(0x01, 0x02, 0x03)
+        val expectedBytes = signer.sign(DomainTag.USER_DOMAIN_TAG + inputBytes)
+
+        val actualBytes = signer.signAsUser(inputBytes)
+        assertArrayEquals(expectedBytes, actualBytes)
+    }
+
+    @Test
+    fun `Test signAsTransaction`() {
+        val inputBytes = byteArrayOf(0x01, 0x02, 0x03)
+        val expectedBytes = signer.sign(DomainTag.TRANSACTION_DOMAIN_TAG + inputBytes)
+
+        val actualBytes = signer.signAsTransaction(inputBytes)
+        assertArrayEquals(expectedBytes, actualBytes)
+    }
+}