ucert 0.2.57

data/test/test_t24_tracker.rb

@@ -0,0 +1,151 @@
+require "minitest/autorun"
+require "ucert"
+
+class TestT24Tracker < Minitest::Test
+
+  def setup
+    @t24_tracker = Ucert::T24Tracker.new(:verbose=>false)
+  end
+
+  def test_instance_initialize_success
+     refute_nil @t24_tracker.t24_user_entitlement_report
+     refute_nil @t24_tracker.t24_group_entitlement_report
+     refute_nil @t24_tracker.file_user_map
+     refute_nil @t24_tracker.t24_user_entitlement
+     refute_nil @t24_tracker.t24_group_entitlement
+     refute_nil @t24_tracker.t24_2_ad_user
+  end
+
+  def test_parse_t24_user_report
+    @test_t24_entitlement=@t24_tracker.send(:parse_t24_user_report, File.dirname(__FILE__)+"/t24_testfiles/T24_User_Rpt_Test.csv")
+    titles=File.open(File.dirname(__FILE__)+"/t24_testfiles/T24_User_Rpt_Test.csv", 'r').first.split(",").map {|x| x.gsub("\"","").gsub(/\n/, "")}
+    user_group=Hash.new
+    for i in 1..3
+      user_group[i]=Hash.new
+      j=1
+      titles.each do |title|
+        title.tr!('@','')
+          if title=="APPLICATION"
+            q=j+1
+            user_group[i][title]={"TEST#{i}-#{j}"=>"TEST#{i}-#{q}"}
+            j+=1
+          elsif title=="FUNCTION"
+            j+=1
+          elsif title=="TIME.LAST.SIGN.ON"
+          else
+            user_group[i][title]="TEST#{i}-#{j}"
+            j+=1
+          end
+      end
+      k=1
+      flag=false
+      while k<i do
+        key=user_group[i]["APPLICATION"].keys[0]+"-"+k.to_s
+        user_group[i]["APPLICATION"][key]=key
+        k+=1
+        flag=true
+      end
+      if flag
+        user_group[i]["GROUP"]=["@TEST#{i}-3"]
+        user_group[i]["APPLICATION"].delete(user_group[i]["APPLICATION"].first[0])
+      end
+      user_group[i].delete(user_group[i].first[0])
+      user_group[i]["ID"]="TEST#{i}-1"
+    end
+    assert_equal user_group, @test_t24_entitlement
+  end
+
+  def test_parse_t24_group_report
+    @test_t24_entitlement=@t24_tracker.send(:parse_t24_group_report, File.dirname(__FILE__)+"/t24_testfiles/T24_Grp_Rpt_Test.csv")
+    titles=["ID","DESCRIPTION","APPLICATION"]
+    test_group=Hash.new
+    for i in 1..3
+      test_group[i]=Hash.new
+      j=1
+      titles.each do |title|
+        if title=="APPLICATION"
+          test_group[i]["APPLICATION"]=Hash.new
+          application_name="TEST#{i}-#{j}"
+          test_group[i]["APPLICATION"][application_name]=Hash.new
+          j+=1
+          test_group[i]["APPLICATION"][application_name]["VERSION"]="TEST#{i}-#{j}"
+          j+=1
+          test_group[i]["APPLICATION"][application_name]["FUNCTION"]="TEST#{i}-#{j}"
+        else
+          test_group[i][title]="TEST#{i}-#{j}"
+        end
+        j+=1
+      end
+      k=1
+      while k<i do
+        application_name=test_group[i]["APPLICATION"].keys[0]+"-"+k.to_s
+        test_group[i]["APPLICATION"][application_name]=Hash.new
+        test_group[i]["APPLICATION"][application_name]["VERSION"]=""
+        test_group[i]["APPLICATION"][application_name]["FUNCTION"]=application_name
+        k+=1
+      end
+    end
+    assert_equal test_group, @test_t24_entitlement
+    refute_nil @test_t24_entitlement
+  end
+
+  def test_insert_dn
+    @t24_tracker.t24_user_entitlement.each do |index, user|
+      dn_regex=/\ACN=.+\,OU=.+\,DC=ny,DC=cmbchina,DC=com\z/
+      assert_match dn_regex, user["DN"] unless user["DN"].nil?
+    end
+  end
+
+  def test_save_t24_user_map!
+    @t24_tracker.send(:save!, File.dirname(__FILE__)+"/t24_testfiles/t24_user_map_test.txt")
+    line_count=0
+    File.open(File.dirname(__FILE__)+"/t24_testfiles/t24_user_map_test.txt", 'r').each_line do |line|
+      if line_count==0
+        assert_match (/\# local T24 to AD user map file created by the .+ class .+ method at: .+/), line
+      else
+        user_id=line.split(":")[0]
+        user_name=line.split(":")[1].split("|")[0]
+        dn=line.split(":")[1].split("|")[1].nil? ? dn="" : dn=line.split(":")[1].split("|")[1].strip
+        assert_equal @t24_tracker.t24_user_entitlement[line_count]["USER.NAME"].upcase, user_name
+        assert_equal @t24_tracker.t24_user_entitlement[line_count]["ID"].upcase, user_id
+        if dn.empty?
+          assert_nil @t24_tracker.t24_user_entitlement[line_count]["DN"]
+        else
+          assert_equal @t24_tracker.t24_user_entitlement[line_count]["DN"], dn
+        end
+      end
+      line_count+=1
+    end
+    File.delete(File.dirname(__FILE__)+"/t24_testfiles/t24_user_map_test.txt")
+  end
+
+  def test_t24_search_by_dn
+    index=0
+    File.open(@t24_tracker.file_user_map, 'r').each_line do |line|
+      if index>0
+        line.split("|")[1].nil? ? dn="" : dn=line.split("|")[1].strip
+        assert_equal dn, @t24_tracker.send(:search_by_dn, dn)['DN'] unless dn.empty?
+      end
+      index=index+1
+    end
+  end
+
+  def test_print_user_entitlement
+    $stdout = StringIO.new
+    @t24_tracker.print_user.each
+    items=$stdout.string.split("\n")
+    items.each do |line|
+     assert_equal 6, line.count("|")
+    end
+  end
+
+  def test_print_group_entitlement
+    $stdout = StringIO.new
+    @t24_tracker.print_group.each
+    items=$stdout.string.split("\n")
+    items.each do |line|
+     assert_equal 2, line.count("|")
+    end
+  end
+
+end

data/test/test_utils.rb

@@ -0,0 +1,46 @@
+require "minitest/autorun"
+require "ucert"
+
+class TestClass
+  include Ucert::Utils
+end
+
+class TestUtils < Minitest::Test
+
+  def setup
+    @test_instance = TestClass.new
+  end
+
+  def test_file_2_list
+    test_array=["Test1","Test2","Test3","Test4","Test5"]
+    assert_equal test_array, @test_instance.file_2_list(File.dirname(__FILE__)+"/utils_testfiles/file2list_test.txt")
+  end
+
+  def test_search_ad
+    test_dn="CN=HOAuditor,OU=Disabled Users,OU=New York Branch,DC=ny,DC=cmbchina,DC=com"
+    assert_equal test_dn, @test_instance.search_ad("HOAuditor")
+  end
+
+  def test_load_known_user_map_from_file
+    test_hash={"1"=>"test1", "2"=>"test2", "3"=>"test3"}
+    assert_equal test_hash, @test_instance.load_known_user_map_from_file(File.dirname(__FILE__)+"/utils_testfiles/load_know_user_map_testfile.txt")
+  end
+
+# ip =~ /\d+\.\d+\.\d+.\d+/ return 0 in method for a valid ip
+  def test_is_ip?
+    test_ip=["1.1.1.1","192.1.10.1","00.00.0.1","999.0.1.3","11.00.1.","19216818","a.b.c.1","1.1.1.1.1"]
+    assert  @test_instance.is_ip?(test_ip[0])
+    assert  @test_instance.is_ip?(test_ip[1])
+    refute  @test_instance.is_ip?(test_ip[2])
+    refute  @test_instance.is_ip?(test_ip[3])
+    refute  @test_instance.is_ip?(test_ip[4])
+    refute  @test_instance.is_ip?(test_ip[5])
+    refute  @test_instance.is_ip?(test_ip[6])
+    refute  @test_instance.is_ip?(test_ip[7])
+  end
+
+  def test_nslookup
+    assert @test_instance.nslookup("www.ruby-lang.org")
+  end
+
+end

data/test/test_vpn_tracker.rb

@@ -0,0 +1,56 @@
+require "minitest/autorun"
+require "ucert"
+
+class TestVpnTracker < Minitest::Test
+
+  def setup
+    @vpn_tracker = Ucert::VpnTracker.new(:verbose=>false)
+  end
+
+  def test_instance_initialize_success
+     refute_nil @vpn_tracker.vpn_user_entitlement
+  end
+
+  #user CN=Sherry's department is "Gompliance" instead of "Compliance", the typo caused the test fail.
+  def test_parse_vpn_user_info
+=begin
+    department_catagory=["IT","FI","Banking Operations","Management","Compliance","Corporate Banking US Group","Corporate Banking China Group","TransactionBanking","Treasury","Risk Management","Credit Products Group","Finance","SPD","PB","Human Resource","CPG","General Office"]
+    @test_vpn_entitlement=@vpn_tracker.send(:parse_vpn_user_info)
+    cout=1
+    @test_vpn_entitlement.each do |index, people|
+      assert_equal true, people.has_key?("CN")
+      user_cn=people["CN"]
+      puts "checking index:"+cout.to_s+" CN:"+user_cn
+      puts people
+      people.each do |key, value|
+        if key=="department"
+          assert department_catagory.include? value
+        elsif key=="VPN Class"
+          assert_match (/\Aclass_\d\z/), value
+        elsif key=="DN"
+          assert_match (/\A#{user_cn}\,OU=Users,OU=.+\,DC=ny,DC=cmbchina,DC=com\z/), value
+        else
+          refute_nil value
+        end
+      end
+      cout=cout+1
+    end
+=end
+  end
+
+  def test_print_user_entitlement
+    column_num=Array.new
+    @vpn_tracker.vpn_user_entitlement.values.each do |rec|
+      column_num.push(rec.each.size)
+    end
+    assert_equal 1, column_num.uniq!.size
+    $stdout = StringIO.new
+    @vpn_tracker.print_user
+    items=$stdout.string.split("\n")
+    items.each do |line|
+     assert_match (/\A.+\|(.*\|){#{column_num[0]-1}}\z/), line
+     assert_equal column_num[0], line.count("|")
+    end
+  end
+
+end

data/test/test_wms_tracker.rb

@@ -0,0 +1,109 @@
+require "minitest/autorun"
+require "ucert"
+
+class TestWmsTracker < Minitest::Test
+
+    def setup
+      @wms_tracker = Ucert::WmsTracker.new(:verbose=>false)
+    end
+
+    def test_instance_initialize_success
+       refute_nil @wms_tracker.wms_user_report
+       refute_nil @wms_tracker.wms_role_report
+       refute_nil @wms_tracker.file_user_map
+       refute_nil @wms_tracker.wms_2_ad_user
+       refute_nil @wms_tracker.wms_user_entitlement
+       refute_nil @wms_tracker.wms_role_entitlement
+    end
+
+    def test_parse_wms_user_report_method
+      @test_user_entitlement=@wms_tracker.send(:parse_wms_user_report, File.dirname(__FILE__)+"/wms_testfiles/user_rpt_test.txt")
+      test_group=Hash.new
+      test_group[1]={"USERID"=>"test","USERNAME"=>"测试员","STATUS"=>"1","ORGNAME"=>"全行"}
+      test_group[2]={"USERID"=>"admin","USERNAME"=>"管理员","STATUS"=>"1","ORGNAME"=>"纽约分行信息技术部"}
+      test_group[3]={"USERID"=>"developer","USERNAME"=>"开发员","STATUS"=>"0","ORGNAME"=>"纽约分行风险管理部"}
+      test_group[1]["ROLEID"]={"0000"=>true}
+      test_group[2]["ROLEID"]={"0001"=>true,"0002"=>true,"0003"=>true}
+      test_group[3]["ROLEID"]={"0004"=>true}
+      for i in 1..3
+        assert_equal test_group[i], @test_user_entitlement[i]
+      end
+    end
+
+    def test_parse_wms_role_report_method
+      @test_role_entitlement=@wms_tracker.send(:parse_wms_role_report, File.dirname(__FILE__)+"/wms_testfiles/role_rpt_test.txt")
+      test_group=Hash.new
+      test_group["0000"]={"ROLENAME"=>"超级测试员"}
+      test_group["0001"]={"ROLENAME"=>"测试管理员"}
+      test_group["0002"]={"ROLENAME"=>"一般测试员"}
+      test_group["0000"]["PRV"]=[{"PRVID"=>"0000_SA1","PRVNAME"=>"Super Admin1","URL"=>"/ad1"},{"PRVID"=>"0000_SA2","PRVNAME"=>"Super Admin2","URL"=>"/ad2"}]
+      test_group["0001"]["PRV"]=[{"PRVID"=>"0000_SA2","PRVNAME"=>"Super Admin2","URL"=>"/ad2"},{"PRVID"=>"0001_TA1","PRVNAME"=>"Test Admin1","URL"=>"/ta1"}]
+      test_group["0002"]["PRV"]=[{"PRVID"=>"0002_TA2","PRVNAME"=>"Test Admin2","URL"=>"/ta2"}]
+      for i in ["0000","0001","0002"]
+        assert_equal test_group[i], @test_role_entitlement[i]
+      end
+    end
+
+    def test_insert_dn
+      @wms_tracker.wms_user_entitlement.each do |index, user|
+        dn_regex=/\ACN=.+\,OU=.+\,DC=ny,DC=cmbchina,DC=com\z/
+        assert_match dn_regex, user["DN"] unless user["DN"].nil?
+      end
+    end
+
+    def test_print_user_entitlement
+      $stdout = StringIO.new
+      @wms_tracker.print_user
+      items=$stdout.string.split("\n")
+      items.each do |line|
+        if line.include?("CN=")
+          assert_match (/\A.+\|.+\|(0|1)\|.+\|\{.+\}\|CN=[a-zA-Z\s]+,OU=[a-zA-Z\s]+,.*DC=ny,DC=cmbchina,DC=com\|\z/), line
+        else
+          assert_match (/\A.+\|.+\|(0|1)\|.+\|\{.+\}\|\|\z/), line
+        end
+      end
+    end
+
+    def test_print_user_role
+      $stdout = StringIO.new
+      @wms_tracker.print_role
+      items=$stdout.string.split("\n")
+      items.each do |line|
+        assert_match (/\A.+\|.+\|\[\{(\".+\"\=\>.+\,)*\".+\"\=\>.+\}\]\z/), line
+      end
+    end
+
+    def test_save_wms_user_map!
+      @wms_tracker.send(:save!, File.dirname(__FILE__)+"/wms_testfiles/wms_user_map_test.txt")
+      line_count=0
+      File.open(File.dirname(__FILE__)+"/wms_testfiles/wms_user_map_test.txt", 'r').each_line do |line|
+        if line_count==0
+          assert_match (/\# local Wms to AD user map file created by the .+ class .+ method at: .+/), line
+        else
+          user_name=line.split(":")[0]
+          user_id=line.split(":")[1].split("|")[0]
+          dn=line.split(":")[1].split("|")[1].nil? ? dn="" : dn=line.split(":")[1].split("|")[1].strip
+          assert_equal @wms_tracker.wms_user_entitlement[line_count]["USERNAME"].upcase, user_name
+          assert_equal @wms_tracker.wms_user_entitlement[line_count]["USERID"].upcase, user_id
+          if dn.empty?
+            assert_nil @wms_tracker.wms_user_entitlement[line_count]["DN"]
+          else
+            assert_equal @wms_tracker.wms_user_entitlement[line_count]["DN"], dn
+          end
+        end
+        line_count+=1
+      end
+      File.delete(File.dirname(__FILE__)+"/wms_testfiles/wms_user_map_test.txt")
+    end
+
+    def test_wms_search_by_dn
+      index=0
+      File.open(@wms_tracker.file_user_map, 'r').each_line do |line|
+        if index>0
+          line.split("|")[1].nil? ? dn="" : dn=line.split("|")[1].strip
+          assert_equal dn, @wms_tracker.send(:search_by_dn, dn)['DN'] unless dn.empty?
+        end
+        index=index+1
+      end
+    end
+end

data/test/test_yst_tracker.rb

@@ -0,0 +1,133 @@
+require "minitest/autorun"
+require "ucert"
+
+class TestYstTracker < Minitest::Test
+
+  def setup
+    @yst_tracker = Ucert::YstTracker.new(:verbose=>false)
+  end
+
+  def test_instance_initialize_success
+     refute_nil @yst_tracker.yst_user_report
+     refute_nil @yst_tracker.yst_org_report
+     refute_nil @yst_tracker.file_user_map
+     refute_nil @yst_tracker.yst_2_ad_user
+     refute_nil @yst_tracker.yst_user_entitlement
+     refute_nil @yst_tracker.yst_org_entitlement
+  end
+
+  def test_parse_yst_user_report_method
+    @test_user_entitlement=@yst_tracker.send(:parse_yst_user_report, File.dirname(__FILE__)+"/yst_testfiles/YiShiTong_User_Test.csv")
+    line_cnt=0
+    File.open(File.dirname(__FILE__)+"/yst_testfiles/YiShiTong_User_Test.csv",'r:gb18030:UTF-8').each do |line|
+      if line_cnt==0
+        @title=line.chomp.split(",")
+      else
+        user=line.chomp.split(",").map {|x| x.gsub(/(\s|\")/,"")}
+        assert_equal @title.size, @test_user_entitlement.values[line_cnt-1].size
+        assert_equal user, @test_user_entitlement.values[line_cnt-1].values
+      end
+      line_cnt+=1
+    end
+    assert_equal line_cnt-1, @test_user_entitlement.size
+  end
+
+  def test_parse_yst_org_report_method
+    @test_org_entitlement=@yst_tracker.send(:parse_yst_org_report,File.dirname(__FILE__)+"/yst_testfiles/YiShiTong_Org_Test.csv")
+    line_cnt=0
+    f=File.open( File.dirname(__FILE__)+"/yst_testfiles/YiShiTong_Org_Test.csv",'r:gb18030:UTF-8')
+    f.each do |line|
+      if line_cnt==0
+        @title=line.chomp.split(",")
+      else
+        org=line.chomp.split(",").map {|x| x.gsub(/(\s|\")/,"")}
+        assert_equal @title.size, @test_org_entitlement.values[line_cnt-1].size
+        assert_equal org, @test_org_entitlement.values[line_cnt-1].values
+      end
+      line_cnt+=1
+    end
+    assert_equal line_cnt-1, @test_org_entitlement.size
+  end
+
+    def test_insert_dn
+      @yst_tracker.yst_user_entitlement.each do |index, user|
+        dn_regex=/\ACN=.+\,OU=.+\,DC=ny,DC=cmbchina,DC=com\z/
+        assert_match dn_regex, user["DN"] unless user["DN"].nil?
+      end
+    end
+
+    def test_save_yst_user_map!
+      @yst_tracker.send(:save!, File.dirname(__FILE__)+"/yst_testfiles/yst_user_map_test.txt")
+      line_count=0
+      File.open(File.dirname(__FILE__)+"/yst_testfiles/yst_user_map_test.txt", 'r').each_line do |line|
+        if line_count==0
+          assert_match (/\# local Yst to AD user map file created by the .+ class .+ method at: .+/), line
+        else
+          user_name=line.split(":")[0]
+          email=line.split(":")[1].split("|")[0]
+          dn=line.split(":")[1].split("|")[1].nil? ? dn="" : dn=line.split(":")[1].split("|")[1].strip
+          assert_equal @yst_tracker.yst_user_entitlement[line_count]["USER_NAME"].upcase, user_name
+          assert_equal @yst_tracker.yst_user_entitlement[line_count]["EMAIL"].upcase, email
+          if dn.empty?
+            assert_nil @yst_tracker.yst_user_entitlement[line_count]["DN"]
+          else
+            assert_equal @yst_tracker.yst_user_entitlement[line_count]["DN"], dn
+          end
+        end
+        line_count+=1
+      end
+      File.delete(File.dirname(__FILE__)+"/yst_testfiles/yst_user_map_test.txt")
+    end
+
+    def test_yst_search_by_dn
+      index=0
+      File.open(@yst_tracker.file_user_map, 'r').each_line do |line|
+        if index>0
+          line.split("|")[1].nil? ? dn="" : dn=line.split("|")[1].strip
+          assert_equal dn, @yst_tracker.send(:search_by_dn, dn)['DN'] unless dn.empty?
+        end
+        index=index+1
+      end
+    end
+
+    def test_yst_search_by_id
+      index=0
+      File.open(@yst_tracker.yst_user_report, 'r:gb18030:UTF-8').each_line do |line|
+        if index>0
+          id=line.split(",")[0].gsub('"', '')
+          assert_equal id, @yst_tracker.send(:search_by_id, id)['USER_ID']
+        end
+        index=index+1
+      end
+    end
+
+  def test_yst_user_entitlement
+    column_num=Array.new
+    @yst_tracker.yst_user_entitlement.values.map do |rec|
+      column_num.push(rec.each.size)
+    end
+    assert_equal 1, column_num.uniq!.size
+    $stdout = StringIO.new
+    @yst_tracker.print_user.each
+    items=$stdout.string.split("\n")
+    items.each do |line|
+      assert_match (/\A.+\|(.*\|){#{column_num[0]-1}}\z/), line
+      assert_equal column_num[0], line.count("|")
+    end
+  end
+
+  def test_print_org_entitlement
+    column_num=Array.new
+    @yst_tracker.yst_org_entitlement.values.map do |rec|
+      column_num.push(rec.each.size)
+    end
+    assert_equal 1, column_num.uniq!.size
+    $stdout = StringIO.new
+    @yst_tracker.print_org.each
+    items=$stdout.string.split("\n")
+    items.each do |line|
+     assert_match (/\A.+\|(.*\|){#{column_num[0]-1}}\z/), line
+     assert_equal column_num[0], line.count("|")
+    end
+  end
+end