def setUp(self): self.network_name = "test_net_name" self.driver = DefaultPolicyDriver(self.network_name) assert_equal(self.driver.profile_name, self.network_name) # Mock the DatastoreClient self.client = MagicMock(spec=DatastoreClient) self.driver._client = self.client