time.sleep(1) def tearDown(self): self.consumer.shutdown() self.consumer2.shutdown() self.producer.shutdown() self.producer2.shutdown() def test_send_recv(self): msg = 'ScalrMessage' msg2 = 'InternalMessage' expected_msg = None def listener(message, queue): self.assertFalse(expected_msg is None) self.assertEqual(message.name, expected_msg) self.consumer.listeners.append(listener) self.consumer2.listeners.append(listener) expected_msg = msg self.producer.send(Queues.CONTROL, P2pMessage(msg)) expected_msg = msg2 self.producer2.send(Queues.CONTROL, P2pMessage(msg2)) if __name__ == "__main__": szr_unittest.main() unittest.main()
pk = os.environ['PRIVATE_KEY'] bus.platform.set_access_data(dict( key=pk, service_account_name=email )) class CloudStorageTest(unittest.TestCase, TransferTestMixin): conn = None def setUp(self): TransferTestMixin.setUp(self) self.container = uuid.uuid4() self.key = 'path/to/candies' self.rdst = 'gcs://%s/%s' % (self.container, self.key) def tearDown(self): TransferTestMixin.tearDown(self) def native_upload(self, files): return self.trn.upload(files, self.rdst) if __name__ == "__main__": main() unittest.main()
#U.upload(self.files, cfud) url = 'cf://%s/%s/' % (self.container, self.prefix) U.upload(self.files, url) connection = cloudfiles.get_connection(username=os.environ["username"], api_key=os.environ["api_key"], serviceNet=True) #check container exists containers = [c.name for c in connection.get_all_containers()] self.assertTrue(self.container in containers) #check file uploaded and prefix is ok container = connection.get_container(self.container) objects = [c.name for c in container.get_objects()] self.assertTrue(self.obj_name in objects) #check file contains appropriate data U.download(url, self.new_dir) t_content = open(os.path.join(self.new_dir, os.path.basename(self.fname)), 'r').read() self.assertEquals(self.content, t_content) container.delete_object(self.obj_name) #delete container connection.delete_container(self.container) ''' if __name__ == "__main__": main() unittest.main()