Add test for nil due_on value during import

1 files changed, 21 insertions, 0 deletions

diff --git a/spec/lib/gitlab/github_import/importer/milestones_importer_spec.rb b/spec/lib/gitlab/github_import/importer/milestones_importer_spec.rb
index 471f499dd19..35c37f9499d 100644
--- a/spec/lib/gitlab/github_import/importer/milestones_importer_spec.rb
+++ b/spec/lib/gitlab/github_import/importer/milestones_importer_spec.rb
@@ -21,6 +21,19 @@ describe Gitlab::GithubImport::Importer::MilestonesImporter, :clean_gitlab_redis
     )
   end
 
+  let(:milestone2) do
+    double(
+      :milestone,
+      number: 1,
+      title: '1.0',
+      description: 'The first release',
+      state: 'open',
+      due_on: nil,
+      created_at: created_at,
+      updated_at: updated_at
+    )
+  end
+
   describe '#execute' do
     it 'imports the milestones in bulk' do
       milestone_hash = { number: 1, title: '1.0' }
@@ -114,6 +127,14 @@ describe Gitlab::GithubImport::Importer::MilestonesImporter, :clean_gitlab_redis
     end
   end
 
+  describe '#nil_due_on' do
+    let(:milestone_hash) { importer.build(milestone2) }
+
+    it 'should handle missing due_on correctly' do
+      expect(milestone_hash[:due_date]).to be nil
+    end
+  end
+
   describe '#each_milestone' do
     it 'returns the milestones' do
       expect(client)