git-contest 0.0.1

data/spec/bin/t009_git_contest_init_spec.rb

@@ -0,0 +1,82 @@
+require "spec_helper"
+
+# Don't forget --defaults option
+
+describe "T009: git-contest-init" do
+
+  before do
+    init_env
+    @test_dir = "#{ENV['GIT_CONTEST_TEMP_DIR']}/t009"
+    Dir.mkdir @test_dir
+    Dir.chdir @test_dir
+    # debug_on
+  end
+
+  after do
+    Dir.chdir '..'
+    Dir.rmdir @test_dir
+  end
+
+  describe "001: --force" do
+    before do
+      Dir.mkdir "001"
+      Dir.chdir "001"
+    end
+
+    after do
+      FileUtils.remove_dir ".git", :force => true
+      Dir.chdir ".."
+      Dir.rmdir "001"
+    end
+
+    it "001: init -> init" do
+      ret1 = bin_exec "init --defaults"
+      ret_config1 = git_do("config --get git.contest.branch.master")
+      ret2 = bin_exec "init --defaults"
+
+      ret_config1.should                                           === "master"
+      ret1.include?("Error: unknown argument").should              === false
+      ret2.include?("Error: unknown argument").should              === false
+      ret1.include?("Already initialized for git-contest.").should === false
+      ret1.include?("use: git contest init -f").should             === false
+      ret2.include?("Already initialized for git-contest.").should === true
+      ret2.include?("use: git contest init -f").should             === true
+    end
+
+    it "002: init -> init -f -> init --force" do
+      ret1 = bin_exec "init --defaults"
+      ret_config1 = git_do("config --get git.contest.branch.master")
+      ret2 = bin_exec "init --defaults -f"
+      ret3 = bin_exec "init --defaults --force"
+      ret_config1.should === "master"
+      ret1.include?("Error: unknown argument").should              === false
+      ret2.include?("Error: unknown argument").should              === false
+      ret3.include?("Error: unknown argument").should              === false
+      ret1.include?("Already initialized for git-contest.").should === false
+      ret1.include?("use: git contest init -f").should             === false
+      ret2.include?("Already initialized for git-contest.").should === false
+      ret2.include?("use: git contest init -f").should             === false
+      ret3.include?("Already initialized for git-contest.").should === false
+      ret3.include?("use: git contest init -f").should             === false
+    end
+
+    it "003: init -f -> init -f -> init --force" do
+      ret1 = bin_exec "init --defaults -f"
+      ret_config1 = git_do("config --get git.contest.branch.master")
+      ret2 = bin_exec "init --defaults -f"
+      ret3 = bin_exec "init --defaults --force"
+      ret_config1.should                                           === "master"
+      ret1.include?("Error: unknown argument").should              === false
+      ret2.include?("Error: unknown argument").should              === false
+      ret3.include?("Error: unknown argument").should              === false
+      ret1.include?("Already initialized for git-contest.").should === false
+      ret1.include?("use: git contest init -f").should             === false
+      ret2.include?("Already initialized for git-contest.").should === false
+      ret2.include?("use: git contest init -f").should             === false
+      ret3.include?("Already initialized for git-contest.").should === false
+      ret3.include?("use: git contest init -f").should             === false
+    end
+  end
+
+end
+

data/spec/lib/contest/driver/t001_aizu_online_judge_spec.rb

