using Falcon.SugarApi.Encryption;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using System;
using System.Text;

namespace Falcon.SugarApi.Test.Encryption
{
    /// <summary>
    /// RSA测试类
    /// </summary>
    [TestClass]
    public class EncryptionTest
    {

        /// <summary>
        /// 测试公钥、秘钥生成，加密解密
        /// </summary>
        [TestMethod("DES测试")]
        public void DESTest() {
            var config = new DESConfig() { };
            IDESEncryption r = new DESProvider(config);
            var k = r.GenerateKey();

            //var bs = Encoding.UTF8.GetBytes(mingw);
            //var a1 = Convert.ToBase64String(bs);
            //var a2 = Convert.FromBase64String(a1);
            //Assert.AreEqual(bs.Length, a2.Length);

            for (int i = 0; i < 1000; i++) {
                var mingw = GenerateStr();
                Console.WriteLine(mingw);
                var p = new DESProvider(config);
                var miwen = p.Encrypt(k, mingw);

                var p1 = new DESProvider(config);
                var mingw1 = p1.Decrypt(k, miwen);

                Assert.AreEqual(mingw, mingw1, "解密后明文不同");

            }
        }
        /// <summary>
        /// 测试公钥、秘钥生成，加密解密
        /// </summary>
        [TestMethod("AES测试")]
        public void AESTest() {
            var config = new AESConfig() { };
            IAESEncryption r = new AESProvider(config);
            var k = r.GenerateKey();

            for (int i = 0; i < 1000; i++) {
                var mingw = GenerateStr();
                Console.WriteLine(mingw);
                var p = new AESProvider(config);
                var miwen = p.Encrypt(k, mingw);

                var p1 = new AESProvider(config);
                var mingw1 = p1.Decrypt(k, miwen);

                Assert.AreEqual(mingw, mingw1, "解密后明文不同");

            }
        }

        /// <summary>
        /// 生成随机长度，由字符表内字符组成的字符串
        /// </summary>
        /// <returns></returns>
        private string GenerateStr() {
            var chars = @"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ123456789,.!/*\";
            var r = new Random();
            var len = r.Next(100);
            var sb = new StringBuilder(len);
            for (int i = 0; i < len; i++) {
                sb.Append(chars[r.Next(0, chars.Length)]);
            }
            return sb.ToString();
        }
    }
}