def setUp(self): self.loc_tree = Mock() self.loc_tree.get_hierarchy_path.return_value = ['hierarchy'] self.dbm = Mock(spec=DatabaseManager) self.submission_handler_mock = Mock(spec=SubmissionHandler) self.player = Player(self.dbm, self.submission_handler_mock, self.loc_tree)
def __init__(self, dbm, parser, channel_name, location_tree=None, is_datasender=False): Player.__init__(self, dbm, location_tree) self.parser = parser self.channel_name = channel_name self.form_code = None self.is_datasender = is_datasender
class TestSMSPlayer(TestCase): def setUp(self): self.loc_tree = Mock() self.loc_tree.get_hierarchy_path.return_value = ['hierarchy'] self.dbm = Mock(spec=DatabaseManager) self.submission_handler_mock = Mock(spec=SubmissionHandler) self.player = Player(self.dbm, self.submission_handler_mock, self.loc_tree) def test_should_not_resolve_location_hierarchy_if_hierarchy_already_passed_in(self): values = dict(l='a,b,c',t='clinic') self.player._get_location_data(values=values) self.assertEqual(['c', 'b', 'a'],values['l']) def test_should_resolve_location_hierarchy_if_hierarchy_not_passed_in(self): values = dict(l='no_hierarchy',t='clinic') self.player._get_location_data(values=values) self.assertEqual(['hierarchy'],values['l'])
class TestSMSPlayer(TestCase): def setUp(self): self.loc_tree = Mock() self.loc_tree.get_hierarchy_path.return_value = ['hierarchy'] self.dbm = Mock(spec=DatabaseManager) self.submission_handler_mock = Mock(spec=SubmissionHandler) self.player = Player(self.dbm, self.submission_handler_mock, self.loc_tree) def test_should_not_resolve_location_hierarchy_if_hierarchy_already_passed_in( self): values = dict(l='a,b,c', t='clinic') self.player._get_location_data(values=values) self.assertEqual(['c', 'b', 'a'], values['l']) def test_should_resolve_location_hierarchy_if_hierarchy_not_passed_in( self): values = dict(l='no_hierarchy', t='clinic') self.player._get_location_data(values=values) self.assertEqual(['hierarchy'], values['l'])
def __init__(self, dbm, parser, channel_name, location_tree=None): Player.__init__(self, dbm, location_tree) self.parser = parser self.channel_name = channel_name self.form_code = None self.logger = logging.getLogger("websubmission")