summaryrefslogtreecommitdiff
path: root/certs/test/unit
diff options
context:
space:
mode:
Diffstat (limited to 'certs/test/unit')
-rw-r--r--certs/test/unit/cert_pool_test.rb51
-rw-r--r--certs/test/unit/cert_test.rb48
2 files changed, 99 insertions, 0 deletions
diff --git a/certs/test/unit/cert_pool_test.rb b/certs/test/unit/cert_pool_test.rb
new file mode 100644
index 0000000..24ace57
--- /dev/null
+++ b/certs/test/unit/cert_pool_test.rb
@@ -0,0 +1,51 @@
+require 'test_helper'
+
+class CertPoolTest < ActiveSupport::TestCase
+
+ setup do
+ 2.times { Cert.create! }
+ end
+
+ teardown do
+ Cert.all.each {|c| c.destroy}
+ end
+
+ test "picks random sample" do
+ Cert.create! # with 3 certs chances are pretty low we pick the same one 40 times.
+ picked = []
+ first = Cert.sample.id
+ current = Cert.sample.id
+ 40.times do
+ break if current != first
+ current = Cert.sample.id
+ end
+ assert_not_equal current, first
+ end
+
+ test "picks cert from the pool" do
+ assert_difference "Cert.count", -1 do
+ cert = Cert.pick_from_pool
+ end
+ end
+
+ test "err's out if all certs have been destroyed" do
+ sample = Cert.first.tap{|c| c.destroy}
+ Cert.all.each {|c| c.destroy}
+ assert_raises RECORD_NOT_FOUND do
+ Cert.expects(:sample).returns(sample)
+ cert = Cert.pick_from_pool
+ end
+ end
+
+ test "picks other cert if first pick has been destroyed" do
+ first = Cert.first.tap{|c| c.destroy}
+ second = Cert.first
+ Cert.expects(:sample).at_least_once.
+ returns(first).
+ then.returns(second)
+ cert = Cert.pick_from_pool
+ assert_equal second, cert
+ assert_nil Cert.first
+ end
+
+end
diff --git a/certs/test/unit/cert_test.rb b/certs/test/unit/cert_test.rb
new file mode 100644
index 0000000..21ef169
--- /dev/null
+++ b/certs/test/unit/cert_test.rb
@@ -0,0 +1,48 @@
+require 'test_helper'
+
+class CertTest < ActiveSupport::TestCase
+
+ setup do
+ @sample = Cert.new
+ @sample.set_random
+ @sample.attach_zip
+ end
+
+ test "certs come with attachments" do
+ assert @sample.has_attachment? "cert.txt"
+ end
+
+ test "cert.zip_attachment returns couchDB attachment" do
+ assert_equal "text/plain", @sample.zip_attachment["content_type"]
+ end
+
+ test "cert.zipped returns the actual data" do
+ @sample.save # This is required!
+ assert lines = @sample.zipped.split("\n")
+ assert_equal "-----BEGIN RSA PRIVATE KEY-----", lines.first.chomp
+ assert_equal "-----END CERTIFICATE-----", lines.last.chomp
+ end
+
+ test "cert.zipname returns name for the zip file" do
+ assert_equal "cert.txt", @sample.zipname
+ end
+
+ test "test data is valid" do
+ assert @sample.valid?
+ end
+
+ test "validates random" do
+ @sample.stubs(:set_random)
+ [0, 1, nil, "asdf"].each do |invalid|
+ @sample.random = invalid
+ assert !@sample.valid?, "#{invalid} should not be a valid value for random"
+ end
+ end
+
+ test "validates attachment" do
+ @sample.stubs(:attach_zip)
+ @sample.delete_attachment(@sample.zipname)
+ assert !@sample.valid?, "Cert should require zipped attachment"
+ end
+
+end