\n"
ret
end
def add_remove_column_htmlform
if $cgi.include?("add_remove_column_month")
begin
startdate = Date.parse("#{$cgi["add_remove_column_month"]}-1")
rescue ArgumentError
olddate = $cgi.params["add_remove_column_month"][1]
case $cgi["add_remove_column_month"]
when CGI.unescapeHTML(YEARBACK)
startdate = Date.parse("#{olddate}-1")-365
when CGI.unescapeHTML(MONTHBACK)
startdate = Date.parse("#{olddate}-1")-1
when CGI.unescapeHTML(MONTHFORWARD)
startdate = Date.parse("#{olddate}-1")+31
when CGI.unescapeHTML(YEARFORWARD)
startdate = Date.parse("#{olddate}-1")+366
else
exit
end
startdate = Date.parse("#{startdate.year}-#{startdate.month}-1")
end
else
startdate = Date.parse("#{Date.today.year}-#{Date.today.month}-1")
end
ret = <
END
ret
end
def add_remove_column name,description
begin
parsed_name = Date.parse("#{$cgi["add_remove_column_month"]}-#{name}")
rescue ArgumentError
return false
end
add_remove_parsed_column(parsed_name,CGI.escapeHTML(description))
end
end
if __FILE__ == $0
require 'test/unit'
require 'pp'
class DatePoll
def store comment
end
end
SITE="gbfuaibe"
require "cgi"
CGI_PARAMS={"add_remove_column_month" => ["2008-02"]}
CGI_COOKIES={}
$cgi = CGI.new
class DatePollTest < Test::Unit::TestCase
def setup
@poll = DatePoll.new(SITE, false)
end
def test_add_remove_column
assert(!@poll.add_remove_column("foo", "bar"))
assert(!@poll.add_remove_column("31", "31.02.2008 ;--)"))
assert(@poll.add_remove_column("20", "correct date"))
assert_equal("correct date",@poll.head[Date.parse("2008-02-20")])
assert(@poll.add_remove_column("20", "foobar"))
assert(@poll.head.empty?)
end
end
end