From f64898deab3bfc04e9a46d6d17ba6ab3db8b1554 Mon Sep 17 00:00:00 2001
From: Oleg Hasjanov <oleg.phenomenon@gmail.com>
Date: Mon, 10 May 2021 09:48:16 +0300
Subject: [PATCH] added test

---
 .../api/v1/registrant/domains_test.rb         | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 test/integration/api/v1/registrant/domains_test.rb

diff --git a/test/integration/api/v1/registrant/domains_test.rb b/test/integration/api/v1/registrant/domains_test.rb
new file mode 100644
index 000000000..367df9e48
--- /dev/null
+++ b/test/integration/api/v1/registrant/domains_test.rb
@@ -0,0 +1,46 @@
+require 'test_helper'
+require 'auth_token/auth_token_creator'
+
+CompanyRegisterClientStub = Struct.new(:any_method) do
+  def representation_rights(citizen_personal_code:, citizen_country_code:)
+    raise CompanyRegister::NotAvailableError
+  end
+end
+
+class RegistrantApiV1DomainsTest < ActionDispatch::IntegrationTest
+  setup do
+    @user = users(:registrant)
+    @registrar = registrars(:bestnames)
+    @contact = contacts(:john)
+  end
+
+  def test_get_default_counts_of_domains
+    get api_v1_registrant_domains_path + "?tech=init", as: :json,
+    headers: { 'HTTP_AUTHORIZATION' => auth_token }
+
+    assert_response :ok
+
+    response_json = JSON.parse(response.body)
+    assert_equal response_json['total'], 4
+    assert_equal response_json['count'], 4
+  end
+
+  def test_get_default_counts_of_direct_domains
+    CompanyRegister::Client.stub(:new, CompanyRegisterClientStub.new) do
+      get api_v1_registrant_domains_path + "?tech=init", as: :json,
+          headers: { 'HTTP_AUTHORIZATION' => auth_token }
+    end
+
+    response_json = JSON.parse(response.body)
+    assert_equal response_json['total'], 4
+    assert_equal response_json['count'], 4
+  end
+
+  private
+
+  def auth_token
+    token_creator = AuthTokenCreator.create_with_defaults(@user)
+    hash = token_creator.token_in_hash
+    "Bearer #{hash[:access_token]}"
+  end
+end