@@ -0,0 +1,177 @@
+require 'spec_helper'
+
+require 'mechanize'
+require 'contest/driver/aizu_online_judge'
+
+describe "T001: Contest::Driver::AizuOnlineJudge" do
+  before do
+    # setup driver
+    @driver = Contest::Driver::AizuOnlineJudge.new
+    @driver.stub(:sleep).and_return(0)
+    @driver.client = Mechanize.new {|agent|
+      agent.user_agent_alias = 'Windows IE 7'
+    }
+    ENV['GIT_CONTEST_CONFIG'] = get_path('/mock/t001/config.yml')
+    init
+
+    # basic status_log
+    WebMock
+    .stub_request(
+      :get,
+      /http:\/\/judge\.u-aizu\.ac\.jp\/onlinejudge\/webservice\/status_log/,
+    )
+    .to_return(
+      :status => 200,
+      :body => read_file('/mock/t001/status_log.xml'),
+      :headers => {
+        'Content-Type' => 'text/xml',
+      },
+    )
+
+    # status.jsp: http://judge.u-aizu.ac.jp/onlinejudge/status.jsp
+    WebMock
+    .stub_request(
+      :get,
+      /http:\/\/judge\.u-aizu\.ac\.jp\/onlinejudge\/status/,
+    )
+    .to_return(
+      :status => 200,
+      :body => read_file('/mock/t001/status.html'),
+      :headers => {
+        'Content-Type' => 'text/html',
+      },
+    )
+
+    # description
+    WebMock
+    .stub_request(
+      :get,
+      /http:\/\/judge\.u-aizu\.ac\.jp\/onlinejudge\/description\.jsp\?id=/,
+    )
+    .to_return(
+      :status => 200,
+      :body => read_file('/mock/t001/description.html'),
+      :headers => {
+        'Content-Type' => 'text/html',
+      },
+    )
+
+    # servlet/Submit
+    WebMock
+    .stub_request(
+      :post,
+      /http:\/\/judge\.u-aizu\.ac\.jp\/onlinejudge\/servlet\/Submit/,
+    )
+    .to_return(
+      :status => 200,
+      :body => '',
+      :headers => {
+        'Content-Type' => 'text/html',
+      },
+    )
+  end
+
+  describe "001: #get_status_wait" do
+    it "001: Check Status" do
+      ret = @driver.get_status_wait 'test_user', '111'
+      ret.should === "Wrong Answer"
+    end
+    it "002: Check Timeout" do
+      @flag = false
+      proc = Proc.new do
+        @flag = true
+      end
+      @driver.on 'timeout', proc
+      @driver.get_status_wait 'test_user', '999'
+      @driver.off 'timeout', proc
+      @flag.should === true
+    end
+    it "002: Check Timeout noset" do
+      @flag = false
+      proc = Proc.new do
+        @flag = true
+      end
+      @driver.on 'timeout', proc
+      @driver.off 'timeout', proc
+      @driver.get_status_wait 'test_user', '999'
+      @flag.should === false
+    end
+  end
+
+  describe "002: #get_status_wait" do
+    before do
+      # has 2 statuses
+      WebMock.stub_request(
+        :get,
+        /http:\/\/judge\.u-aizu\.ac\.jp\/onlinejudge\/webservice\/status_log\??.*/
+      ).to_return(
+        :status => 200,
+        :body => read_file('/mock/t001/002.status_log.xml'),
+        :headers => {
+          'Content-Type' => 'text/xml',
+        },
+      )
+    end
+    it "001: Check Status" do
+      ret = @driver.get_status_wait 'test_user', '111'
+      ret.should === "Wrong Answer"
+      ret = @driver.get_status_wait 'test_user', '112'
+      ret.should === "Accepted"
+    end
+  end
+
+  describe "003: #submit" do
+    it "001: Check Event" do
+      FileUtils.touch '/tmp/main.rb'
+
+      @flag_start         = false
+      @flag_before_submit = false
+      @flag_after_submit  = false
+      @flag_before_wait   = false
+      @flag_after_wait    = false
+      @flag_finish        = false
+
+      proc_start = Proc.new do
+        @flag_start = true
+      end
+      proc_before_submit = Proc.new do |info|
+        @flag_before_submit = true
+      end
+      proc_after_submit = Proc.new do
+        @flag_after_submit = true
+      end
+      proc_before_wait = Proc.new do
+        @flag_before_wait = true
+      end
+      proc_after_wait = Proc.new do
+        @flag_after_wait = true
+      end
+      proc_finish = Proc.new do
+        @flag_finish = true
+      end
+
+      @driver.on 'start', proc_start
+      @driver.on 'before_submit', proc_before_submit
+      @driver.on 'after_submit', proc_after_submit
+      @driver.on 'before_wait', proc_before_wait
+      @driver.on 'after_wait', proc_after_wait
+      @driver.on 'finish', proc_finish
+
+      @driver.submit(
+        {
+          "user" => "test_user",
+          "password" => "password",
+        },
+        '/tmp/main.rb',
+        {
+          :contest_id => '11111',
+          :problem_id => '22222',
+        },
+      )
+
+      @flag = @flag_start && @flag_before_submit && @flag_after_submit && @flag_before_wait && @flag_after_wait && @flag_finish
+      @flag.should === true
+    end
+  end
+end
+

data/spec/lib/contest/driver/t002_codeforces_spec.rb

@@ -0,0 +1,34 @@
+require 'spec_helper'
+
+require 'contest/driver/codeforces'
+require 'mechanize'
+
+describe "T002: Contest::Driver::Codeforces" do
+  before do
+    @driver = Contest::Driver::Codeforces.new
+    @driver.stub(:sleep).and_return(0)
+    @driver.client = Mechanize.new {|agent|
+      agent.user_agent_alias = 'Windows IE 7'
+    }
+
+    # basic status_log
+    WebMock.stub_request(
+      :get,
+      /http:\/\/codeforces.com\/contest\/[0-9A-Z]*\/my/
+    ).to_return(
+      :status => 200,
+      :body => read_file('/mock/t002/my_submissions.html'),
+      :header => {
+        'Content-Type' => 'text/html',
+      },
+    )
+  end
+
+  describe '001: #get_status_wait' do
+    it '001: Check Status' do
+      ret = @driver.get_status_wait 11111, 22222
+      ret.should === "Accepted"
+    end
+  end
+end
+

