Added Checksum::AlgorythmHelper#sum_of_figues_for_pt_si

anshulacro · Aug 1, 2013 · e1572dd · e1572dd
1 parent b8d0168
commit e1572dd
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 6 deletions.
diff --git a/lib/valvat/checksum.rb b/lib/valvat/checksum.rb
@@ -64,6 +64,12 @@ def sum_of_figures(reverse_ints=false)
         (fig*(i.modulo(2) == 0 ? ints[0] : ints[1])).to_s.split("").inject(0) { |sum, n| sum + n.to_i }
       end.inject(:+)
     end
+
+    def sum_of_figues_for_pt_si
+      11 - figures.reverse.each_with_index.map do |fig, i|
+        fig*(i+2)
+      end.inject(:+).modulo(11)
+    end
   end
 end
 

diff --git a/lib/valvat/checksum/pt.rb b/lib/valvat/checksum/pt.rb
@@ -3,10 +3,10 @@
 class Valvat
   module Checksum
     class PT < Base
+      include AlgorythmHelper
+
       def check_digit
-        chk = 11 - figures.reverse.each_with_index.map do |fig, i|
-          fig*(i+2)
-        end.inject(:+).modulo(11)
+        chk = sum_of_figues_for_pt_si
         chk > 9 ? 0 : chk
       end
     end

diff --git a/lib/valvat/checksum/si.rb b/lib/valvat/checksum/si.rb
@@ -3,15 +3,15 @@
 class Valvat
   module Checksum
     class SI < Base
+      include AlgorythmHelper
+
       def validate
         figures_str.to_i > 999999 &&
         super
       end
 
       def check_digit
-        chk = 11 - figures.reverse.each_with_index.map do |fig, i|
-          fig*(i+2)
-        end.inject(:+).modulo(11)
+        chk = sum_of_figues_for_pt_si
         chk == 1 ? 0 : chk
       end
     end