lib/contentful_importer.rb in storageroom-to-contentful-0.0.3 vs lib/contentful_importer.rb in storageroom-to-contentful-0.0.4

@@ -7,10 +7,34 @@
 
   def initialize
     Contentful::Management::Client.new(CREDENTIALS['ACCESS_TOKEN'])
   end
 
+  def test_credentials
+    test_contetful_credentials
+    test_storageroom_credentials
+  end
+
+  def test_contetful_credentials
+    space = Contentful::Management::Space.all
+    if space.is_a? Contentful::Management::Array
+      puts 'Contentful Management API credentials: OK'
+    end
+  rescue NoMethodError => e
+    puts 'Contentful Management API credentials: INVALID (check README)'
+  end
+
+  def test_storageroom_credentials
+    request = StorageRoomExporter.new.send(:get_request, 'collections')
+    if request.is_a? Hash
+      puts 'StorageRoom API credentials: OK'
+    end
+  rescue RuntimeError => e
+    puts 'StorageRoom API credentials: INVALID (check README)'
+    puts e
+  end
+
   def create_space
     puts 'Name for a new created space on Contentful:'
     name_space = gets.strip
     @space = Contentful::Management::Space.create(name: name_space, organization_id: CREDENTIALS['ORGANIZATION_ID'])
   end
@@ -257,6 +281,7 @@
     Contentful::Management::Field.new.tap do |field|
       field.type = params['link'] || 'Link'
       field.link_type = params['link_type']
     end
   end
+
 end