소스 검색

HUE-8728 [jb] Fix download test.

jdesjean 6 년 전
부모
커밋
c41b8ec
1개의 변경된 파일6개의 추가작업 그리고 3개의 파일을 삭제
  1. 6 3
      apps/filebrowser/src/filebrowser/views_test.py

+ 6 - 3
apps/filebrowser/src/filebrowser/views_test.py

@@ -825,9 +825,12 @@ alert("XSS")
     f.write(sdf_string)
     f.close()
 
-    response = self.c.get('/filebrowser/download=%s/xss?disposition=inline' % prefix, follow=True)
-    assert_equal(200, response.status_code)
-    assert_equal('attachment', response['Content-Disposition'])
+    response = self.c.get('/filebrowser/download=%s/xss?disposition=inline' % prefix, follow=False) # The client does not support redirecting to another host. follow=False
+    if response.status_code == 302: # Redirects to webhdfs
+      assert_true(response.url.find('webhdfs') >= 0)
+    else:
+      assert_equal(200, response.status_code)
+      assert_equal('attachment', response['Content-Disposition'])
 
     # Download fails and displays exception because of missing permissions
     self.cluster.fs.chmod(prefix + '/xss', 0700)