diff --git a/test/codegen-generated-classes-test/src/test/java/software/amazon/awssdk/services/compiledendpointrules/endpoints/internal/RuleUrlTest.java b/test/codegen-generated-classes-test/src/test/java/software/amazon/awssdk/services/compiledendpointrules/endpoints/internal/RuleUrlTest.java
index 6133cf7c65c..c05c4e14b7c 100644
--- a/test/codegen-generated-classes-test/src/test/java/software/amazon/awssdk/services/compiledendpointrules/endpoints/internal/RuleUrlTest.java
+++ b/test/codegen-generated-classes-test/src/test/java/software/amazon/awssdk/services/compiledendpointrules/endpoints/internal/RuleUrlTest.java
@@ -56,6 +56,11 @@ public static Collection<TestIsIpCase> testIsIpCases() {
             , new TestIsIpCase("1.1.010.1", false)
             , new TestIsIpCase("1.1.1.010", false)
 
+            , new TestIsIpCase("001.1.1.1", false)
+            , new TestIsIpCase("1.001.1.1", false)
+            , new TestIsIpCase("1.1.001.1", false)
+            , new TestIsIpCase("1.1.1.001", false)
+
             , new TestIsIpCase("01.1.1.1", false)
             , new TestIsIpCase("1.01.1.1", false)
             , new TestIsIpCase("1.1.01.1", false)
@@ -79,6 +84,11 @@ public static Collection<TestIsIpCase> testIsIpCases() {
             , new TestIsIpCase("127.0.1mm.0.1", false)
             , new TestIsIpCase("127.0.0.1mm.1", false)
 
+            // Less than 4 segments
+            , new TestIsIpCase("127.0.0", false)
+            , new TestIsIpCase("127.0", false)
+            , new TestIsIpCase("127", false)
+
             // name
             , new TestIsIpCase("amazon.com", false)
             , new TestIsIpCase("localhost", false)