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
