Ensure Owner Or Organization Present

/cc https://github.com/terraform-providers/terraform-provider-github/pull/521#discussion_r461593258
integrations · Jul 31, 2020 · eee8ea8 · eee8ea8
1 parent 4f1a7ea
commit eee8ea8
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/github/provider_utils.go b/github/provider_utils.go
@@ -47,13 +47,23 @@ func testAccPreCheck(t *testing.T) {
 }
 
 func testAccCheckOrganization() error {
+
 	baseURL := os.Getenv("GITHUB_BASE_URL")
 	token := os.Getenv("GITHUB_TOKEN")
 
+	owner := os.Getenv("GITHUB_OWNER")
+	if owner == "" {
+		organization := os.Getenv("GITHUB_ORGANIZATION")
+		if organization == "" {
+			return fmt.Errorf("Neither `GITHUB_OWNER` or `GITHUB_ORGANIZATION` set in environment")
+		}
+		owner = organization
+	}
+
 	config := Config{
 		BaseURL: baseURL,
 		Token:   token,
-		Owner:   testOwner,
+		Owner:   owner,
 	}
 
 	meta, err := config.Meta()