@@ -22,7 +22,7 @@ func TestAccSumologicCollector_basic(t *testing.T) {
2222 Config : testAccSumologicCollectorConfigBasic (rname ),
2323 Check : resource .ComposeTestCheckFunc (
2424 testAccCheckCollectorExists (resourceName , & collector ),
25- testAccCheckCollectorValues (& collector , rname , "" , "" , "Etc/UTC" ),
25+ testAccCheckCollectorValues (& collector , rname , "" , "" , "Etc/UTC" , "" ),
2626 resource .TestCheckResourceAttrSet (resourceName , "id" ),
2727 resource .TestCheckResourceAttr (resourceName , "name" , rname ),
2828 resource .TestCheckResourceAttr (resourceName , "description" , "" ),
@@ -56,7 +56,7 @@ func TestAccSumologicCollector_create(t *testing.T) {
5656 // query the API to retrieve the collector
5757 testAccCheckCollectorExists (resourceName , & collector ),
5858 // verify remote values
59- testAccCheckCollectorValues (& collector , rname , rdescription , rcategory , "Etc/UTC" ),
59+ testAccCheckCollectorValues (& collector , rname , rdescription , rcategory , "Etc/UTC" , "" ),
6060 // verify local values
6161 resource .TestCheckResourceAttrSet (resourceName , "id" ),
6262 resource .TestCheckResourceAttr (resourceName , "name" , rname ),
@@ -83,7 +83,7 @@ func TestAccSumologicCollector_update(t *testing.T) {
8383 Config : testAccSumologicCollectorConfig (rname , rdescription , rcategory ),
8484 Check : resource .ComposeTestCheckFunc (
8585 testAccCheckCollectorExists (resourceName , & collector ),
86- testAccCheckCollectorValues (& collector , rname , rdescription , rcategory , "Etc/UTC" ),
86+ testAccCheckCollectorValues (& collector , rname , rdescription , rcategory , "Etc/UTC" , "" ),
8787 resource .TestCheckResourceAttrSet (resourceName , "id" ),
8888 resource .TestCheckResourceAttr (resourceName , "name" , rname ),
8989 resource .TestCheckResourceAttr (resourceName , "description" , rdescription ),
@@ -95,7 +95,7 @@ func TestAccSumologicCollector_update(t *testing.T) {
9595 Config : testAccSumologicCollectorConfigUpdate (rname , rdescription , rcategory ),
9696 Check : resource .ComposeTestCheckFunc (
9797 testAccCheckCollectorExists (resourceName , & collector ),
98- testAccCheckCollectorValues (& collector , rname , rdescription , rcategory , "Europe/Berlin" ),
98+ testAccCheckCollectorValues (& collector , rname , rdescription , rcategory , "Europe/Berlin" , "" ),
9999 resource .TestCheckResourceAttrSet (resourceName , "id" ),
100100 resource .TestCheckResourceAttr (resourceName , "name" , rname ),
101101 resource .TestCheckResourceAttr (resourceName , "description" , rdescription ),
@@ -161,19 +161,24 @@ func testAccCheckCollectorExists(n string, collector *Collector) resource.TestCh
161161 }
162162}
163163
164- func testAccCheckCollectorValues (collector * Collector , name , description , category , timezone string ) resource.TestCheckFunc {
164+ func testAccCheckCollectorValues (collector * Collector , name , description , category , timezone , budgetValue string ) resource.TestCheckFunc {
165165 return func (s * terraform.State ) error {
166166 if collector .Name != name {
167167 return fmt .Errorf ("bad name, expected \" %s\" , got: %#v" , name , collector .Name )
168168 }
169169 if collector .Description != description {
170- return fmt .Errorf ("bad name , expected \" %s\" , got: %#v" , description , collector .Description )
170+ return fmt .Errorf ("bad description , expected \" %s\" , got: %#v" , description , collector .Description )
171171 }
172172 if collector .Category != category {
173- return fmt .Errorf ("bad name , expected \" %s\" , got: %#v" , category , collector .Category )
173+ return fmt .Errorf ("bad category , expected \" %s\" , got: %#v" , category , collector .Category )
174174 }
175175 if collector .TimeZone != timezone {
176- return fmt .Errorf ("bad name, expected \" %s\" , got: %#v" , timezone , collector .TimeZone )
176+ return fmt .Errorf ("bad timezone, expected \" %s\" , got: %#v" , timezone , collector .TimeZone )
177+ }
178+ if value , ok := collector .Fields ["_budget" ]; ok {
179+ if value != budgetValue {
180+ return fmt .Errorf ("bad budgetValue, expected \" %s\" , got: %#v" , budgetValue , collector .Fields ["_budget" ])
181+ }
177182 }
178183 return nil
179184 }
0 commit comments