test refactor

Author: NegaScout

test/padding.jl

@@ -19,17 +19,19 @@ end
     test_vector = Vector{UInt8}([1,2,3])
     Random.seed!(42)
     padded = ToyPublicKeys.pad(ToyPublicKeys.pkcs1_v1_5, test_vector)
-    @test ToyPublicKeys.unpad(ToyPublicKeys.pkcs1_v1_5, padded) == test_vector
+    unpadded = ToyPublicKeys.unpad(ToyPublicKeys.pkcs1_v1_5, padded)
+    @test unpadded == test_vector
 end
 
 @testset "padding/pkcs_1_v2_2 pad(unpad) is identity" begin
-    test_vector = Vector{UInt8}([1,2,3])
+    test_vector = Vector{UInt8}([3,2,1])
     Random.seed!(42)
     private_key, public_key = ToyPublicKeys.generate_rsa_key_pair(ToyPublicKeys.pkcs1_v1_5, 2048)
     padded = ToyPublicKeys.pad(ToyPublicKeys.pkcs1_v2_2,
                                test_vector,
                                public_key)
-    @test test_vector == ToyPublicKeys.unpad(ToyPublicKeys.pkcs1_v2_2,
-                                             padded,
-                                             public_key)
+    unpadded = ToyPublicKeys.unpad(ToyPublicKeys.pkcs1_v2_2,
+                                    padded,
+                                    public_key)
+    @test unpadded == test_vector
 end

test/rsa.jl

@@ -117,3 +117,12 @@ end
     signature = ToyPublicKeys.RSASP1(ToyPublicKeys.pkcs1_v1_5, msg, private_key)
     @test ToyPublicKeys.RSAVP1(ToyPublicKeys.pkcs1_v1_5, signature, public_key) == msg
 end
+
+@testset "rsaes_oaep_decrypt(rsaes_oaep_encrypt) is true" begin
+    Random.seed!(42)
+    private_key, public_key = ToyPublicKeys.generate_rsa_key_pair(ToyPublicKeys.pkcs1_v1_5, 2048)
+    msg = Vector{UInt8}("123")
+    C = ToyPublicKeys.rsaes_oaep_encrypt(msg, public_key)
+    ret = ToyPublicKeys.rsaes_oaep_decrypt(C, private_key)
+    @test ret == msg
+end

test/runtests.jl

@@ -2,7 +2,7 @@ using ToyPublicKeys
 using Test
 import Random
 
+include("utils.jl")
+include("padding.jl")
 include("rsa.jl")
 include("dh.jl")
-include("padding.jl")
-include("utils.jl")

test/utils.jl

@@ -1,15 +1,15 @@
-@testset "I2OSP" begin
+@testset "i2osp" begin
     b = big"255"
-    @test b |> ToyPublicKeys.I2OSP == "FF"
-    @test b |> x -> ToyPublicKeys.I2OSP(x, 3) == "00:FF"
+    @test b |> ToyPublicKeys.i2osp == "FF"
+    @test b |> x -> ToyPublicKeys.i2osp(x, 3) == "00:FF"
 end
 
-@testset "OS2IP" begin
+@testset "os2ip" begin
     b = "FF"
-    @test b |> ToyPublicKeys.OS2IP == big"255"
+    @test b |> ToyPublicKeys.os2ip == big"255"
 end
 
-@testset "I2OSP |> OS2IP" begin
+@testset "i2osp |> os2ip" begin
     b = big"255"
-    @test b |> ToyPublicKeys.I2OSP |> ToyPublicKeys.OS2IP == b
+    @test b |> ToyPublicKeys.i2osp |> ToyPublicKeys.os2ip == b
 end