Cohort not modified should not throw exception

amplitude · Jul 31, 2024 · 73672b2 · 73672b2
1 parent c4c4260
commit 73672b2
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 15 deletions.
diff --git a/pkg/experiment/local/cohort_download_api.go b/pkg/experiment/local/cohort_download_api.go
@@ -48,7 +48,7 @@ func (api *directCohortDownloadApi) getCohort(cohortID string, cohort *Cohort) (
 			errors++
 			if errors >= 3 || func(err error) bool {
 				switch err.(type) {
-				case *CohortNotModifiedException, *CohortTooLargeException:
+				case *CohortTooLargeException:
 					return true
 				default:
 					return false
@@ -85,7 +85,7 @@ func (api *directCohortDownloadApi) getCohort(cohortID string, cohort *Cohort) (
 				}(),
 			}, nil
 		} else if response.StatusCode == http.StatusNoContent {
-			return nil, &CohortNotModifiedException{Message: "Cohort not modified"}
+			return nil, nil
 		} else if response.StatusCode == http.StatusRequestEntityTooLarge {
 			return nil, &CohortTooLargeException{Message: "Cohort exceeds max cohort size"}
 		} else {

diff --git a/pkg/experiment/local/cohort_download_api_test.go b/pkg/experiment/local/cohort_download_api_test.go
@@ -214,9 +214,8 @@ func TestCohortDownloadApi(t *testing.T) {
 			httpmock.NewStringResponder(204, ""),
 		)
 
-		_, err := api.getCohort("1234", cohort)
-		assert.Error(t, err)
-		_, isCohortNotModifiedException := err.(*CohortNotModifiedException)
-		assert.True(t, isCohortNotModifiedException)
+		result, err := api.getCohort("1234", cohort)
+		assert.Nil(t, result)
+		assert.NoError(t, err)
 	})
 }
diff --git a/pkg/experiment/local/cohort_loader.go b/pkg/experiment/local/cohort_loader.go
@@ -69,7 +69,9 @@ func (task *CohortLoaderTask) run() {
 	if err != nil {
 		task.err = err
 	} else {
-		task.loader.cohortStorage.putCohort(cohort)
+		if cohort != nil {
+			task.loader.cohortStorage.putCohort(cohort)
+		}
 	}
 
 	task.loader.removeJob(task.cohortId)

diff --git a/pkg/experiment/local/exception.go b/pkg/experiment/local/exception.go
@@ -16,11 +16,3 @@ type CohortTooLargeException struct {
 func (e *CohortTooLargeException) Error() string {
 	return e.Message
 }
-
-type CohortNotModifiedException struct {
-	Message string
-}
-
-func (e *CohortNotModifiedException) Error() string {
-	return e.Message
-}