| # Licensed to the Apache Software Foundation (ASF) under one * |
| # or more contributor license agreements. See the NOTICE file * |
| # distributed with this work for additional information * |
| # regarding copyright ownership. The ASF licenses this file * |
| # to you under the Apache License, Version 2.0 (the * |
| # "License"); you may not use this file except in compliance * |
| # with the License. You may obtain a copy of the License at * |
| # * |
| # http://www.apache.org/licenses/LICENSE-2.0 * |
| # * |
| # Unless required by applicable law or agreed to in writing, * |
| # software distributed under the License is distributed on an * |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * |
| # KIND, either express or implied. See the License for the * |
| # specific language governing permissions and limitations * |
| # under the License. |
| |
| import unittest |
| from etch.python.Types import * |
| from etch.binding.support.RemoteBase import * |
| |
| class What: |
| SEND = "SEND" |
| CALL = "CALL" |
| BEGINCALL = "BEGINCALL" |
| |
| class MyDeliveryService(DeliveryService): |
| |
| def _setUnittest(self, ut): |
| self.unittest = ut |
| |
| def send(self, msg): |
| self.unittest.assertEqual(None, self.what) |
| self.what = What.SEND |
| self.xmsg = msg |
| |
| def begincall(self, msg): |
| self.unittest.assertEqual(None, self.what) |
| self.what = What.BEGINCALL |
| self.xmsg = msg |
| self.xmb = PlainMailBox(None, 0L, 0, 0, 1) |
| return self.xmb |
| |
| def endcall(self, mb, responseType, responseField, timeout): |
| self.unittest.assertEqual(What.BEGINCALL, self.what) |
| self.unittest.assertEqual(self.xmb, mb) |
| self.what = What.CALL |
| self.xmb = None |
| self.xresponseType = responseType |
| self.xtimeout = timeout |
| return 23 |
| |
| def transportControl(self, control, value): |
| pass |
| |
| def transportNotify(self, event): |
| pass |
| |
| def transportQuery(self, query): |
| pass |
| |
| def addStub(self, stub): |
| pass |
| |
| def getSource(self): |
| return None |
| |
| class MyDefaultValueFactory(DefaultValueFactory): |
| typs = TypeMap() |
| class2type = Class2TypeMap() |
| DefaultValueFactory.init(typs, class2type) |
| |
| def __init__(self): |
| super(MyDefaultValueFactory,self).__init__(self.typs, self.class2type) |
| |
| |
| class Test_EtchSupportRemoteBase(unittest.TestCase): |
| |
| def setUp(self): |
| self.svc = MyDeliveryService() |
| self.vf = MyDefaultValueFactory() |
| self.rb = RemoteBase(self.svc, self.vf) |
| self.mt = Type("foo") |
| self.rmt = Type("bar") |
| self.rmf = Type("baz") |
| |
| def test_newMessage(self): |
| msg = self.rb._newMessage(self.mt) |
| msg.checkType(self.mt) |
| self.assertEqual(None, self.svc.what) |
| self.assertEqual(None, self.svc.xmsg) |
| self.assertEqual(None, self.svc.xresponseType) |
| self.assertEqual(None, self.svc.xresponseField) |
| self.assertEqual(None, self.svc.xtimeout) |
| |
| def test_send(self): |
| msg = self.rb._newMessage(self.mt) |
| self.rb._svc.send(self.msg) |
| |
| self.assertEqual(self.svc.what, What.SEND) |
| self.assertEqual(self.svc.xmsg, msg) |
| self.assertEqual(self.svc.xresponseType) |
| self.assertEqual(self.svc.xresponseField) |
| self.assertEqual(self.svc.xtimeout) |
| |
| def test_call(self): |
| msg = self.rb._newMessage(self.mt) |
| |
| result = self.rb._svc.endcall(self.rb._svc.begincall(msg), self.rmt, self.rmf, 98) |
| |
| self.assertEqual(self.svc.what , What.CALL) |
| self.assertEqual(self.svc.xmsg, msg) |
| self.assertEqual(self.svc.xresponseType, self.rmt) |
| self.assertEqual(self.svc.xresponseField, self.rmf) |
| self.assertEqual(self.svc.xtimeout, 98) |
| self.assertEqual(result, 23) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
| |
