From 0bf72af37383c2c3cfafe630d91028169ff503c6 Mon Sep 17 00:00:00 2001
From: Tim Kramer <tim@thethinkery.net>
Date: Mon, 23 Apr 2018 22:21:21 -0700
Subject: [PATCH] more tests

---
 lib/RESO_UPI_Tester_Ruby.rb       | 21 ++++++++++++++++++++-
 test/RESO_UPI_Tester_Ruby_test.rb | 13 +++++++++++++
 2 files changed, 33 insertions(+), 1 deletion(-)

diff --git a/lib/RESO_UPI_Tester_Ruby.rb b/lib/RESO_UPI_Tester_Ruby.rb
index 0900fbf..da78d0a 100644
--- a/lib/RESO_UPI_Tester_Ruby.rb
+++ b/lib/RESO_UPI_Tester_Ruby.rb
@@ -10,6 +10,7 @@ class UPI
     @country_code
     @sub_country_name
     @sub_country_code
+    @sub_county_name
     @sub_county_code
     @property_id
     @property_code
@@ -65,7 +66,12 @@ def get_sub_country_name
     end
 
     def set_sub_country_name(sub_country_name)
-    	@sub_country_name = sub_country_name
+      @valid_codes.each do |i|
+        if i[0] == sub_country_name
+          set_sub_country_code(i[1])
+          @sub_country_name = sub_country_name
+        end
+      end
     end
 
     def get_sub_country_code
@@ -84,6 +90,19 @@ def set_sub_county_code(sub_county_code)
     	@sub_county_code = sub_county_code
     end
 
+    def get_sub_county_name
+    	@sub_county_code
+    end
+
+    def set_sub_county_name(sub_county_name)
+      @valid_codes.each do |i|
+        if i[3] == sub_county_name and i[1] == get_sub_country_code
+          set_sub_county_code(i[2])
+          @sub_county_name = sub_county_name
+        end
+      end
+    end
+
     def get_property_id
     	@property_id
     end
diff --git a/test/RESO_UPI_Tester_Ruby_test.rb b/test/RESO_UPI_Tester_Ruby_test.rb
index 02ddca0..ed57d4c 100644
--- a/test/RESO_UPI_Tester_Ruby_test.rb
+++ b/test/RESO_UPI_Tester_Ruby_test.rb
@@ -34,4 +34,17 @@ def test_basic_functions
     	refute(t.is_valid, upi)
     end
   end
+
+  def test_state_code
+    t = RESOUPITesterRuby::UPI.new
+    t.set_sub_country_name('AL')
+    assert_equal(t.get_sub_country_code, "01")
+  end
+
+  def test_county_code
+    t = RESOUPITesterRuby::UPI.new
+    t.set_sub_country_name('AL')
+    t.set_sub_county_name('Autauga County')
+    assert_equal(t.get_sub_county_code, "001")
+  end
 end