Merge remote-tracking branch 'ryandeivert/ryandeivert-add-flow-log-s3…

…-dest' into coveo-master

aws/ec2_helpers.go

@@ -0,0 +1,18 @@
+package aws
+
+import (
+	"regexp"
+)
+
+var (
+	ec2ResourceIdRegexp = regexp.MustCompile(`\A([a-z]+)-([a-z0-9]+)\z`)
+)
+
+// parseEc2ResourceId parses an EC2 resource ID into prefix and unique part.
+func parseEc2ResourceId(s string) (string, string) {
+	matches := ec2ResourceIdRegexp.FindStringSubmatch(s)
+	if matches == nil {
+		return "", ""
+	}
+	return matches[1], matches[2]
+}

aws/ec2_helpers_test.go

@@ -0,0 +1,45 @@
+package aws
+
+import (
+	"testing"
+)
+
+func TestParseEc2ResourceId(t *testing.T) {
+	type TestCase struct {
+		ResourceId     string
+		ExpectedPrefix string
+		ExpectedUnique string
+	}
+	testCases := []TestCase{
+		{
+			ResourceId:     "",
+			ExpectedPrefix: "",
+			ExpectedUnique: "",
+		},
+		{
+			ResourceId:     "vpc12345678",
+			ExpectedPrefix: "",
+			ExpectedUnique: "",
+		},
+		{
+			ResourceId:     "vpc-12345678",
+			ExpectedPrefix: "vpc",
+			ExpectedUnique: "12345678",
+		},
+		{
+			ResourceId:     "i-12345678abcdef00",
+			ExpectedPrefix: "i",
+			ExpectedUnique: "12345678abcdef00",
+		},
+	}
+
+	for i, testCase := range testCases {
+		resultPrefix, resultUnique := parseEc2ResourceId(testCase.ResourceId)
+		if resultPrefix != testCase.ExpectedPrefix || resultUnique != testCase.ExpectedUnique {
+			t.Errorf(
+				"test case %d: got (%s, %s), but want (%s, %s)",
+				i, resultPrefix, resultUnique, testCase.ExpectedPrefix, testCase.ExpectedUnique,
+			)
+		}
+	}
+}