X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/9fcb58db8d47a42e52eead600ddabf57ab7c29ed..2b30b239c8c66c3a1a7be35c61a9fb707df1e7ee:/services/nodemanager/tests/test_config.py

diff --git a/services/nodemanager/tests/test_config.py b/services/nodemanager/tests/test_config.py
index 15716d5277..8002b3b921 100644
--- a/services/nodemanager/tests/test_config.py
+++ b/services/nodemanager/tests/test_config.py
@@ -29,6 +29,12 @@ creds = dummy_creds
 cores = 1
 price = 0.8
 
+[Size 1.preemptible]
+instance_type = 1
+preemptible = true
+cores = 1
+price = 0.8
+
 [Logging]
 file = /dev/null
 level = DEBUG
@@ -54,11 +60,24 @@ testlogger = INFO
     def test_list_sizes(self):
         config = self.load_config()
         sizes = config.node_sizes()
-        self.assertEqual(1, len(sizes))
+        self.assertEqual(2, len(sizes))
         size, kwargs = sizes[0]
         self.assertEqual('Small', size.name)
         self.assertEqual(1, kwargs['cores'])
         self.assertEqual(0.8, kwargs['price'])
+        # preemptible is False by default
+        self.assertEqual(False, kwargs['preemptible'])
+        # instance_type == arvados node size id by default
+        self.assertEqual(kwargs['id'], kwargs['instance_type'])
+        # Now retrieve the preemptible version
+        size, kwargs = sizes[1]
+        self.assertEqual('Small', size.name)
+        self.assertEqual('1.preemptible', kwargs['id'])
+        self.assertEqual(1, kwargs['cores'])
+        self.assertEqual(0.8, kwargs['price'])
+        self.assertEqual(True, kwargs['preemptible'])
+        self.assertEqual('1', kwargs['instance_type'])
+
 
     def test_default_node_mem_scaling(self):
         config = self.load_config()