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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
require File.dirname(__FILE__) + '/test_helper.rb'
require File.dirname(__FILE__) + '/../test_credentials.rb'
class TestAcf < Test::Unit::TestCase
RIGHT_OBJECT_TEXT = 'Right test message'
STDOUT.sync = true
def setup
TestCredentials.get_credentials
@acf= Aws::AcfInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
@s3 = Aws::S3.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
@bucket_name = "right-acf-awesome-test-bucket-0001"
@bucket_domain = "#{@bucket_name}.s3.amazonaws.com"
end
def test_01_list_distributions_part1
distributions = nil
assert_nothing_raised(Aws::AwsError) do
distributions = @acf.list_distributions
end
assert distributions.is_a?(Array)
end
def test_02_try_to_create_for_bad_bucket
# a bucket does not exist
assert_raise(Aws::AwsError) do
@acf.create_distribution("right-cloudfront-awesome-test-bucket-not-exist", "Mustn't to be born", true)
end
# a bucket is not a domain naming complied guy
bucket_name = 'right_cloudfront_awesome_test_bucket_BAD'
@s3.bucket(bucket_name, :create)
assert_raise(Aws::AwsError) do
@acf.create_distribution(bucket_name, "Mustn't to be born", true)
end
end
def test_03_create
comment = 'WooHoo!!!'
# create a test bucket
@s3.bucket(@bucket_name, :create)
# create a distribution
distribution = @acf.create_distribution(@bucket_domain, comment, true)
assert_equal comment, distribution[:comment]
assert distribution[:cnames].size == 0
assert distribution[:enabled]
end
def test_04_list_distributions_part2
distributions = @acf.list_distributions
assert distributions.size > 0
end
def get_test_distribution
@acf.list_distributions.select{ |d| d[:origin] == @bucket_domain }.first
end
def test_05_get_distribution
old = get_test_distribution
assert_nothing_raised do
@acf.get_distribution(old[:aws_id])
end
end
def test_06_get_and_set_config
config = nil
old = get_test_distribution
assert_nothing_raised do
config = @acf.get_distribution_config(old[:aws_id])
end
# change a config
config[:enabled] = false
config[:cnames] << 'x1.myawesomesite.com'
config[:cnames] << 'x2.myawesomesite.com'
# set config
set_config_result = nil
assert_nothing_raised do
set_config_result = @acf.set_distribution_config(old[:aws_id], config)
end
assert set_config_result
# reget the config and check
new_config = nil
assert_nothing_raised do
new_config = @acf.get_distribution_config(old[:aws_id])
end
assert !new_config[:enabled]
assert_equal new_config[:cnames].sort, ['x1.myawesomesite.com', 'x2.myawesomesite.com']
assert_not_equal config[:e_tag], new_config[:e_tag]
# try to update the old config again (must fail because ETAG has changed)
assert_raise(Aws::AwsError) do
@acf.set_distribution_config(old[:aws_id], config)
end
end
def test_07_caching
# enable caching
@acf.params[:cache] = true
# list distributions
@acf.list_distributions
# list the distributions again - cache should hit
assert_raise(Aws::AwsNoChange) do
@acf.list_distributions
end
# disable caching
@acf.params[:cache] = true
end
def test_08_delete_distribution
# we need ETAG so use get_distribution
distribution = @acf.get_distribution(get_test_distribution[:aws_id])
# try to delete a distribution
# should fail because
if distribution[:status] == 'InProgress'
# should fail because the distribution is not deployed yet
assert_raise(Aws::AwsError) do
@acf.delete_distribution(distribution[:aws_id], distribution[:e_tag])
end
# wait for a deployed state
print "waiting up to 5 min while the distribution is being deployed: "
100.times do
print '.'
distribution = @acf.get_distribution(distribution[:aws_id])
if distribution[:status] == 'Deployed'
print ' done'
break
end
sleep 3
end
puts
end
# only disabled and deployed distribution can be deleted
assert_equal 'Deployed', distribution[:status]
assert !distribution[:enabled]
# delete the distribution
assert_nothing_raised do
@acf.delete_distribution(distribution[:aws_id], distribution[:e_tag])
end
end
def test_09_drop_bucket
assert @s3.bucket(@bucket_name).delete
end
end
|
