Fix crash when user data source id or email not found

Author: emichaeli

sumologic/data_source_sumologic_user.go

@@ -59,6 +59,9 @@ func dataSourceSumologicUserRead(d *schema.ResourceData, meta interface{}) error
 		if err != nil {
 			return fmt.Errorf("user with id %v not found: %v", id, err)
 		}
+		if user == nil {
+			return fmt.Errorf("user with id %v not found", id)
+		}
 	} else {
 		if userEmail, ok := d.GetOk("email"); ok {
 			email := userEmail.(string)
@@ -102,6 +105,9 @@ func (s *Client) GetUserByEmail(email string) (*User, error) {
 	if err != nil {
 		return nil, err
 	}
+	if len(response.User) == 0 {
+		return nil, fmt.Errorf("user with email address '%s' does not exist", email)
+	}
 
 	return &response.User[0], nil
 }

sumologic/data_source_sumologic_user_test.go

@@ -2,12 +2,13 @@ package sumologic
 
 import (
 	"fmt"
+	"regexp"
 	"testing"
 
 	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
 )
 
-func TestAccDataSourcSumologicUser_basic(t *testing.T) {
+func TestAccDataSourceSumologicUser_basic(t *testing.T) {
 	resource.Test(t, resource.TestCase{
 		PreCheck:  func() { testAccPreCheck(t) },
 		Providers: testAccProviders,
@@ -34,6 +35,40 @@ func testAccDataSourceUserCheck(email, reference string) resource.TestCheckFunc
 	)
 }
 
+func TestAccDataSourceSumologicUser_user_email_doesnt_exist(t *testing.T) {
+	userDoestExistConfig := `
+  data "sumologic_user" "user_email_doesnt_exist" {
+    email = "[email protected]"
+  }`
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config:      userDoestExistConfig,
+				ExpectError: regexp.MustCompile("user with email address '[email protected]' does not exist"),
+			},
+		},
+	})
+}
+
+func TestAccDataSourceSumologicUser_user_id_doesnt_exist(t *testing.T) {
+	userDoestExistConfig := `
+  data "sumologic_user" "user_id_doesnt_exist" {
+    id = 99999999999999
+  }`
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config:      userDoestExistConfig,
+				ExpectError: regexp.MustCompile("user with id 99999999999999 not found"),
+			},
+		},
+	})
+}
+
 var testDataSourceAccSumologicUserConfig = fmt.Sprintf(`
 resource "sumologic_user" "test_user" {
   first_name = "Test"