From 142e976c4087022be924c496883edc183cdfc3d8 Mon Sep 17 00:00:00 2001
From: boukeversteegh <info@boukeversteegh.nl>
Date: Mon, 25 May 2020 21:56:03 +0200
Subject: [PATCH] Add extra related test cases for #11

---
 .../casing_message_field_uppercase.json                     | 4 +++-
 .../casing_message_field_uppercase.proto                    | 4 +++-
 .../casing_message_field_uppercase.py                       | 6 ++++++
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.json b/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.json
index 4c62045..83bd111 100644
--- a/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.json
+++ b/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.json
@@ -1,3 +1,5 @@
 {
-  "UPPERCASE": 10
+  "UPPERCASE": 10,
+  "UPPERCASE_V2": 10,
+  "UPPER_CAMEL_CASE": 10
 }
diff --git a/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.proto b/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.proto
index 5e75bbd..9964dfa 100644
--- a/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.proto
+++ b/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.proto
@@ -1,5 +1,7 @@
 syntax = "proto3";
 
 message Test {
-  int32 UPPER_CAMEL_CASE = 1;
+  int32 UPPERCASE = 1;
+  int32 UPPERCASE_V2 = 2;
+  int32 UPPER_CAMEL_CASE = 3;
 }
\ No newline at end of file
diff --git a/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.py b/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.py
index cf9119d..d77119e 100644
--- a/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.py
+++ b/betterproto/tests/inputs/casing_message_field_uppercase/casing_message_field_uppercase.py
@@ -5,6 +5,12 @@ from betterproto.tests.output_betterproto.casing_message_field_uppercase.casing_
 
 def test_message_casing():
     message = Test()
+    assert hasattr(
+        message, "uppercase"
+    ), "UPPERCASE attribute is converted to 'uppercase' in python"
+    assert hasattr(
+        message, "uppercase_v2"
+    ), "UPPERCASE_V2 attribute is converted to 'uppercase_v2' in python"
     assert hasattr(
         message, "upper_camel_case"
     ), "UPPER_CAMEL_CASE attribute is converted to upper_camel_case in python"