test/regression_test.rb in urlcrypt-0.1.2 vs test/regression_test.rb in urlcrypt-0.2.0

@@ -1,36 +1,31 @@
 # encoding: utf-8
 class URLcryptRegressionTest < TestClass
+  def setup
+    # this key was generated via rake secret in a rails app, the pack() converts it into a byte array
+    @secret = ['d25883a27b9a639da85ea7e159b661218799c9efa63069fac13a6778c954fb6d721968887a19bdb01af8f59eb5a90d256bd9903355c20b0b4b39bf4048b9b17b'].pack('H*')
+  end
+
   def test_encryption_and_decryption
     original = '{"some":"json_data","token":"dfsfsdfsdf"}'
-    encrypted = URLcrypt.encrypt(original)
+    encrypted = URLcrypt.encrypt(original, key: @secret)
 
-    encrypted = URLcrypt::encrypt(original)
-    assert_equal(URLcrypt::decrypt(encrypted), original)
+    assert_equal(URLcrypt::decrypt(encrypted, key: @secret), original)
   end
 
   def test_encryption_with_too_long_key
-    # this key was generated via rake secret in a rails app, the pack() converts it into a byte array
-    secret = ['d25883a27b9a639da85ea7e159b661218799c9efa63069fac13a6778c954fb6d721968887a19bdb01af8f59eb5a90d256bd9903355c20b0b4b39bf4048b9b17b'].pack('H*')
-    URLcrypt::key = secret
+    assert OpenSSL::Cipher.new('aes-256-cbc').key_len < @secret.bytesize
 
-    assert OpenSSL::Cipher.new('aes-256-cbc').key_len < secret.bytesize
-
     original  = "hello world!"
-    encrypted = URLcrypt::encrypt(original)
-    assert_equal(URLcrypt::decrypt(encrypted), original)
+    encrypted = URLcrypt::encrypt(original, key: @secret)
+    assert_equal(URLcrypt::decrypt(encrypted, key: @secret), original)
   end
 
   def test_encryption_and_decryption_with_too_long_key
-    # this key was generated via rake secret in a rails app, the pack() converts it into a byte array
-    secret = ['d25883a27b9a639da85ea7e159b661218799c9efa63069fac13a6778c954fb6d721968887a19bdb01af8f59eb5a90d256bd9903355c20b0b4b39bf4048b9b17b'].pack('H*')
-    URLcrypt::key = secret
+    assert OpenSSL::Cipher.new('aes-256-cbc').key_len < @secret.bytesize
 
-    assert OpenSSL::Cipher.new('aes-256-cbc').key_len < secret.bytesize
-
     original = '{"some":"json_data","token":"dfsfsdfsdf"}'
-    encrypted = URLcrypt.encrypt(original)
+    encrypted = URLcrypt.encrypt(original, key: @secret)
 
-    encrypted = URLcrypt::encrypt(original)
-    assert_equal(URLcrypt::decrypt(encrypted), original)
+    assert_equal(URLcrypt::decrypt(encrypted, key: @secret), original)
   end
 end
\ No newline at end of file