def setUp(self): self.store = StateGroupStore() storage = Mock(main=self.store, state=self.store) hs = Mock( spec_set=[ "config", "get_datastore", "get_storage", "get_auth", "get_state_handler", "get_clock", "get_state_resolution_handler", "hostname", ] ) hs.config = default_config("tesths", True) hs.get_datastore.return_value = self.store hs.get_state_handler.return_value = None hs.get_clock.return_value = MockClock() hs.get_auth.return_value = Auth(hs) hs.get_state_resolution_handler = lambda: StateResolutionHandler(hs) hs.get_storage.return_value = storage self.state = StateHandler(hs) self.event_id = 0
def setUp(self): self.dummy_store = _DummyStore() storage_controllers = Mock(main=self.dummy_store, state=self.dummy_store) hs = Mock( spec_set=[ "config", "get_datastores", "get_storage_controllers", "get_auth", "get_state_handler", "get_clock", "get_state_resolution_handler", "get_account_validity_handler", "get_macaroon_generator", "hostname", ] ) clock = cast(Clock, MockClock()) hs.config = default_config("tesths", True) hs.get_datastores.return_value = Mock(main=self.dummy_store) hs.get_state_handler.return_value = None hs.get_clock.return_value = clock hs.get_macaroon_generator.return_value = MacaroonGenerator( clock, "tesths", b"verysecret" ) hs.get_auth.return_value = Auth(hs) hs.get_state_resolution_handler = lambda: StateResolutionHandler(hs) hs.get_storage_controllers.return_value = storage_controllers self.state = StateHandler(hs) self.event_id = 0
def setUp(self): self.store = Mock( spec_set=[ "get_state_groups", "add_event_hashes", ] ) hs = Mock(spec=["get_datastore"]) hs.get_datastore.return_value = self.store self.state = StateHandler(hs) self.event_id = 0
def setUp(self): self.store = StateGroupStore() hs = Mock(spec_set=[ "get_datastore", "get_auth", "get_state_handler", "get_clock", "get_state_resolution_handler", ]) hs.get_datastore.return_value = self.store hs.get_state_handler.return_value = None hs.get_clock.return_value = MockClock() hs.get_auth.return_value = Auth(hs) hs.get_state_resolution_handler = lambda: StateResolutionHandler(hs) self.state = StateHandler(hs) self.event_id = 0
def setUp(self): self.store = Mock( spec_set=[ "get_state_groups", "add_event_hashes", ] ) hs = Mock(spec=[ "get_datastore", "get_auth", "get_state_handler", "get_clock", ]) hs.get_datastore.return_value = self.store hs.get_state_handler.return_value = None hs.get_auth.return_value = Auth(hs) hs.get_clock.return_value = MockClock() self.state = StateHandler(hs) self.event_id = 0
def setUp(self): self.persistence = Mock(spec=[ "get_unresolved_state_tree", "update_current_state", "get_latest_pdus_in_context", "get_current_state_pdu", "get_pdu", "get_power_level", ]) self.replication = Mock(spec=["get_pdu"]) hs = Mock(spec=["get_datastore", "get_replication_layer"]) hs.get_datastore.return_value = self.persistence hs.get_replication_layer.return_value = self.replication hs.hostname = "bob.com" self.state = StateHandler(hs)
def setUp(self): self.store = Mock( spec_set=[ "get_state_groups_ids", "add_event_hashes", "get_events", "get_next_state_group", "get_state_group_delta", ] ) hs = Mock(spec_set=[ "get_datastore", "get_auth", "get_state_handler", "get_clock", ]) hs.get_datastore.return_value = self.store hs.get_state_handler.return_value = None hs.get_clock.return_value = MockClock() hs.get_auth.return_value = Auth(hs) self.store.get_next_state_group.side_effect = Mock self.store.get_state_group_delta.return_value = (None, None) self.state = StateHandler(hs) self.event_id = 0
def build_state_handler(self): return StateHandler(self)
def get_state_handler(self) -> StateHandler: return StateHandler(self)