data/spec/lib/contest/driver/t003_uva_online_judge_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+
+require 'contest/driver/codeforces'
+require 'mechanize'
+
+describe "T003: Contest::Driver::UvaOnlineJudge" do
+  before do
+    @driver = Contest::Driver::UvaOnlineJudge.new
+    @driver.stub(:sleep).and_return(0)
+    @driver.client = Mechanize.new {|agent|
+      agent.user_agent_alias = 'Windows IE 7'
+    }
+  end
+
+  describe '001: #get_submission_id' do
+    it '001' do
+      ret = @driver.get_submission_id(read_file('/mock/t003/after_submit.html'))
+      ret.should eq '99999'
+    end
+  end
+
+  describe '002: #get_submission_status' do
+    it '001: Sent to Judge' do
+      ret = @driver.get_submission_status('99999', read_file('/mock/t003/my_submissions.sent_to_judge.html'))
+      ret.should eq 'Sent to judge'
+    end
+    it '002: Compile Error' do
+      ret = @driver.get_submission_status('99999', read_file('/mock/t003/my_submissions.compile_error.html'))
+      ret.should eq 'Compilation error'
+    end
+  end
+end
+

data/spec/mock/default_config/plugins/driver_dummy.rb

@@ -0,0 +1,113 @@
+require 'contest/driver/common'
+require 'rexml/document'
+
+module Contest
+  module Driver
+    class Dummy < DriverBase
+      def get_opts_ext
+        define_options do
+          opt(
+            :problem_id,
+            "Problem ID (Ex: 1000, 123, 0123, etc...)",
+            :type => :string,
+            :required => true,
+          )
+          opt(
+            :contest_id,
+            "Contest ID (Ex: 1000, 123, 0123, etc...)",
+            :type => :string,
+            :required => true,
+          )
+        end
+      end
+
+      def get_site_name
+        "Dummy"
+      end
+
+      def get_problem_id(options)
+        "#{options[:contest_id]}#{options[:problem_id]}"
+      end
+
+      def get_desc
+        "This is Dummy"
+      end
+
+      def resolve_language(label)
+        case label
+        when "clang"
+          return "C"
+        when "cpp"
+          return "C++"
+        when "java"
+          return "JAVA"
+        when "cpp11"
+          return "C++11"
+        when "cs"
+          return "C#"
+        when "dlang"
+          return "D"
+        when "golang"
+          return "Go"
+        when "ruby"
+          return "Ruby"
+        when "python2"
+          return "Python2"
+        when "python3"
+          return "Python"
+        when "php"
+          return "PHP"
+        when "javascript"
+          return "JavaScript"
+        else
+          abort "unknown languag"
+        end
+      end
+
+      def submit_ext(config, source_path, options)
+        # start
+        trigger 'start'
+
+        # submit
+        trigger(
+          'before_submit',
+          {
+            :source => source_path
+          },
+        )
+        trigger 'after_submit'
+
+        # need to get the newest waiting submissionId
+        trigger 'before_wait'
+
+        # wait result
+        status = ''
+        File.open source_path, 'r' do |file|
+          line = file.read
+          if line == 'wa-code'
+            status = 'Wrong Answer'
+          elsif line == 'ac-code'
+            status = 'Accepted'
+          elsif line == 'tle-code'
+            status = 'Time Limit Exceeded'
+          end
+        end
+        trigger(
+          'after_wait',
+          {
+            :submission_id => '99999',
+            :status => status,
+            :result => get_commit_message($config["submit_rules"]["message"], status, options),
+          }
+        )
+        trigger 'finish'
+
+        status
+      end
+
+      if ENV['TEST_MODE'] === 'TRUE'
+        attr_writer :client
+      end
+    end
+  end
+end

data/spec/mock/t001/002.status_log.xml

@@ -0,0 +1,55 @@
+<?xml version="1.0"?>
+<status_list>
+  <status>
+    <run_id>
+      112
+    </run_id>
+    <user_id>
+      test_user
+    </user_id>
+    <problem_id>
+      222
+    </problem_id>
+    <status>
+      Accepted
+    </status>
+    <language>
+      Ruby
+    </language>
+    <cputime>
+      333
+    </cputime>
+    <memory>
+      444
+    </memory>
+    <code_size>
+      555
+    </code_size>
+  </status>
+  <status>
+    <run_id>
+      111
+    </run_id>
+    <user_id>
+      test_user
+    </user_id>
+    <problem_id>
+      222
+    </problem_id>
+    <status>
+      Wrong Answer
+    </status>
+    <language>
+      Ruby
+    </language>
+    <cputime>
+      333
+    </cputime>
+    <memory>
+      444
+    </memory>
+    <code_size>
+      555
+    </code_size>
+  </status>
+</status_list>