fixed dtc modules drift (#465)

Author: Chaithra001

infoblox/resource_infoblox_dtc_lbdn.go

@@ -302,12 +302,12 @@ func resourceDtcLbdnGet(d *schema.ResourceData, m interface{}) error {
 
 	rec, err := searchObjectByRefOrInternalId("DtcLbdn", d, m)
 	if err != nil {
-		if _, ok := err.(*ibclient.NotFoundError); !ok {
-			return ibclient.NewNotFoundError(fmt.Sprintf(
-				"cannot find appropriate object on NIOS side for resource with ID '%s': %s;", d.Id(), err))
-		} else {
+		if _, ok := err.(*ibclient.NotFoundError); ok {
 			d.SetId("")
 			return nil
+		} else {
+			return ibclient.NewNotFoundError(fmt.Sprintf(
+				"cannot find appropriate object on NIOS side for resource with ID '%s': %s;", d.Id(), err))
 		}
 	}
 
@@ -429,6 +429,7 @@ func resourceDtcLbdnGet(d *schema.ResourceData, m interface{}) error {
 		return err
 	}
 
+	d.SetId(dtcLbdn.Ref)
 	return nil
 }
 

infoblox/resource_infoblox_dtc_pool.go

@@ -510,12 +510,12 @@ func resourceDtcPoolGet(d *schema.ResourceData, m interface{}) error {
 	connector := m.(ibclient.IBConnector)
 	rec, err := searchObjectByRefOrInternalId("DtcPool", d, m)
 	if err != nil {
-		if _, ok := err.(*ibclient.NotFoundError); !ok {
-			return ibclient.NewNotFoundError(fmt.Sprintf(
-				"cannot find appropriate object on NIOS side for resource with ID '%s': %s;", d.Id(), err))
-		} else {
+		if _, ok := err.(*ibclient.NotFoundError); ok {
 			d.SetId("")
 			return nil
+		} else {
+			return ibclient.NewNotFoundError(fmt.Sprintf(
+				"cannot find appropriate object on NIOS side for resource with ID '%s': %s;", d.Id(), err))
 		}
 	}
 	var dtcPool *ibclient.DtcPool
@@ -643,6 +643,10 @@ func resourceDtcPoolGet(d *schema.ResourceData, m interface{}) error {
 		}
 	}
 
+	if err = d.Set("ref", dtcPool.Ref); err != nil {
+		return err
+	}
+	d.SetId(dtcPool.Ref)
 	return nil
 }
 

infoblox/resource_infoblox_dtc_server.go

@@ -189,12 +189,12 @@ func resourceDtcServerGet(d *schema.ResourceData, m interface{}) error {
 	connector := m.(ibclient.IBConnector)
 	rec, err := searchObjectByRefOrInternalId("DtcServer", d, m)
 	if err != nil {
-		if _, ok := err.(*ibclient.NotFoundError); !ok {
-			return ibclient.NewNotFoundError(fmt.Sprintf(
-				"cannot find appropriate object on NIOS side for resource with ID '%s': %s;", d.Id(), err))
-		} else {
+		if _, ok := err.(*ibclient.NotFoundError); ok {
 			d.SetId("")
 			return nil
+		} else {
+			return ibclient.NewNotFoundError(fmt.Sprintf(
+				"cannot find appropriate object on NIOS side for resource with ID '%s': %s;", d.Id(), err))
 		}
 	}
 	var dtcServer *ibclient.DtcServer
@@ -244,6 +244,10 @@ func resourceDtcServerGet(d *schema.ResourceData, m interface{}) error {
 	if err = d.Set("use_sni_hostname", dtcServer.UseSniHostname); err != nil {
 		return err
 	}
+	if err = d.Set("ref", dtcServer.Ref); err != nil {
+		return err
+	}
+	d.SetId(dtcServer.Ref)
 	return nil
 }