From 1fba3a731c84c86f076424ee17362ace03512a7a Mon Sep 17 00:00:00 2001
From: Sergei Tsoganov <sergei@Sergeis-MacBook-Pro.local>
Date: Wed, 23 Nov 2022 14:55:34 +0200
Subject: [PATCH] Added tests

---
 .../v1/registrar/xml_console_controller.rb    |   3 +
 .../repp/v1/registrar/xml_console_test.rb     | 105 ++++++++++++++++++
 2 files changed, 108 insertions(+)
 create mode 100644 test/integration/repp/v1/registrar/xml_console_test.rb

diff --git a/app/controllers/repp/v1/registrar/xml_console_controller.rb b/app/controllers/repp/v1/registrar/xml_console_controller.rb
index e022a31a6..37d28192f 100644
--- a/app/controllers/repp/v1/registrar/xml_console_controller.rb
+++ b/app/controllers/repp/v1/registrar/xml_console_controller.rb
@@ -4,6 +4,9 @@ module Repp
       class XmlConsoleController < BaseController
         include EppRequestable
 
+        THROTTLED_ACTIONS = %i[load_xml create].freeze
+        include Shunter::Integration::Throttle
+
         PREFS = %w[
           domain-ee
           contact-ee
diff --git a/test/integration/repp/v1/registrar/xml_console_test.rb b/test/integration/repp/v1/registrar/xml_console_test.rb
new file mode 100644
index 000000000..19c82496f
--- /dev/null
+++ b/test/integration/repp/v1/registrar/xml_console_test.rb
@@ -0,0 +1,105 @@
+require 'test_helper'
+
+class ReppV1RegistrarXmlConsoleTest < ActionDispatch::IntegrationTest
+  def setup
+    @user = users(:api_bestnames)
+    token = Base64.encode64("#{@user.username}:#{@user.plain_text_password}")
+    token = "Basic #{token}"
+
+    @auth_headers = { 'Authorization' => token }
+    adapter = ENV['shunter_default_adapter'].constantize.new
+    adapter&.clear!
+  end
+
+  def test_load_schema_path
+    get load_xml_repp_v1_registrar_xml_console_path, params: { obj: 'domain', epp_action: 'update' },
+                                                     headers: @auth_headers
+
+    assert_response :ok
+    json = JSON.parse(response.body, symbolize_names: true)
+    assert_equal update_payload, json[:data][:xml]
+  end
+
+  def test_check_schema_path
+    post repp_v1_registrar_xml_console_path, params: { xml_console: { payload: payload } }.to_json,
+                                             headers: @auth_headers
+
+    assert_response :ok
+  end
+
+  private
+
+  def payload
+    <<~XML
+      <?xml version="1.0" encoding="UTF-8" standalone="no"?>
+      <epp xmlns="https://epp.tld.ee/schema/epp-ee-1.0.xsd">
+        <command>
+          <check>
+            <domain:check xmlns:domain="https://epp.tld.ee/schema/domain-ee-1.1.xsd">
+              <domain:name>auction.test</domain:name>
+            </domain:check>
+          </check>
+        </command>
+      </epp>
+    XML
+  end
+
+  def update_payload
+    <<~XML
+      <?xml version="1.0" encoding="UTF-8" standalone="no"?>
+      <epp xmlns="https://epp.tld.ee/schema/epp-ee-1.0.xsd">
+        <command>
+          <update>
+            <domain:update
+             xmlns:domain="https://epp.tld.ee/schema/domain-ee-1.2.xsd">
+              <domain:name>example.ee</domain:name>
+              <domain:add>
+                <domain:ns>
+                  <domain:hostAttr>
+                    <domain:hostName>ns1.example.com</domain:hostName>
+                  </domain:hostAttr>
+                  <domain:hostAttr>
+                    <domain:hostName>ns2.example.com</domain:hostName>
+                  </domain:hostAttr>
+                </domain:ns>
+                <domain:contact type="tech">mak21</domain:contact>
+              </domain:add>
+              <domain:rem>
+                <domain:ns>
+                  <domain:hostAttr>
+                    <domain:hostName>ns1.example.net</domain:hostName>
+                  </domain:hostAttr>
+                </domain:ns>
+                <domain:contact type="tech">mak21</domain:contact>
+              </domain:rem>
+              <domain:chg>
+                <domain:registrant>mak21</domain:registrant>
+                <domain:authInfo>
+                  <domain:pw>newpw</domain:pw>
+                </domain:authInfo>
+              </domain:chg>
+            </domain:update>
+          </update>
+          <extension>
+            <secDNS:update xmlns:secDNS="urn:ietf:params:xml:ns:secDNS-1.1">
+              <secDNS:rem>
+                <secDNS:keyData>
+                  <secDNS:flags>257</secDNS:flags>
+                  <secDNS:protocol>3</secDNS:protocol>
+                  <secDNS:alg>8</secDNS:alg>
+                  <secDNS:pubKey>700b97b591ed27ec2590d19f06f88bba700b97b591ed27ec2590d19f</secDNS:pubKey>
+                </secDNS:keyData>
+              </secDNS:rem>
+            </secDNS:update>
+            <eis:extdata xmlns:eis="https://epp.tld.ee/schema/eis-1.0.xsd">
+              <eis:legalDocument type="pdf">
+                dGVzdCBmYWlsCg==
+              </eis:legalDocument>
+            </eis:extdata>
+          </extension>
+          <clTRID>test_bestnames-#{Time.zone.now.to_i}</clTRID>
+        </command>
+      </epp>
+    XML
+  end
+end
\ No newline at end of file