getRDSUpgradePath/test_awsrdscli.py at master · robins/getRDSUpgradePath · GitHub

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
import unittest

from warnings import filterwarnings
from awsrdscli import isValidRegion
from awsrdscli import getRegionTypoRecommendation

class TestMethods(unittest.TestCase):

  def test_getRegionTypoRecommendation_positives(self):
    self.assertEqual(getRegionTypoRecommendation('ap-northeast-1a'), 'ap-northeast-1')
    self.assertEqual(getRegionTypoRecommendation('ap-northeast1-1'), 'ap-northeast-1')
    self.assertEqual(getRegionTypoRecommendation('ap1-northeast-1'), 'ap-northeast-1')
    self.assertEqual(getRegionTypoRecommendation('AP-NORTHEAST-1'), 'ap-northeast-1')

  def test_isValidRegion_positives(self):
    self.assertEqual(isValidRegion('ap-northeast-1'), 1)
    self.assertEqual(isValidRegion('ap-northeast-2'), 1)
    self.assertEqual(isValidRegion('ap-northeast-3'), 1)
    self.assertEqual(isValidRegion('ap-south-1'), 1)
    self.assertEqual(isValidRegion('ap-southeast-1'), 1)
    self.assertEqual(isValidRegion('ca-central-1'), 1)
    self.assertEqual(isValidRegion('eu-central-1'), 1)
    self.assertEqual(isValidRegion('eu-north-1'), 1)
    self.assertEqual(isValidRegion('eu-west-1'), 1)
    self.assertEqual(isValidRegion('eu-west-2'), 1)
    self.assertEqual(isValidRegion('eu-west-3'), 1)
    self.assertEqual(isValidRegion('sa-east-1'), 1)
    self.assertEqual(isValidRegion('us-east-1'), 1)
    self.assertEqual(isValidRegion('us-east-2'), 1)
    self.assertEqual(isValidRegion('us-west-1'), 1)
    self.assertEqual(isValidRegion('us-west-2'), 1)

  def test_isValidRegion_negatives(self):
    self.assertEqual(isValidRegion('ap-northeast-1a'), 0)
    self.assertEqual(isValidRegion('ap-northeast-10'), 0)
    self.assertEqual(isValidRegion('ap-northeast-a1'), 0)
    self.assertEqual(isValidRegion('ap-northeast-'), 0)
    self.assertEqual(isValidRegion('ap-northeast'), 0)
    self.assertEqual(isValidRegion('ap-northeasta-1'), 0)
    self.assertEqual(isValidRegion('ap1-northeast-1'), 0)
    self.assertEqual(isValidRegion('APNE1'), 0)
    self.assertEqual(isValidRegion('apne1'), 0)
    self.assertEqual(isValidRegion('AP-NORTHEAST-1'), 0)

  def setUp(self):
      filterwarnings("ignore", category=ResourceWarning, message="unclosed.*<ssl.SSLSocket.*>")

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