Example #1
0
        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()
Example #2
0
        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()
Example #3
0
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()
 
Example #4
0
                #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()