summaryrefslogtreecommitdiff
path: root/tests/unit/ec2/test_instancetype.py
blob: dd81ca495d2a5a27823607dec19eae5bf4cc3604 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#!/usr/bin/env python
from tests.unit import unittest
from tests.unit import AWSMockServiceTestCase

import boto.ec2

from boto.ec2.instancetype import InstanceType
from boto.ec2.connection import EC2Connection

class TestEC2ConnectionBase(AWSMockServiceTestCase):
    connection_class = EC2Connection

    def setUp(self):
        super(TestEC2ConnectionBase, self).setUp()
        self.ec2 = self.service_connection


class TestReservedInstanceOfferings(TestEC2ConnectionBase):

    def default_body(self):
        return b"""
            <DescribeInstanceTypesResponseType xmlns="http://ec2.amazonaws.com/doc/2013-02-01/">
                <VmTypeMessage/>
                <instanceTypeDetails>
                    <item>
                        <name>m1.small</name><cpu>1</cpu><disk>5</disk><memory>256</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>t1.micro</name><cpu>1</cpu><disk>5</disk><memory>256</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>m1.medium</name><cpu>1</cpu><disk>10</disk><memory>512</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>c1.medium</name><cpu>2</cpu><disk>10</disk><memory>512</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>m1.large</name><cpu>2</cpu><disk>10</disk><memory>512</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>m1.xlarge</name><cpu>2</cpu><disk>10</disk><memory>1024</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>c1.xlarge</name><cpu>2</cpu><disk>10</disk><memory>2048</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>m2.xlarge</name><cpu>2</cpu><disk>10</disk><memory>2048</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>m3.xlarge</name><cpu>4</cpu><disk>15</disk><memory>2048</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>m2.2xlarge</name><cpu>2</cpu><disk>30</disk><memory>4096</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>m3.2xlarge</name><cpu>4</cpu><disk>30</disk><memory>4096</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>cc1.4xlarge</name><cpu>8</cpu><disk>60</disk><memory>3072</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>m2.4xlarge</name><cpu>8</cpu><disk>60</disk><memory>4096</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>hi1.4xlarge</name><cpu>8</cpu><disk>120</disk><memory>6144</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>cc2.8xlarge</name><cpu>16</cpu><disk>120</disk><memory>6144</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>cg1.4xlarge</name><cpu>16</cpu><disk>200</disk><memory>12288</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>cr1.8xlarge</name><cpu>16</cpu><disk>240</disk><memory>16384</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                    <item>
                        <name>hs1.8xlarge</name><cpu>48</cpu><disk>24000</disk><memory>119808</memory>
                        <availability/><ephemeralDisk/>
                    </item>
                </instanceTypeDetails>
            </DescribeInstanceTypesResponseType>
        """

    def test_get_instance_types(self):
        self.set_http_response(status_code=200)
        response = self.ec2.get_all_instance_types()
        self.assertEqual(len(response), 18)
        instance_type = response[0]
        self.assertEqual(instance_type.name, 'm1.small')
        self.assertEqual(instance_type.cores, '1')
        self.assertEqual(instance_type.disk, '5')
        self.assertEqual(instance_type.memory, '256')
        instance_type = response[17]
        self.assertEqual(instance_type.name, 'hs1.8xlarge')
        self.assertEqual(instance_type.cores, '48')
        self.assertEqual(instance_type.disk, '24000')
        self.assertEqual(instance_type.memory, '119808')


if __name__ == '__main__':
    unittest.main()