fix panic when updating org with existing schedule (#196)

control/schedule.go

@@ -310,17 +310,17 @@ func (c *Client) ScheduleNow(ctx context.Context, org string, info *OrgInfo, pha
 		if !phases[0].Effective.IsZero() {
 			return errors.New("first phase must be effective now")
 		}
-	}
-	cps, err := c.LookupPhases(ctx, org)
-	if err != nil && !errors.Is(err, ErrOrgNotFound) {
-		return err
-	}
-	for _, p := range cps {
-		if p.Current {
-			p0 := phases[0]
-			p.Features = p0.Features
-			phases[0] = p
-			break
+		cps, err := c.LookupPhases(ctx, org)
+		if err != nil && !errors.Is(err, ErrOrgNotFound) {
+			return err
+		}
+		for _, p := range cps {
+			if p.Current {
+				p0 := phases[0]
+				p.Features = p0.Features
+				phases[0] = p
+				break
+			}
 		}
 	}
 	return c.Schedule(ctx, org, info, phases)

control/schedule_test.go

@@ -133,6 +133,27 @@ func TestSchedule(t *testing.T) {
 	})
 }
 
+func TestScheduleUpdateOrgOnSchedule(t *testing.T) {
+	info := &OrgInfo{Email: "test@foo.com"}
+	c := newTestClient(t)
+	ctx := context.Background()
+	c.Push(ctx, []Feature{{
+		FeaturePlan: mpf("feature:x@plan:test@0"),
+		Interval:    "@daily",
+		Currency:    "usd",
+	}}, pushLogger(t))
+	err := c.Schedule(ctx, "org:example", info, []Phase{{
+		Features: []refs.FeaturePlan{mpf("feature:x@plan:test@0")},
+	}})
+	if err != nil {
+		t.Fatalf("got %v, want nil", err)
+	}
+	err = c.ScheduleNow(ctx, "org:example", info, nil) // org update only
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
 func TestScheduleMinMaxItems(t *testing.T) {
 	c := newTestClient(t)
 	ctx := context.Background()