Revision 8ecbb5e8 kamaki/clients/tests.py

b/kamaki/clients/tests.py
70 70
	def setUp(self):
71 71
		cyclades_url = 'https://cyclades.okeanos.grnet.gr/api/v1.1'
72 72
		url = 'https://cyclades.okeanos.grnet.gr/plankton'
73
		token = 'Kn+G9dfmlPLR2WFnhfBOow=='
73
		self.token = 'Kn+G9dfmlPLR2WFnhfBOow=='
74 74
		self.imgid = 'b2dffe52-64a4-48c3-8a4c-8214cc3165cf'
75 75
		self.now = time.mktime(time.gmtime())
76 76
		self.imgname = 'img_%s'%self.now
77 77
		self.imglocation = 'pithos://saxtouri@grnet.gr/pithos/my.img'
78
		self.client = image(url, token)
79
		self.cyclades = cyclades(cyclades_url, token)
80

  
78
		self.client = image(url, self.token)
79
		self.cyclades = cyclades(cyclades_url, self.token)
81 80
		self._imglist={}
82 81

  
82
	def _prepare_img(self):
83
		imglocalpath = '/home/saxtouri/src/kamaki-settings/files/centos.diskdump'
84
		f = open(imglocalpath, 'rb')
85
		pithcli = pithos('https://pithos.okeanos.grnet.gr/v1',
86
			self.token,
87
			'saxtouri@grnet.gr',
88
			'pithos')
89
		pithcli.upload_object('my.img', f)
90
		f.close()
91

  
92
		self.client.register(self.imgname, self.imglocation, params=dict(is_public=True))
93
		img = self._get_img_by_name(self.imgname)
94
		self._imglist[self.imgname]=img
95

  
83 96
	def tearDown(self):
84 97
		for img in self._imglist.values():
85
			print('\tDelete img %s (%s)'%(img['name'], img['id']))
86 98
			self.cyclades.delete_image(img['id'])
87 99

  
88 100
	def _get_img_by_name(self, name):
......
178 190

  
179 191
	def test_register(self):
180 192
		"""Test register"""
181
		print('// register %s %s --public'%(self.imgname, self.imglocation))
182
		self.client.register(self.imgname, self.imglocation, params=dict(is_public=True))
183
		img = self._get_img_by_name(self.imgname)
184
		self.assertTrue(img != None)
185
		self._imglist[self.imgname]=img
193
		self._prepare_img()
194
		self.assertTrue(len(self._imglist)>0)
195
		for img in self._imglist.values():
196
			self.assertTrue(img != None)
197

  
198
	def test_set_members(self):
199
		"""Test set_members"""
200
		self._prepare_img()
201
		members=['%s@fake.net'%self.now]
202
		for img in self._imglist.values():
203
			self.client.set_members(img['id'], members)
204
			r = self.client.list_members(img['id'])
205
			self.assertEqual(r[0]['member_id'], members[0])
206

  
207
	def test_list_members(self):
208
		"""Test list_members"""
209
		self.test_set_members()
210

  
211
	def test_list_shared(self):
212
		"""Test list_shared"""
213
		#No way to test this, if I dont have member images
214
		pass
186 215

  
187 216
class testCyclades(unittest.TestCase):
188 217
	"""Set up a Cyclades thorough test"""

Also available in: Unified diff