Fix connection refused error

apache · Jun 3, 2017 · 61e8e64 · 61e8e64
1 parent cc3ec30
commit 61e8e64
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/superset/views/core.py b/superset/views/core.py
@@ -403,9 +403,7 @@ def csv_to_df(filepath_or_buffer, sep, header, names, index_col, squeeze,
                   dayfirst, thousands, decimal, quotechar, escapechar, comment,
                   encoding, error_bad_lines, chunksize):
         # Use Pandas to parse csv file to a dataframe
-        upload_path = 'http://' + config['SUPERSET_WEBSERVER_ADDRESS'] + ':' \
-                      + str(config['SUPERSET_WEBSERVER_PORT']) \
-                      + url_for('uploaded_file', filename=filepath_or_buffer)
+        upload_path = config['UPLOAD_FOLDER'] + filepath_or_buffer
         # Expose this to api so can specify each field
         chunks = pandas.read_csv(filepath_or_buffer=upload_path,
                                  sep=sep,

diff --git a/tests/core_tests.py b/tests/core_tests.py
@@ -647,18 +647,19 @@ def test_user_profile(self, username='admin'):
 
     def test_import_csv(self):
         self.login(username='admin')
-        config = app.config
 
-        test_file = open('tests/testCSV.csv', 'w+')
+        os.environ['SUPERSET_CONFIG'] = 'tests.superset_test_config'
+        con = app.config.get('SQLALCHEMY_DATABASE_URI')
 
+        test_file = open('tests/testCSV.csv', 'w+')
         test_file.write(u'Column 1, Column 2\n')
-        test_file.write(u'Test1, Test 2')
+        test_file.write(u'Test 1, Test 2')
         test_file.seek(0)
 
         form_data = {'csv_file': test_file,
                             'sep': ',',
                             'name': 'TestName',
-                            'con': config['SQLALCHEMY_DATABASE_URI'],
+                            'con': con,
                             'if_exists': 'append',
                             'index_label': 'test_label',
                             'chunksize': 1,