git-contest 0.0.3 → 0.1.0

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

@@ -3,31 +3,178 @@ require 'spec_helper'
 require 'contest/driver/codeforces'
 require 'mechanize'
 
-describe "T003: Contest::Driver::UvaOnlineJudge" do
-  before do
+describe "T003: UvaOnlineJudge Driver" do
+  before :each do
+    @test_dir = "#{ENV['GIT_CONTEST_TEMP_DIR']}/t003"
+    Dir.mkdir @test_dir
+    Dir.chdir @test_dir
+  end
+
+  after :each do
+    Dir.chdir @test_dir
+    Dir.chdir '..'
+    FileUtils.remove_dir @test_dir, :force => true
+  end
+
+  before(:each) 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
+  context "A001: #get_submission_id" do
+    it "should return last submission id" 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
+  context "A002: #get_submission_status" do
+    it "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
+
+    it "Compilation error" do
       ret = @driver.get_submission_status('99999', read_file('/mock/t003/my_submissions.compile_error.html'))
       ret.should eq 'Compilation error'
     end
   end
+
+  context "A003: #submit" do
+    before do
+      FileUtils.touch 'test_source.java'
+    end
+
+    before do
+      # index.html
+      WebMock
+      .stub_request(
+        :get,
+        /^http:\/\/uva\.onlinejudge\.org\/$/,
+      )
+      .to_return(
+        :status => 200,
+        :body => read_file('/mock/t003/uva_home.html'),
+        :headers => {
+          'Content-Type' => 'text/html',
+        },
+      )
+
+      # login
+      WebMock
+      .stub_request(
+        :post,
+        /^http:\/\/uva\.onlinejudge\.org\/index\.php\?option=com_comprofiler&task=login$/,
+      )
+      .to_return(
+        :status => 200,
+        :headers => {
+          'Content-Type' => 'text/html',
+        },
+      )
+
+      # quick submit page
+      WebMock
+      .stub_request(
+        :get,
+        /^http:\/\/uva\.onlinejudge\.org\/index\.php\?Itemid=25&option=com_onlinejudge$/,
+      )
+      .to_return(
+        :status => 200,
+        :body => read_file('/mock/t003/uva_quick_submit.html'),
+        :headers => {
+          'Content-Type' => 'text/html',
+        },
+      )
+
+      # quick submit
+      WebMock
+      .stub_request(
+        :post,
+        /^http:\/\/uva\.onlinejudge\.org\/index\.php\?Itemid=25&option=com_onlinejudge&page=save_submission$/,
+      )
+      .to_return(
+        :status => 200,
+        :body => read_file('/mock/t003/uva_after_quick_submit.html'),
+        :headers => {
+          'Content-Type' => 'text/html',
+        },
+      )
+
+      # status
+      WebMock
+      .stub_request(
+        :get,
+        /^http:\/\/uva\.onlinejudge\.org\/index\.php\?Itemid=9&option=com_onlinejudge$/
+      )
+      .to_return(
+        :status => 200,
+        :body => read_file('/mock/t003/uva_my_submissions.html'),
+        :headers => {
+          'Content-Type' => 'text/html',
+        },
+      )
+    end
+
+    it "should return commit message" do
+      @driver.config.merge!(
+        "user" => "test_user",
+        "password" => "password",
+      )
+      @driver.options.merge!(
+        :problem_id => '333333',
+        :source => 'test_source.java',
+      )
+      @driver.submit.should include "UVa 333333: Wrong answer"
+    end
+
+    it "Check events" do
+      @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.config.merge!(
+        "user" => "test_user",
+        "password" => "password",
+      )
+      @driver.options.merge!(
+        :problem_id => '333333',
+        :source => 'test_source.java',
+      )
+      @driver.submit()
+
+      @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/t010_kattis_spec.rb

