forked from phillbaker/terraform-provider-elasticsearch
-
Notifications
You must be signed in to change notification settings - Fork 1
/
resource_elasticsearch_snapshot_repository_test.go
105 lines (88 loc) · 2.63 KB
/
resource_elasticsearch_snapshot_repository_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
package main
import (
"context"
"fmt"
"testing"
elastic7 "github.com/olivere/elastic/v7"
elastic5 "gopkg.in/olivere/elastic.v5"
elastic6 "gopkg.in/olivere/elastic.v6"
"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
"github.com/hashicorp/terraform-plugin-sdk/terraform"
)
func TestAccElasticsearchSnapshotRepository(t *testing.T) {
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testCheckElasticsearchSnapshotRepositoryDestroy,
Steps: []resource.TestStep{
resource.TestStep{
Config: testAccElasticsearchSnapshotRepository,
Check: resource.ComposeTestCheckFunc(
testCheckElasticsearchSnapshotRepositoryExists("elasticsearch_snapshot_repository.test"),
),
},
},
})
}
func testCheckElasticsearchSnapshotRepositoryExists(name string) resource.TestCheckFunc {
return func(s *terraform.State) error {
rs, ok := s.RootModule().Resources[name]
if !ok {
return fmt.Errorf("Not found: %s", name)
}
if rs.Primary.ID == "" {
return fmt.Errorf("No snapshot repository ID is set")
}
meta := testAccProvider.Meta()
var err error
switch meta.(type) {
case *elastic7.Client:
client := meta.(*elastic7.Client)
_, err = client.SnapshotGetRepository(rs.Primary.ID).Do(context.TODO())
case *elastic6.Client:
client := meta.(*elastic6.Client)
_, err = client.SnapshotGetRepository(rs.Primary.ID).Do(context.TODO())
default:
client := meta.(*elastic5.Client)
_, err = client.SnapshotGetRepository(rs.Primary.ID).Do(context.TODO())
}
if err != nil {
return err
}
return nil
}
}
func testCheckElasticsearchSnapshotRepositoryDestroy(s *terraform.State) error {
for _, rs := range s.RootModule().Resources {
if rs.Type != "elasticsearch_snapshot_repository" {
continue
}
meta := testAccProvider.Meta()
var err error
switch meta.(type) {
case *elastic7.Client:
client := meta.(*elastic7.Client)
_, err = client.SnapshotGetRepository(rs.Primary.ID).Do(context.TODO())
case *elastic6.Client:
client := meta.(*elastic6.Client)
_, err = client.SnapshotGetRepository(rs.Primary.ID).Do(context.TODO())
default:
client := meta.(*elastic5.Client)
_, err = client.SnapshotGetRepository(rs.Primary.ID).Do(context.TODO())
}
if err != nil {
return nil // should be not found error
}
return fmt.Errorf("Snapshot repository %q still exists", rs.Primary.ID)
}
return nil
}
var testAccElasticsearchSnapshotRepository = `
resource "elasticsearch_snapshot_repository" "test" {
name = "terraform-test"
type = "fs"
settings = {
location = "/tmp/elasticsearch"
}
}
`