Merge pull request #224 from stephanerobert/commit_transaction

Commit transaction in Arista
internap · Oct 25, 2018 · 26345c1 · 26345c1
2 parents 1a27b29 + 3cb3108
commit 26345c1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/netman/adapters/switches/arista.py b/netman/adapters/switches/arista.py
@@ -40,6 +40,9 @@ def _end_transaction(self):
     def _start_transaction(self):
         pass
 
+    def commit_transaction(self):
+        self.node.enable('write memory')
+
     def get_vlan(self, number):
         try:
             vlans_result, interfaces_result = self.node.enable(

diff --git a/tests/adapters/switches/arista_test.py b/tests/adapters/switches/arista_test.py
@@ -324,3 +324,8 @@ def test_remove_ip_from_unknown_vlan(self):
 
         with self.assertRaises(UnknownVlan):
             self.switch.remove_ip_from_vlan(123, IPNetwork("1.1.1.1/30"))
+
+    def test_transactions_commit_write_memory(self):
+        self.switch.node.should_receive("enable").with_args("write memory").once()
+
+        self.switch.commit_transaction()