@@ -0,0 +1,207 @@
+require "spec_helper"
+require "contest/driver/kattis"
+
+describe "T010: Kattis Driver" do
+  before :each do
+    @test_dir = "#{ENV['GIT_CONTEST_TEMP_DIR']}/t010"
+    Dir.mkdir @test_dir
+    Dir.chdir @test_dir
+  end
+
+  after :each do
+    Dir.chdir @test_dir
+    Dir.chdir '..'
+    FileUtils.remove_dir @test_dir, :force => true
+  end
+
+  before do
+    @driver = Contest::Driver::Kattis.new
+    @driver.stub(:sleep).and_return(0)
+  end
+
+  context "A001: #get_submission_id" do
+    subject do
+      dummy_html = read_file('/mock/t010/user_submissions.html')
+      @driver.get_submission_id dummy_html
+    end
+    it "must return last submission id" do
+      should eq "111111"
+    end
+  end
+
+  context "A002: #get_submissions_status" do
+    context "111111" do
+      subject do
+        dummy_html = read_file('/mock/t010/user_submission_111111.html')
+        @driver.get_submission_status "111111", dummy_html
+      end
+      it "must return the status of specified submission" do
+        should eq "Accepted"
+      end
+    end
+    context "222222" do
+      subject do
+        dummy_html = read_file('/mock/t010/user_submission_222222.html')
+        @driver.get_submission_status "222222", dummy_html
+      end
+      it "must return the status of specified submission" do
+        should eq "Wrong Answer"
+      end
+    end
+  end
+
+  context "A003: #submit" do
+    before do
+      FileUtils.touch 'test_source.go'
+    end
+
+    before do
+      # login page
+      WebMock.stub_request(
+        :get,
+        /^https:\/\/open\.kattis\.com\/login\?email_login=true$/
+      )
+      .to_return(
+        :status => 200,
+        :body => read_file('/mock/t010/open_kattis_com_login.html'),
+        :headers => {
+          'Content-Type' => 'text/html',
+        },
+      )
+
+      # login
+      WebMock.stub_request(
+        :post,
+        /^https:\/\/open\.kattis\.com\/login\?email_login=true$/
+      )
+      .to_return(
+        :status => 200,
+        :headers => {
+          'Content-Type' => 'text/html',
+        },
+      )
+
+      # submit page
+      WebMock.stub_request(
+        :get,
+        /^https:\/\/open\.kattis\.com\/submit$/,
+      )
+      .to_return(
+        :status => 200,
+        :body => read_file('/mock/t010/open_kattis_com_submit.html'),
+        :headers => {
+          'Content-Type' => 'text/html',
+        },
+      )
+
+      # submit
+      WebMock.stub_request(
+        :post,
+        /^https:\/\/open\.kattis\.com\/submit$/,
+      )
+      .to_return(
+        :status => 200,
+        :headers => {
+          'Content-Type' => 'text/html',
+        },
+      )
+
+      # user submissions
+      WebMock.stub_request(
+        :get,
+        /^https:\/\/open\.kattis\.com\/users\/test_user\?show=submissions$/
+      )
+      .to_return(
+        :status => 200,
+        :body => read_file('/mock/t010/open_kattis_com_user_submissions.html'),
+        :headers => {
+          'Content-Type' => 'text/html',
+        },
+      )
+
+      # submission 999999
+      WebMock.stub_request(
+        :get,
+        /^https:\/\/open\.kattis\.com\/submission\?id=999999$/
+      )
+      .to_return(
+        :status => 200,
+        :body => read_file('/mock/t010/open_kattis_com_user_submissions.html'),
+        :headers => {
+          'Content-Type' => 'text/html',
+        },
+      )
+    end
+
+    it "should return commit message" do
+      @driver.config.merge!(
+        "user" => "test_user",
+        "password" => "password",
+      )
+      @driver.options.merge!(
+        :problem_id => '333333',
+        :source => 'test_source.go',
+      )
+      @driver.submit.should include "Kattis 333333: Wrong Answer"
+    end
+
+    it "check events" do
+      @flag_start         = false
+      @flag_before_login  = false
+      @flag_after_login   = 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_login = Proc.new do |info|
+        @flag_before_login = true
+      end
+      proc_after_login = Proc.new do
+        @flag_after_login = 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_login', proc_before_login
+      @driver.on 'after_login', proc_after_login
+      @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.config.merge!(
+        "user" => "test_user",
+        "password" => "password",
+      )
+      @driver.options.merge!(
+        :problem_id => '333333',
+        :source => 'test_source.go',
+      )
+      @driver.submit
+
+      @flag = @flag_start && @flag_before_login && @flag_after_login && @flag_before_submit && @flag_after_submit && @flag_before_wait && @flag_after_wait && @flag_finish
+      @flag.should === true
+    end
+  end
+end
+

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

@@ -64,15 +64,20 @@ module Contest
         end
       end
 
-      def submit_ext(config, source_path, options)
+      def submit_ext
         # start
         trigger 'start'
 
         # submit
+        if @options[:source].is_a? Array
+          source_text = @options[:source].join(", ")
+        else
+          source_text = @options[:source]
+        end
         trigger(
           'before_submit',
           {
-            :source => source_path
+            :source => source_text
           },
         )
         trigger 'after_submit'
@@ -82,22 +87,37 @@ module Contest
 
         # 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'
+
+        if @options[:source].is_a? Array
+          File.open @options[:source][0], '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
+        elsif @options[:source].is_a? String
+          File.open @options[:source], '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
         end
+
         trigger(
           'after_wait',
           {
             :submission_id => '99999',
             :status => status,
-            :result => get_commit_message($config["submit_rules"]["message"], status, options),
+            :result => get_commit_message(status),
           }
         )
         trigger 'finish'