binapigen: added enumflags type

Change-Id: I2f46504bd05862e415dab518fad349d08aedf919
Signed-off-by: Vladimir Lavor <vlavor@cisco.com>

1 files changed, 16 insertions, 1 deletions

diff --git a/binapigen/generate_test.go b/binapigen/generate_test.go
index 2fa5dc6..b1d4d70 100644
--- a/binapigen/generate_test.go
+++ b/binapigen/generate_test.go
@@ -47,7 +47,7 @@ func GenerateFromFile(file string, opts Options) error {
 	return nil
 }
 
-func TestGenerateFromFile(t *testing.T) {
+func TestGenerateFromFileACL(t *testing.T) {
 	RegisterTestingT(t)
 
 	// remove directory created during test
@@ -62,6 +62,21 @@ func TestGenerateFromFile(t *testing.T) {
 	Expect(fileInfo.Name()).To(BeEquivalentTo("acl.ba.go"))
 }
 
+func TestGenerateFromFileIP(t *testing.T) {
+	RegisterTestingT(t)
+
+	// remove directory created during test
+	defer os.RemoveAll(testOutputDir)
+
+	opts := Options{OutputDir: testOutputDir}
+	err := GenerateFromFile("vppapi/testdata/ip.api.json", opts)
+	Expect(err).ShouldNot(HaveOccurred())
+	fileInfo, err := os.Stat(testOutputDir + "/ip/ip.ba.go")
+	Expect(err).ShouldNot(HaveOccurred())
+	Expect(fileInfo.IsDir()).To(BeFalse())
+	Expect(fileInfo.Name()).To(BeEquivalentTo("ip.ba.go"))
+}
+
 func TestGenerateFromFileInputError(t *testing.T) {
 	RegisterTestingT(t)