updated `leap test init` to work with openvpn service levels.

1 files changed, 17 insertions, 7 deletions

diff --git a/lib/leap_cli/commands/test.rb b/lib/leap_cli/commands/test.rb
index 1da4f6d..3f0feb0 100644
--- a/lib/leap_cli/commands/test.rb
+++ b/lib/leap_cli/commands/test.rb
@@ -5,8 +5,7 @@ module LeapCli; module Commands
     test.desc 'Creates files needed to run tests.'
     test.command :init do |init|
       init.action do |global_options,options,args|
-        generate_test_client_cert
-        generate_test_client_openvpn_config
+        generate_test_client_openvpn_configs
       end
     end
 
@@ -22,14 +21,25 @@ module LeapCli; module Commands
 
   private
 
-  def generate_test_client_openvpn_config
+  #
+  # generates a whole bunch of openvpn configs that can be used to connect to different openvpn gateways
+  #
+  def generate_test_client_openvpn_configs
+    assert_config! 'provider.ca.client_certificates.unlimited_prefix'
+    assert_config! 'provider.ca.client_certificates.limited_prefix'
     template = read_file! Path.find_file(:test_client_openvpn_template)
-
     ['production', 'testing', 'local', 'development'].each do |env|
-      vpn_nodes = manager.nodes[:environment => env][:services => 'openvpn']
+      vpn_nodes = manager.nodes[:environment => env][:services => 'openvpn']['openvpn.allow_limited' => true]
+      if vpn_nodes.any?
+        generate_test_client_cert(provider.ca.client_certificates.limited_prefix) do |key, cert|
+          write_file! [:test_openvpn_config, env+'_limited'], Util.erb_eval(template, binding)
+        end
+      end
+      vpn_nodes = manager.nodes[:environment => env][:services => 'openvpn']['openvpn.allow_unlimited' => true]
       if vpn_nodes.any?
-        config = Util.erb_eval(template, binding)
-        write_file! [:test_openvpn_config, env], config
+        generate_test_client_cert(provider.ca.client_certificates.unlimited_prefix) do |key, cert|
+          write_file! [:test_openvpn_config, env+'_unlimited'], Util.erb_eval(template, binding)
+        end
       end
     end
   end