Permalink
Browse files

Added a script to create new sanitization tests easily.

  • Loading branch information...
1 parent b02ee40 commit 450323c5dba9d054cf9aa15e9380d5be405377d2 @kenpratt committed Apr 19, 2010
Showing with 22 additions and 0 deletions.
  1. +22 −0 script/add_sanitization_test
@@ -0,0 +1,22 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH.unshift File.join(File.dirname(__FILE__), "..", "lib")
+require "wikipedia"
+
+SANITIZATION_SAMPLE_PATH = File.join(File.dirname(__FILE__), "..", "spec", "fixtures", "sanitization_samples")
+
+def add_sanitization_test(url)
+ title = Wikipedia::Url.new(url).title
+ page = Wikipedia.find(title, :rvsection => 0)
+ File.open(File.join(SANITIZATION_SAMPLE_PATH, "#{title}-raw.txt"), 'w') {|f| f << page.content }
+ File.open(File.join(SANITIZATION_SAMPLE_PATH, "#{title}-sanitized.txt"), 'w') {|f| f << page.sanitized_content }
+end
+
+if __FILE__ == $0
+ unless ARGV.size == 1
+ puts "Usage: #{$0} http://en.wikipedia.org/wiki/Social_Democratic_Party_of_Kyrgyzstan"
+ exit 0
+ end
+ url = ARGV[0]
+ add_sanitization_test(url)
+end

0 comments on commit 450323c

Please sign in to comment.