示例#1
0
 def setUp(self):
     self.testrun_id = 666
    
     self.client = ResponseClient("localhost", 666)
     self.conn = ConnectionStub()
     self.channel = ChannelStub()
     self.client.conn = self.conn
     self.client.channel = self.channel
    def setUp(self):
        self.testrun_id = 666

        self.client = ResponseClient("localhost", 666)
        self.conn = ConnectionStub()
        self.channel = ChannelStub()
        self.client.conn = self.conn
        self.client.channel = self.channel
class TestResponseClient(unittest.TestCase):
    """Unittests for ResponseClient class"""

    def setUp(self):
        self.testrun_id = 666

        self.client = ResponseClient("localhost", 666)
        self.conn = ConnectionStub()
        self.channel = ChannelStub()
        self.client.conn = self.conn
        self.client.channel = self.channel

    def tearDown(self):
        pass

    def test_init(self):
        self.assertEquals(self.client.testrun_id, self.testrun_id)

    def test_predefined_response_queue(self):
        queue = "test_queue1"
        another_client = ResponseClient("localhost", 666, queue)
        self.assertEquals(another_client.response_queue, queue)

    def test_close(self):
        del self.client
        self.assertTrue(self.conn.close_called)
        self.assertTrue(self.channel.close_called)

    def test_set_error(self):
        error_info = "Flashing failed"
        error_code = "666"
        self.client.set_error(error_info, error_code)
        msg = self.channel.msg
        self.assertTrue(msg)
        content = loads(msg.body)
        self.assertTrue(isinstance(content, OTSException))
        self.assertEquals(content.errno, error_code)
        self.assertEquals(content.strerror, error_info)

    def test_add_executed_packages(self):
        environment = "Flashing failed"
        packages = ["pkg1-tests", "pkg2-tests"]
        self.client.add_executed_packages(environment, packages)
        msg = self.channel.msg
        self.assertTrue(msg)
        executed_pkgs = loads(msg.body)
        self.assertEquals(executed_pkgs.environments[0], environment)
        self.assertEquals(executed_pkgs.packages(environment), packages)

    def test_add_result(self):
        filename = "result.xml"
        file_content = "<xm>foo</xml>"
        origin = "localhost"
        test_package = "dummy-tests"
        environment = "hardware"

        self.client.add_result(filename, file_content, origin, test_package, environment)

        msg = self.channel.msg
        self.assertTrue(msg)
        result = loads(msg.body)
        self.assertEquals(result.data.name, filename)
        self.assertEquals(result.data.read(), file_content)
        self.assertEquals(result.hostname, origin)
        self.assertEquals(result.package, test_package)
        self.assertEquals(result.environment, Environment(environment))
示例#4
0
 def test_predefined_response_queue(self):
     queue = "test_queue1"
     another_client = ResponseClient("localhost", 666, queue)
     self.assertEquals(another_client.response_queue, queue)
示例#5
0
class TestResponseClient(unittest.TestCase):
    """Unittests for ResponseClient class"""
    def setUp(self):
        self.testrun_id = 666
       
        self.client = ResponseClient("localhost", 666)
        self.conn = ConnectionStub()
        self.channel = ChannelStub()
        self.client.conn = self.conn
        self.client.channel = self.channel
    
    def tearDown(self):
        pass

    def test_init(self):
        self.assertEquals(self.client.testrun_id, self.testrun_id)
        
    def test_predefined_response_queue(self):
        queue = "test_queue1"
        another_client = ResponseClient("localhost", 666, queue)
        self.assertEquals(another_client.response_queue, queue)

    def test_close(self):
        del self.client
        self.assertTrue(self.conn.close_called)
        self.assertTrue(self.channel.close_called)

    def test_set_error(self):
        error_info = "Flashing failed"
        error_code = "666"
        self.client.set_error(error_info, error_code)
        msg = self.channel.msg
        self.assertTrue(msg)
        content = loads(msg.body)
        self.assertTrue(isinstance(content, OTSException))
        self.assertEquals(content.errno, error_code)
        self.assertEquals(content.strerror, error_info)



    def test_add_executed_packages(self):
        environment = "Flashing failed"
        packages = ['pkg1-tests', 'pkg2-tests']
        self.client.add_executed_packages(environment, packages)
        msg = self.channel.msg
        self.assertTrue(msg)
        executed_pkgs = loads(msg.body)
        self.assertEquals(executed_pkgs.environments[0], environment)
        self.assertEquals(executed_pkgs.packages(environment), packages)

    def test_add_result(self):
        filename = "result.xml"
        file_content = "<xm>foo</xml>"
        origin = "localhost"
        test_package = "dummy-tests"
        environment = "hardware"

        self.client.add_result(filename,
                               file_content,
                               origin, 
                               test_package,
                               environment)

        msg = self.channel.msg
        self.assertTrue(msg)
        result = loads(msg.body)
        self.assertEquals(result.data.name, filename)
        self.assertEquals(result.data.read(), file_content)
        self.assertEquals(result.hostname, origin)
        self.assertEquals(result.package, test_package)
        self.assertEquals(result.environment, Environment(environment))