Project: Unified Migration (#364)

---------

Co-authored-by: Ania Misiorek <[email protected]>
Co-authored-by: Ania Misiorek <[email protected]>

Author: zliang-akamai

linode_api4/objects/linode.py

@@ -612,6 +612,11 @@ def _interface_create(self, body: Dict[str, Any]) -> NetworkInterface:
         return i
 
 
+class MigrationType:
+    COLD = "cold"
+    WARM = "warm"
+
+
 class Instance(Base):
     """
     A Linode Instance.
@@ -948,7 +953,13 @@ def reboot(self):
             return False
         return True
 
-    def resize(self, new_type, allow_auto_disk_resize=True, **kwargs):
+    def resize(
+        self,
+        new_type,
+        allow_auto_disk_resize=True,
+        migration_type: MigrationType = MigrationType.COLD,
+        **kwargs,
+    ):
         """
         Resizes a Linode you have the read_write permission to a different Type. If any
         actions are currently running or queued, those actions must be completed first
@@ -970,6 +981,10 @@ def resize(self, new_type, allow_auto_disk_resize=True, **kwargs):
                                        data must fit within the smaller disk size. Defaults to true.
         :type: allow_auto_disk_resize: bool
 
+        :param migration_type: Type of migration to be used when resizing a Linode.
+                               Customers can choose between warm and cold, the default type is cold.
+        :type: migration_type: str
+
         :returns: True if the operation was successful.
         :rtype: bool
         """
@@ -979,6 +994,7 @@ def resize(self, new_type, allow_auto_disk_resize=True, **kwargs):
         params = {
             "type": new_type,
             "allow_auto_disk_resize": allow_auto_disk_resize,
+            "migration_type": migration_type,
         }
         params.update(kwargs)
 
@@ -1438,7 +1454,12 @@ def mutate(self, allow_auto_disk_resize=True):
 
         return True
 
-    def initiate_migration(self, region=None, upgrade=None):
+    def initiate_migration(
+        self,
+        region=None,
+        upgrade=None,
+        migration_type: MigrationType = MigrationType.COLD,
+    ):
         """
         Initiates a pending migration that is already scheduled for this Linode
         Instance
@@ -1459,10 +1480,16 @@ def initiate_migration(self, region=None, upgrade=None):
                         region field does not allow upgrades, then the endpoint will return a 400 error
                         code and the migration will not be performed.
         :type: upgrade: bool
+
+        :param migration_type: The type of migration that will be used for this Linode migration.
+                               Customers can only use this param when activating a support-created migration.
+                               Customers can choose between a cold and warm migration, cold is the default type.
+        :type: mirgation_type: str
         """
         params = {
             "region": region.id if issubclass(type(region), Base) else region,
             "upgrade": upgrade,
+            "type": migration_type,
         }
 
         util.drop_null_keys(params)

test/integration/helpers.py

@@ -1,4 +1,3 @@
-import random
 import time
 from typing import Callable
 
@@ -8,7 +7,7 @@
 
 
 def get_test_label():
-    unique_timestamp = str(int(time.time()) + random.randint(0, 1000))
+    unique_timestamp = str(time.time_ns())
     label = "IntTestSDK_" + unique_timestamp
     return label
 
@@ -94,13 +93,13 @@ def retry_sending_request(retries: int, condition: Callable, *args) -> object:
 
 
 def send_request_when_resource_available(
-    timeout: int, func: Callable, *args
+    timeout: int, func: Callable, *args, **kwargs
 ) -> object:
     start_time = time.time()
 
     while True:
         try:
-            res = func(*args)
+            res = func(*args, **kwargs)
             return res
         except ApiError as e:
             if (
@@ -110,9 +109,9 @@ def send_request_when_resource_available(
             ):
                 if time.time() - start_time > timeout:
                     raise TimeoutError(
-                        "Timeout Error: resource is not available in"
+                        "Timeout Error: resource is not available in "
                         + str(timeout)
-                        + "seconds"
+                        + " seconds"
                     )
                 time.sleep(10)
             else:

test/integration/models/test_linode.py

@@ -18,6 +18,7 @@
     Instance,
     Type,
 )
+from linode_api4.objects.linode import MigrationType
 
 
 @pytest.fixture(scope="session")
@@ -302,6 +303,29 @@ def test_linode_resize_with_class(
     assert linode.status == "running"
 
 
+def test_linode_resize_with_migration_type(
+    create_linode_for_long_running_tests,
+):
+    linode = create_linode_for_long_running_tests
+    m_type = MigrationType.WARM
+
+    wait_for_condition(10, 100, get_status, linode, "running")
+
+    time.sleep(5)
+    res = linode.resize(new_type="g6-standard-1", migration_type=m_type)
+
+    assert res
+
+    wait_for_condition(10, 300, get_status, linode, "resizing")
+
+    assert linode.status == "resizing"
+
+    # Takes about 3-5 minute to resize, sometimes longer...
+    wait_for_condition(30, 600, get_status, linode, "running")
+
+    assert linode.status == "running"
+
+
 def test_linode_boot_with_config(create_linode):
     linode = create_linode
 
@@ -416,15 +440,16 @@ def test_linode_initate_migration(test_linode_client):
     chosen_region = available_regions[4]
     label = get_test_label() + "_migration"
 
-    linode, password = client.linode.instance_create(
-        "g6-nanode-1", chosen_region, image="linode/debian10", label=label
+    linode, _ = client.linode.instance_create(
+        "g6-nanode-1", chosen_region, image="linode/debian12", label=label
     )
 
-    wait_for_condition(10, 100, get_status, linode, "running")
     # Says it could take up to ~6 hrs for migration to fully complete
-
     send_request_when_resource_available(
-        300, linode.initiate_migration, "us-mia"
+        300,
+        linode.initiate_migration,
+        region="us-mia",
+        migration_type=MigrationType.COLD,
     )
 
     res = linode.delete()