Ensure Owner Or Organization Present

/cc https:/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()