test/symmetric_encryption_test.rb in symmetric-encryption-0.9.1 vs test/symmetric_encryption_test.rb in symmetric-encryption-1.0.0

@@ -10,75 +10,65 @@
 SymmetricEncryption.load!(File.join(File.dirname(__FILE__), 'config', 'symmetric-encryption.yml'), 'test')
 
 # Unit Test for SymmetricEncryption
 #
 class SymmetricEncryptionTest < Test::Unit::TestCase
-  context 'initialized' do
+  context 'SymmetricEncryption' do
 
-    context 'SymmetricEncryption configuration' do
+    context 'configuration' do
       setup do
         @config = SymmetricEncryption.send(:read_config, File.join(File.dirname(__FILE__), 'config', 'symmetric-encryption.yml'), 'test')
       end
 
       should "match config file" do
         assert_equal @config[:ciphers][0][:cipher], SymmetricEncryption.cipher.cipher
       end
     end
 
-    context 'Base64 encoding tests' do
-      setup do
-        @social_security_number = "987654321"
-        @social_security_number_encrypted = "S+8X1NRrqdfEIQyFHVPuVA==\n"
-        @social_security_number_encrypted_with_secondary_1 = "D1UCu38pqJ3jc0GvwJHiow==\n"
-        @encoding = SymmetricEncryption.cipher.encoding
-        SymmetricEncryption.cipher.encoding = :base64
-      end
+    SymmetricEncryption::Cipher::ENCODINGS.each do |encoding|
+      context "encoding: #{encoding}" do
+        setup do
+          @social_security_number = "987654321"
+          @social_security_number_encrypted =
+            case encoding
+          when :base64
+            "S+8X1NRrqdfEIQyFHVPuVA==\n"
+          when :base64strict
+            "S+8X1NRrqdfEIQyFHVPuVA=="
+          when :base16
+            "4bef17d4d46ba9d7c4210c851d53ee54"
+          when :none
+            "K\xEF\x17\xD4\xD4k\xA9\xD7\xC4!\f\x85\x1DS\xEET".force_encoding(Encoding.find("binary"))
+          else
+            raise "Add test for encoding: #{encoding}"
+          end
+          @social_security_number_encrypted_with_secondary_1 = "D1UCu38pqJ3jc0GvwJHiow==\n"
+          @encoding = SymmetricEncryption.cipher.encoding
+          SymmetricEncryption.cipher.encoding = encoding
+        end
 
-      teardown do
-        SymmetricEncryption.cipher.encoding = @encoding
-      end
+        teardown do
+          SymmetricEncryption.cipher.encoding = @encoding
+        end
 
-      should "encrypt simple string" do
-        assert_equal @social_security_number_encrypted, SymmetricEncryption.encrypt(@social_security_number)
-      end
+        should "encrypt simple string" do
+          assert_equal @social_security_number_encrypted, SymmetricEncryption.encrypt(@social_security_number)
+        end
 
-      should "decrypt string" do
-        assert_equal @social_security_number, SymmetricEncryption.decrypt(@social_security_number_encrypted)
-      end
+        should "decrypt string" do
+          assert_equal @social_security_number, SymmetricEncryption.decrypt(@social_security_number_encrypted)
+        end
 
-      should "determine if string is encrypted" do
-        assert_equal true, SymmetricEncryption.encrypted?(@social_security_number_encrypted)
-        assert_equal false, SymmetricEncryption.encrypted?(@social_security_number)
-      end
+        should "determine if string is encrypted" do
+          assert_equal true, SymmetricEncryption.encrypted?(@social_security_number_encrypted)
+          assert_equal false, SymmetricEncryption.encrypted?(@social_security_number)
+        end
 
-      should "decrypt with secondary key when first one fails" do
-        assert_equal @social_security_number, SymmetricEncryption.decrypt(@social_security_number_encrypted)
+        should "decrypt with secondary key when first one fails" do
+          assert_equal @social_security_number, SymmetricEncryption.decrypt(@social_security_number_encrypted_with_secondary_1)
+        end
       end
     end
 
-    context 'Base64Strict tests' do
-      setup do
-        @social_security_number = "987654321"
-        @social_security_number_encrypted = "S+8X1NRrqdfEIQyFHVPuVA=="
-        @social_security_number_encrypted_with_secondary_1 = "D1UCu38pqJ3jc0GvwJHiow=="
-      end
-
-      should "encrypt simple string" do
-        assert_equal @social_security_number_encrypted, SymmetricEncryption.encrypt(@social_security_number)
-      end
-
-      should "decrypt string" do
-        assert_equal @social_security_number, SymmetricEncryption.decrypt(@social_security_number_encrypted)
-      end
-
-      should "determine if string is encrypted" do
-        assert_equal true, SymmetricEncryption.encrypted?(@social_security_number_encrypted)
-        assert_equal false, SymmetricEncryption.encrypted?(@social_security_number)
-      end
-
-      should "decrypt with secondary key when first one fails" do
-        assert_equal @social_security_number, SymmetricEncryption.decrypt(@social_security_number_encrypted)
-      end
-    end
   end
 
 end