add allow and disallow functions back and fix tests

axelarnetwork · Dec 12, 2024 · fde9d16 · fde9d16
1 parent cfc6b27
commit fde9d16
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/move/axelar_gateway/sources/gateway.move b/move/axelar_gateway/sources/gateway.move
@@ -1091,7 +1091,7 @@ fun test_allow_function() {
     let ctx = &mut sui::tx_context::dummy();
     let mut self = dummy(ctx);
     let owner_cap = owner_cap::create(ctx);
-    let version = 0;
+    let version = VERSION;
     let function_name = b"function_name".to_ascii_string();
 
     self.allow_function(&owner_cap, version, function_name);
@@ -1105,7 +1105,7 @@ fun test_disallow_function() {
     let ctx = &mut sui::tx_context::dummy();
     let mut self = dummy(ctx);
     let owner_cap = owner_cap::create(ctx);
-    let version = 0;
+    let version = VERSION;
     let function_name = b"approve_messages".to_ascii_string();
 
     self.disallow_function(&owner_cap, version, function_name);

diff --git a/move/axelar_gateway/sources/versioned/gateway_v1.move b/move/axelar_gateway/sources/versioned/gateway_v1.move
@@ -221,6 +221,22 @@ public(package) fun send_message(
     );
 }
 
+public(package) fun allow_function(
+    self: &mut Gateway_v1,
+    version: u64,
+    function_name: String,
+) {
+    self.version_control.allow_function(version, function_name);
+}
+
+public(package) fun disallow_function(
+    self: &mut Gateway_v1,
+    version: u64,
+    function_name: String,
+) {
+    self.version_control.disallow_function(version, function_name);
+}
+
 // -----------------
 // Private Functions
 // -----------------