Merge pull request drone#49 from wlynch/fake

feat: Implement CreateStatus, return PR labels in fake.
jstrachan · Nov 6, 2019 · 6c4b4d8 · 6c4b4d8
2 parents e4a40b2 + ccc8f6e
commit 6c4b4d8
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/scm/driver/fake/pr.go b/scm/driver/fake/pr.go
@@ -18,6 +18,9 @@ func (s *pullService) Find(ctx context.Context, repo string, number int) (*scm.P
 	if !exists {
 		return nil, nil, fmt.Errorf("Pull request number %d does not exit", number)
 	}
+	if labels, _, err := s.client.Issues.ListLabels(ctx, repo, number, scm.ListOptions{}); err == nil {
+		val.Labels = labels
+	}
 	return val, nil, nil
 }
 

diff --git a/scm/driver/fake/repo.go b/scm/driver/fake/repo.go
@@ -98,8 +98,21 @@ func (s *repositoryService) CreateHook(context.Context, string, *scm.HookInput)
 	panic("implement me")
 }
 
-func (s *repositoryService) CreateStatus(context.Context, string, string, *scm.StatusInput) (*scm.Status, *scm.Response, error) {
-	panic("implement me")
+func (s *repositoryService) CreateStatus(ctx context.Context, repo string, ref string, in *scm.StatusInput) (*scm.Status, *scm.Response, error) {
+	statuses := s.data.Statuses[ref]
+	if statuses == nil {
+		statuses = []*scm.Status{}
+	}
+	status := scm.ConvertStatusInputToStatus(in)
+	for _, existing := range statuses {
+		if existing.Label == status.Label {
+			*existing = *status
+			return status, nil, nil
+		}
+	}
+	statuses = append(statuses, status)
+	s.data.Statuses[ref] = statuses
+	return status, nil, nil
 }
 
 func (s *repositoryService) DeleteHook(context.Context, string, string) (*scm.Response, error) {