added PolarSSL::SSL#set_endpoint and assert_equal on PolarSSL::SSL constant tests

luisbebop · luisbebop · commit cd34f281ba8a · 2013-11-09T18:59:49.000-02:00
diff --git a/src/polarssl.c b/src/polarssl.c
@@ -133,6 +133,16 @@ static mrb_value mrb_ssl_initialize(mrb_state *mrb, mrb_value self) {
   return self;
 }
 
+static mrb_value mrb_ssl_set_endpoint(mrb_state *mrb, mrb_value self) {
+	ssl_context *ssl;
+	mrb_int endpoint_mode;
+	
+	mrb_get_args(mrb, "i", &endpoint_mode);	
+	ssl = DATA_CHECK_GET_PTR(mrb, self, &mrb_ssl_type, ssl_context);
+	ssl_set_endpoint(ssl, endpoint_mode);
+	return mrb_true_value();
+}
+
 void mrb_mruby_polarssl_gem_init(mrb_state *mrb) {
 	struct RClass *p, *e, *c, *s;
 	
@@ -159,6 +169,7 @@ void mrb_mruby_polarssl_gem_init(mrb_state *mrb) {
 	mrb_define_const(mrb, s, "SSL_VERIFY_OPTIONAL", mrb_fixnum_value(SSL_VERIFY_OPTIONAL));
 	// 2: Certificate verification mode for having required verification.
 	mrb_define_const(mrb, s, "SSL_VERIFY_REQUIRED", mrb_fixnum_value(SSL_VERIFY_REQUIRED));
+	mrb_define_method(mrb, s, "set_endpoint", mrb_ssl_set_endpoint, MRB_ARGS_REQ(1));
 }
 
 void mrb_mruby_polarssl_gem_final(mrb_state *mrb) {	
diff --git a/test/polarssl.rb b/test/polarssl.rb
@@ -61,18 +61,27 @@
   
   assert('PolarSSL::SSL::SSL_IS_CLIENT') do
     PolarSSL::SSL.const_defined? :SSL_IS_CLIENT
+    assert_equal(PolarSSL::SSL::SSL_IS_CLIENT, 0)
   end
   
   assert('PolarSSL::SSL::SSL_VERIFY_NONE') do
     PolarSSL::SSL.const_defined? :SSL_VERIFY_NONE
+    assert_equal(PolarSSL::SSL::SSL_VERIFY_NONE, 0)
   end
   
   assert('PolarSSL::SSL::SSL_VERIFY_OPTIONAL') do
     PolarSSL::SSL.const_defined? :SSL_VERIFY_OPTIONAL
+    assert_equal(PolarSSL::SSL::SSL_VERIFY_OPTIONAL, 1)
   end
   
   assert('PolarSSL::SSL::SSL_VERIFY_REQUIRED') do
     PolarSSL::SSL.const_defined? :SSL_VERIFY_REQUIRED
+    assert_equal(PolarSSL::SSL::SSL_VERIFY_REQUIRED, 2)
   end
   
+  assert('PolarSSL::SSL#set_endpoint') do
+    ssl = PolarSSL::SSL.new
+    ssl.set_endpoint(PolarSSL::SSL::SSL_IS_CLIENT)
+  end
+    
 end
