From b0b8267cede57834f965914148e0ef6923636670 Mon Sep 17 00:00:00 2001
From: Christopher Collins <ccollin4@cisco.com>
Date: Wed, 2 Nov 2022 11:23:32 -0700
Subject: [PATCH] Support for epg subnet in gbp client.

Change-Id: Ibdd1274559ef3680a6d5866ed6ec905c4bc44d92
---
 gbpclient/gbp/v2_0/subnet.py        | 13 +++++++++++++
 gbpclient/tests/unit/test_subnet.py |  4 ++++
 2 files changed, 17 insertions(+)

diff --git a/gbpclient/gbp/v2_0/subnet.py b/gbpclient/gbp/v2_0/subnet.py
index 181e54d..3649411 100644
--- a/gbpclient/gbp/v2_0/subnet.py
+++ b/gbpclient/gbp/v2_0/subnet.py
@@ -42,6 +42,9 @@ def _get_attrs_subnet_extension(client_manager, parsed_args, is_create=True):
         attrs['apic:snat_subnet_only'] = True
     if parsed_args.apic_snat_subnet_only_disable:
         attrs['apic:snat_subnet_only'] = False
+    if ('apic_epg_subnet' in parsed_args and
+        parsed_args.apic_epg_subnet):
+        attrs['apic:epg_subnet'] = True
 
     return attrs
 
@@ -58,6 +61,8 @@ subnet_sdk.Subnet.apic_active_active_aap = resource.Body(
     'apic:active_active_aap')
 subnet_sdk.Subnet.apic_snat_subnet_only = resource.Body(
     'apic:snat_subnet_only')
+subnet_sdk.Subnet.apic_epg_subnet = resource.Body(
+    'apic:epg_subnet')
 
 
 class CreateSubnetExtension(hooks.CommandHook):
@@ -108,6 +113,14 @@ class CreateSubnetExtension(hooks.CommandHook):
             help=_("Set APIC snat subnet only to false\n"
                    "Default value for apic_snat_subnet_only is False ")
         )
+        parser.add_argument(
+            '--apic-epg-subnet',
+            action='store_true',
+            default=False,
+            dest='apic_epg_subnet',
+            help=_("Set APIC epg subnet to true\n"
+                   "Default value for apic_epg_subnet is False ")
+        )
         return parser
 
     def get_epilog(self):
diff --git a/gbpclient/tests/unit/test_subnet.py b/gbpclient/tests/unit/test_subnet.py
index 17f6ffc..a4facba 100644
--- a/gbpclient/tests/unit/test_subnet.py
+++ b/gbpclient/tests/unit/test_subnet.py
@@ -52,6 +52,7 @@ class TestSubnetCreate(test_subnet.TestSubnet, test_cli20.CLITestV20Base):
             ('apic_snat_host_pool_enable', None),
             ('apic_active_active_aap_enable', None),
             ('apic_snat_subnet_only_enable', None),
+            ('apic_epg_subnet', False)
         ]
         create_ext = subnet_ext.CreateSubnetExtension(self.app)
         parsed_args = self.check_parser_ext(
@@ -73,6 +74,7 @@ class TestSubnetCreate(test_subnet.TestSubnet, test_cli20.CLITestV20Base):
             "--apic-snat-host-pool-enable",
             "--apic-active-active-aap-enable",
             "--apic-snat-subnet-only-enable",
+            "--apic-epg-subnet"
         ]
         verifylist = [
             ('name', self._subnet.name),
@@ -80,6 +82,7 @@ class TestSubnetCreate(test_subnet.TestSubnet, test_cli20.CLITestV20Base):
             ('apic_snat_host_pool_enable', True),
             ('apic_active_active_aap_enable', True),
             ('apic_snat_subnet_only_enable', True),
+            ('apic_epg_subnet', True)
         ]
         create_ext = subnet_ext.CreateSubnetExtension(self.app)
         parsed_args = self.check_parser_ext(
@@ -94,6 +97,7 @@ class TestSubnetCreate(test_subnet.TestSubnet, test_cli20.CLITestV20Base):
             'apic:active_active_aap': True,
             'apic:snat_host_pool': True,
             'apic:snat_subnet_only': True,
+            'apic:epg_subnet': True
         })