security_group.go

package helpers

import (
	"encoding/json"
	"io/ioutil"
	"os"
	"path/filepath"

	. "github.com/onsi/gomega"
	. "github.com/onsi/gomega/gexec"
)

type SecurityGroup struct {
	Name        string `json:"-"`
	Protocol    string `json:"protocol"`
	Destination string `json:"destination"`
	Ports       string `json:"ports"`
	Description string `json:"description"`
}

func NewSecurityGroup(name string, protocol string, destination string, ports string, description string) SecurityGroup {
	return SecurityGroup{
		Name:        name,
		Protocol:    protocol,
		Destination: destination,
		Ports:       ports,
		Description: description,
	}
}

func (s SecurityGroup) Create() {
	dir, err := ioutil.TempDir("", "simple-security-group")
	Expect(err).ToNot(HaveOccurred())
	defer os.RemoveAll(dir)

	tempfile := filepath.Join(dir, "security-group.json")

	securityGroup, err := json.Marshal([]SecurityGroup{s})
	Expect(err).ToNot(HaveOccurred())

	err = ioutil.WriteFile(tempfile, securityGroup, 0666)
	Eventually(CF("create-security-group", s.Name, tempfile)).Should(Exit(0))
}

func (s SecurityGroup) Delete() {
	Eventually(CF("delete-security-group", s.Name, "-f")).Should(Exit(0))
}