test: fix and add client

sutor-aes/Client.cs

@@ -4,7 +4,7 @@
 
 namespace SutorAes
 {
-    class Client
+    public class Client
     {
         Guid Id { get; set; }
         public string Name { get; set; }
@@ -16,7 +16,7 @@
 
         RSAEncryptionPadding padding = RSAEncryptionPadding.Pkcs1;
 
         public Client(string _name)
         {
             this.Name = _name;
             GenerateRSAKeys();

sutor-aes/Program.cs

@@ -11,82 +11,5 @@ static void Main(string[] args)
         {
             Console.WriteLine("Hello World!");
         }
-
-        public static string DecryptString(byte[] cipherText, byte[] key, byte[] iv)
-        {
-            if (cipherText == null || cipherText.Length <= 0)
-            {
-                throw new ArgumentNullException(nameof(cipherText));
-            }
-            if (key == null || key.Length <= 0)
-            {
-                throw new ArgumentNullException(nameof(key));
-            }
-            if (iv == null || iv.Length <= 0)
-            {
-                throw new ArgumentNullException(nameof(iv));
-            }
-
-            string plaintext = null;
-
-            using (Aes aesAlg = Aes.Create())
-            {
-                aesAlg.Key = key;
-                aesAlg.IV = iv;
-
-                ICryptoTransform decryptor = aesAlg.CreateDecryptor(aesAlg.Key, aesAlg.IV);
-
-                using (MemoryStream msDecrypt = new MemoryStream(cipherText))
-                {
-                    using (CryptoStream csDecrypt = new CryptoStream(msDecrypt, decryptor, CryptoStreamMode.Read))
-                    {
-                        using (StreamReader srDecrypt = new StreamReader(csDecrypt))
-                        {
-                            plaintext = srDecrypt.ReadToEnd();
-                        }
-                    }
-                }
-            }
-
-            return plaintext;
-        }
-
-        public static byte[] EncryptString(string plainText, byte[] key, byte[] iv)
-        {
-            if (plainText == null)
-            {
-                throw new ArgumentNullException(nameof(plainText));
-            }
-            if (key == null || key.Length <= 0)
-            {
-                throw new ArgumentNullException(nameof(key));
-            }
-            if (iv == null || iv.Length <= 0)
-            {
-                throw new ArgumentNullException(nameof(iv));
-            }
-
-            using (Aes aesAlg = Aes.Create())
-            {
-                aesAlg.Key = key;
-                aesAlg.IV = iv;
-
-                ICryptoTransform encryptor = aesAlg.CreateEncryptor(aesAlg.Key, aesAlg.IV);
-
-                using (MemoryStream msEncrypt = new MemoryStream())
-                {
-                    using (CryptoStream csEncrypt = new CryptoStream(msEncrypt, encryptor, CryptoStreamMode.Write))
-                    {
-                        using (StreamWriter swEncrypt = new StreamWriter(csEncrypt))
-                        {
-                            swEncrypt.Write(plainText);
-                        }
-                    }
-                    // Return the encrypted bytes directly without Base64 encoding
-                    return msEncrypt.ToArray();
-                }
-            }
-        }
     }
-
 }

tests/ClientTests.cs

@@ -0,0 +1,39 @@
+using System;
+using System.Text;
+using NUnit.Framework;
+using SutorAes;
+
+namespace tests;
+
+public class ProgramTests
+{
+    private byte[] _key;
+    private byte[] _iv;
+
+    [SetUp]
+    public void Setup()
+    {
+        _key = new byte[32];
+        Random random = new Random();
+        random.NextBytes(_key);
+
+        _iv = new byte[16];
+        random.NextBytes(_iv);
+    }
+
+    [Test]
+    public void EstablishConnection_SendMessage()
+    {
+        string originalText = "This is a test message for encryption and decryption";
+
+        Client bob = new Client("Bob");
+        Client alice = new Client("Alice");
+
+        bob.EstablishConnection(alice);
+
+        bool recieved = false;
+        alice.OnMessage((string msg) => recieved = msg == originalText);
+        bob.SendMessage(alice, originalText);
+        Assert.That(recieved);
+    }
+}