22from __future__ import print_function
33from __future__ import division
44from __future__ import absolute_import
5+ from builtins import object
56from future import standard_library
67standard_library .install_aliases ()
7- from builtins import object
8+
9+ from upcloud_api import ZONE , Server , Storage
10+
811from conftest import Mock
9- from upcloud import ZONE
10- from upcloud import Server
11- from upcloud import Storage
12- import responses
13- import json
14- import pytest
12+ import json , pytest , responses
1513
16- class TestCreateServer (object ):
1714
15+ class TestCreateServer (object ):
1816
1917 def test_storage_prepare_post_body (self , manager ):
2018 s1 = Storage (os = "Ubuntu 14.04" , size = 10 )
2119 body1 = s1 .prepare_post_body ("my.example.com" , 1 )
22-
20+
2321 assert body1 ['title' ] == "my.example.com OS disk"
2422 assert body1 ['tier' ] == "maxiops"
2523 assert body1 ['size' ] == 10
@@ -32,7 +30,7 @@ def test_storage_prepare_post_body(self, manager):
3230 assert body2 ['title' ] == 'my.example.com storage disk 1'
3331 assert body2 ['tier' ] == 'maxiops'
3432 assert body2 ['action' ] == 'create'
35- assert body2 ['size' ] == 100
33+ assert body2 ['size' ] == 100
3634
3735 def test_storage_prepare_post_body_optional_attributes (self , manager ):
3836 s2 = Storage (size = 100 , address = "virtio:0" , type = "disk" )
@@ -84,11 +82,11 @@ def test_server_prepare_post_body(self):
8482 assert body ["server" ]["zone" ] == "us-chi1"
8583
8684 def test_server_prepare_post_body_optional_attributes (self ):
87- server = Server (core_number = 2 , memory_amount = 1024 ,
88- hostname = "my.example.com" ,zone = ZONE .Chicago ,
85+ server = Server (core_number = 2 , memory_amount = 1024 ,
86+ hostname = "my.example.com" ,zone = ZONE .Chicago ,
8987 storage_devices = [ Storage (os = "Ubuntu 14.04" , size = 10 )],
9088 vnc_password = "my-passwd" , password_delivery = "email" )
91-
89+
9290 body = server .prepare_post_body ()
9391 assert body ["server" ]["title" ] == "my.example.com"
9492 assert body ["server" ]["core_number" ] == 2
@@ -100,9 +98,9 @@ def test_server_prepare_post_body_optional_attributes(self):
10098
10199 @responses .activate
102100 def test_create_server (self , manager ):
103-
101+
104102 responses .add (
105- responses .POST ,
103+ responses .POST ,
106104 Mock .base_url + "/server" ,
107105 body = Mock .read_from_file ("server_create.json" ),
108106 status = 202 ,
@@ -115,7 +113,7 @@ def test_create_server(self, manager):
115113 ])
116114
117115 manager .create_server (server1 )
118-
116+
119117 # assert correct values in response
120118 assert type (server1 ).__name__ == "Server"
121119 assert server1 .core_number == "2"
0 commit comments