1 files changed, 50 insertions, 7 deletions
diff --git a/spec/lib/gitlab/github_import_spec.rb b/spec/lib/gitlab/github_import_spec.rb
index 1a690b81d2b..3129da64809 100644
--- a/spec/lib/gitlab/github_import_spec.rb
+++ b/spec/lib/gitlab/github_import_spec.rb
@@ -3,13 +3,13 @@
 require 'spec_helper'
 
 RSpec.describe Gitlab::GithubImport do
-  let(:project) { double(:project) }
+  context 'github.com' do
+    let(:project) { double(:project, import_url: 'http://t0ken@github.com/user/repo.git') }
 
-  describe '.new_client_for' do
     it 'returns a new Client with a custom token' do
       expect(described_class::Client)
         .to receive(:new)
-        .with('123', parallel: true)
+        .with('123', host: nil, parallel: true)
 
       described_class.new_client_for(project, token: '123')
     end
@@ -23,18 +23,57 @@ RSpec.describe Gitlab::GithubImport do
 
       expect(described_class::Client)
         .to receive(:new)
-        .with('123', parallel: true)
+        .with('123', host: nil, parallel: true)
 
       described_class.new_client_for(project)
     end
+
+    it 'returns the ID of the ghost user', :clean_gitlab_redis_cache do
+      expect(described_class.ghost_user_id).to eq(User.ghost.id)
+    end
+
+    it 'caches the ghost user ID', :clean_gitlab_redis_cache do
+      expect(Gitlab::Cache::Import::Caching)
+        .to receive(:write)
+        .once
+        .and_call_original
+
+      2.times do
+        described_class.ghost_user_id
+      end
+    end
   end
 
-  describe '.ghost_user_id', :clean_gitlab_redis_cache do
-    it 'returns the ID of the ghost user' do
+  context 'GitHub Enterprise' do
+    let(:project) { double(:project, import_url: 'http://t0ken@github.another-domain.com/repo-org/repo.git') }
+
+    it 'returns a new Client with a custom token' do
+      expect(described_class::Client)
+        .to receive(:new)
+        .with('123', host: 'http://github.another-domain.com/api/v3', parallel: true)
+
+      described_class.new_client_for(project, token: '123')
+    end
+
+    it 'returns a new Client with a token stored in the import data' do
+      import_data = double(:import_data, credentials: { user: '123' })
+
+      expect(project)
+        .to receive(:import_data)
+        .and_return(import_data)
+
+      expect(described_class::Client)
+        .to receive(:new)
+        .with('123', host: 'http://github.another-domain.com/api/v3', parallel: true)
+
+      described_class.new_client_for(project)
+    end
+
+    it 'returns the ID of the ghost user', :clean_gitlab_redis_cache do
       expect(described_class.ghost_user_id).to eq(User.ghost.id)
     end
 
-    it 'caches the ghost user ID' do
+    it 'caches the ghost user ID', :clean_gitlab_redis_cache do
       expect(Gitlab::Cache::Import::Caching)
         .to receive(:write)
         .once
@@ -44,5 +83,9 @@ RSpec.describe Gitlab::GithubImport do
         described_class.ghost_user_id
       end
     end
+
+    it 'formats the import url' do
+      expect(described_class.formatted_import_url(project)).to eq('http://github.another-domain.com/api/v3')
+    end
   end
 end