/digitalocean/datasource_digitalocean_loadbalancer_test.go

https://github.com/terraform-providers/terraform-provider-digitalocean · Go · 185 lines · 167 code · 18 blank · 0 comment · 7 complexity · c0cca376dfcf5f5a6c42db355bf7028f MD5 · raw file 

    

  1. package digitalocean
    2. 

  2. 

  3. import (
    4. 

  4. 	"context"
    5. 

  5. 	"fmt"
    6. 

  6. 	"regexp"
    7. 

  7. 	"testing"
    8. 

  8. 

  9. 	"github.com/digitalocean/godo"
    10. 

  10. 	"github.com/hashicorp/terraform-plugin-sdk/helper/acctest"
    11. 

  11. 	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
    12. 

  12. 	"github.com/hashicorp/terraform-plugin-sdk/terraform"
    13. 

  13. )
    14. 

  14. 

  15. func TestAccDataSourceDigitalOceanLoadBalancer_Basic(t *testing.T) {
    16. 

  16. 	t.Parallel()
    17. 

  17. 	var loadbalancer godo.LoadBalancer
    18. 

  18. 	rInt := acctest.RandInt()
    19. 

  19. 

  20. 	expectedURNRegEx, _ := regexp.Compile(`do:loadbalancer:[0-9a-fA-F]{8}\-[0-9a-fA-F]{4}\-[0-9a-fA-F]{4}\-[0-9a-fA-F]{4}\-[0-9a-fA-F]{12}`)
    21. 

  21. 

  22. 	resource.Test(t, resource.TestCase{
    23. 

  23. 		PreCheck:  func() { testAccPreCheck(t) },
    24. 

  24. 		Providers: testAccProviders,
    25. 

  25. 		Steps: []resource.TestStep{
    26. 

  26. 			{
    27. 

  27. 				Config: testAccCheckDataSourceDigitalOceanLoadBalancerConfig_basic(rInt),
    28. 

  28. 				Check: resource.ComposeTestCheckFunc(
    29. 

  29. 					testAccCheckDataSourceDigitalOceanLoadBalancerExists("data.digitalocean_loadbalancer.foobar", &loadbalancer),
    30. 

  30. 					resource.TestCheckResourceAttr(
    31. 

  31. 						"data.digitalocean_loadbalancer.foobar", "name", fmt.Sprintf("loadbalancer-%d", rInt)),
    32. 

  32. 					resource.TestCheckResourceAttr(
    33. 

  33. 						"data.digitalocean_loadbalancer.foobar", "region", "nyc3"),
    34. 

  34. 					resource.TestCheckResourceAttr(
    35. 

  35. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.#", "1"),
    36. 

  36. 					resource.TestCheckResourceAttr(
    37. 

  37. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.192790336.entry_port", "80"),
    38. 

  38. 					resource.TestCheckResourceAttr(
    39. 

  39. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.192790336.entry_protocol", "http"),
    40. 

  40. 					resource.TestCheckResourceAttr(
    41. 

  41. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.192790336.target_port", "80"),
    42. 

  42. 					resource.TestCheckResourceAttr(
    43. 

  43. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.192790336.target_protocol", "http"),
    44. 

  44. 					resource.TestCheckResourceAttr(
    45. 

  45. 						"data.digitalocean_loadbalancer.foobar", "healthcheck.#", "1"),
    46. 

  46. 					resource.TestCheckResourceAttr(
    47. 

  47. 						"data.digitalocean_loadbalancer.foobar", "healthcheck.0.port", "22"),
    48. 

  48. 					resource.TestCheckResourceAttr(
    49. 

  49. 						"data.digitalocean_loadbalancer.foobar", "healthcheck.0.protocol", "tcp"),
    50. 

  50. 					resource.TestCheckResourceAttr(
    51. 

  51. 						"data.digitalocean_loadbalancer.foobar", "droplet_ids.#", "2"),
    52. 

  52. 					resource.TestMatchResourceAttr(
    53. 

  53. 						"data.digitalocean_loadbalancer.foobar", "urn", expectedURNRegEx),
    54. 

  54. 					resource.TestCheckResourceAttrSet(
    55. 

  55. 						"data.digitalocean_loadbalancer.foobar", "vpc_uuid"),
    56. 

  56. 					resource.TestCheckResourceAttr(
    57. 

  57. 						"data.digitalocean_loadbalancer.foobar", "enable_proxy_protocol", "false"),
    58. 

  58. 					resource.TestCheckResourceAttr(
    59. 

  59. 						"data.digitalocean_loadbalancer.foobar", "enable_backend_keepalive", "false"),
    60. 

  60. 				),
    61. 

  61. 			},
    62. 

  62. 		},
    63. 

  63. 	})
    64. 

  64. }
    65. 

  65. 

  66. func TestAccDataSourceDigitalOceanLoadBalancer_multipleRules(t *testing.T) {
    67. 

  67. 	t.Parallel()
    68. 

  68. 	var loadbalancer godo.LoadBalancer
    69. 

  69. 	rName := randomTestName()
    70. 

  70. 

  71. 	resource.Test(t, resource.TestCase{
    72. 

  72. 		PreCheck:     func() { testAccPreCheck(t) },
    73. 

  73. 		Providers:    testAccProviders,
    74. 

  74. 		CheckDestroy: testAccCheckDigitalOceanLoadbalancerDestroy,
    75. 

  75. 		Steps: []resource.TestStep{
    76. 

  76. 			{
    77. 

  77. 				Config: testAccCheckDigitalOceanLoadbalancerConfig_multipleRules(rName) + testAccCheckDataSourceDigitalOceanLoadBalancerConfig_multipleRules,
    78. 

  78. 				Check: resource.ComposeAggregateTestCheckFunc(
    79. 

  79. 					testAccCheckDigitalOceanLoadbalancerExists("digitalocean_loadbalancer.foobar", &loadbalancer),
    80. 

  80. 					resource.TestCheckResourceAttr(
    81. 

  81. 						"data.digitalocean_loadbalancer.foobar", "name", rName),
    82. 

  82. 					resource.TestCheckResourceAttr(
    83. 

  83. 						"data.digitalocean_loadbalancer.foobar", "region", "nyc3"),
    84. 

  84. 					resource.TestCheckResourceAttr(
    85. 

  85. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.#", "2"),
    86. 

  86. 					resource.TestCheckResourceAttr(
    87. 

  87. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.236988772.entry_port", "443"),
    88. 

  88. 					resource.TestCheckResourceAttr(
    89. 

  89. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.236988772.entry_protocol", "https"),
    90. 

  90. 					resource.TestCheckResourceAttr(
    91. 

  91. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.236988772.target_port", "443"),
    92. 

  92. 					resource.TestCheckResourceAttr(
    93. 

  93. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.236988772.target_protocol", "https"),
    94. 

  94. 					resource.TestCheckResourceAttr(
    95. 

  95. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.192790336.entry_port", "80"),
    96. 

  96. 					resource.TestCheckResourceAttr(
    97. 

  97. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.192790336.entry_protocol", "http"),
    98. 

  98. 					resource.TestCheckResourceAttr(
    99. 

  99. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.192790336.target_port", "80"),
    100. 

  100. 					resource.TestCheckResourceAttr(
    101. 

  101. 						"data.digitalocean_loadbalancer.foobar", "forwarding_rule.192790336.target_protocol", "http"),
    102. 

  102. 				),
    103. 

  103. 			},
    104. 

  104. 		},
    105. 

  105. 	})
    106. 

  106. }
    107. 

  107. 

  108. func testAccCheckDataSourceDigitalOceanLoadBalancerExists(n string, loadbalancer *godo.LoadBalancer) resource.TestCheckFunc {
    109. 

  109. 	return func(s *terraform.State) error {
    110. 

  110. 		rs, ok := s.RootModule().Resources[n]
    111. 

  111. 

  112. 		if !ok {
    113. 

  113. 			return fmt.Errorf("Not found: %s", n)
    114. 

  114. 		}
    115. 

  115. 

  116. 		if rs.Primary.ID == "" {
    117. 

  117. 			return fmt.Errorf("No Load Balancer ID is set")
    118. 

  118. 		}
    119. 

  119. 

  120. 		client := testAccProvider.Meta().(*CombinedConfig).godoClient()
    121. 

  121. 

  122. 		foundLoadbalancer, _, err := client.LoadBalancers.Get(context.Background(), rs.Primary.ID)
    123. 

  123. 

  124. 		if err != nil {
    125. 

  125. 			return err
    126. 

  126. 		}
    127. 

  127. 

  128. 		if foundLoadbalancer.ID != rs.Primary.ID {
    129. 

  129. 			return fmt.Errorf("Load Balancer not found")
    130. 

  130. 		}
    131. 

  131. 

  132. 		*loadbalancer = *foundLoadbalancer
    133. 

  133. 

  134. 		return nil
    135. 

  135. 	}
    136. 

  136. }
    137. 

  137. 

  138. func testAccCheckDataSourceDigitalOceanLoadBalancerConfig_basic(rInt int) string {
    139. 

  139. 	return fmt.Sprintf(`
    140. 

  140. resource "digitalocean_tag" "foo" {
    141. 

  141.   name = "web"
    142. 

  142. }
    143. 

    144. 

  144. resource "digitalocean_droplet" "foo" {
    145. 

  145.   count              = 2
    146. 

  146.   image              = "ubuntu-18-04-x64"
    147. 

  147.   name               = "foo-%d-${count.index}"
    148. 

  148.   region             = "nyc3"
    149. 

  149.   size               = "512mb"
    150. 

  150.   private_networking = true
    151. 

  151.   tags               = [digitalocean_tag.foo.id]
    152. 

  152. }
    153. 

    154. 

  154. resource "digitalocean_loadbalancer" "foo" {
    155. 

  155.   name   = "loadbalancer-%d"
    156. 

  156.   region = "nyc3"
    157. 

    158. 

  158.   forwarding_rule {
    159. 

  159. 	entry_port     = 80
    160. 

  160. 	entry_protocol = "http"
    161. 

    162. 

  162. 	target_port     = 80
    163. 

  163. 	target_protocol = "http"
    164. 

  164.   }
    165. 

    166. 

  166.   healthcheck {
    167. 

  167. 	port     = 22
    168. 

  168.     protocol = "tcp"
    169. 

  169.   }
    170. 

    171. 

  171.   droplet_tag = digitalocean_tag.foo.id
    172. 

  172.   depends_on  = ["digitalocean_droplet.foo"]
    173. 

  173. }
    174. 

    175. 

  175. data "digitalocean_loadbalancer" "foobar" {
    176. 

  176.   name = digitalocean_loadbalancer.foo.name
    177. 

  177. }`, rInt, rInt)
    178. 

  178. }
    179. 

  179. 

  180. const (
    181. 

  181. 	testAccCheckDataSourceDigitalOceanLoadBalancerConfig_multipleRules = `
    182. 

  182. data "digitalocean_loadbalancer" "foobar" {
    183. 

  183.   name = digitalocean_loadbalancer.foobar.name
    184. 

  184. }`
    185. 

  185. )
    186.