fix: Fix to receive partial contents (#15)

screwdriver-cd · Aug 14, 2018 · fcb48cf · fcb48cf
1 parent 51f6a9c
commit fcb48cf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/hab/hab.go b/hab/hab.go
@@ -53,7 +53,7 @@ func (depo *depot) packagesInfo(pkgName string, from int) (PackagesInfo, error)
 
 	if res.StatusCode == 404 {
 		return PackagesInfo{}, errors.New("Package not found")
-	} else if res.StatusCode != 200 {
+	} else if res.StatusCode/100 != 2 {
 		return PackagesInfo{}, fmt.Errorf("Unexpected status code: %d", res.StatusCode)
 	}
 

diff --git a/hab/hab_test.go b/hab/hab_test.go
@@ -186,7 +186,7 @@ func TestPackagesInfoFromName(t *testing.T) {
 				},
 			},
 			expected:      []string{"0.0.1", "0.0.2", "0.1.0", "0.1.1", "1.0.0"},
-			statusCode:    200,
+			statusCode:    206,
 			httpError:     nil,
 			expectedError: nil,
 		},
@@ -258,7 +258,7 @@ func TestPackagesInfoFromName(t *testing.T) {
 				},
 			},
 			expected:      []string{"0.0.1", "0.0.2", "0.1.0", "0.1.1", "1.0.0", "1.0.1", "1.1.0", "2.0.0"},
-			statusCode:    200,
+			statusCode:    206,
 			httpError:     nil,
 			expectedError: nil,
 		},