2016-07-12 4 views
2

Ich habe Code für cassandra Cluster wieüberprüfen Mock mit Laufzeitobjekt namens erstellen

cluster = Cluster(
     config.CASS_CLUSTER, 
     load_balancing_policy=policies.DCAwareRoundRobinPolicy(
      config.CASS_D_CENTER)) 

Als ich UT für diese schreiben und versuchen Cluster Anruf mit meinem Parameter zu überprüfen.

mock_cluster.assert_called_with(
    config.CASS_CLUSTER, 
    load_balancing_policy=policies.DCAwareRoundRobinPolicy(
     config.CASS_D_CENTER)) 

Es gibt Fehler.

AssertionError: Expected call: Cluster(['192.168.1.1'], load_balancing_policy=<cassandra.policies.DCAwareRoundRobinPolicy object at 0x106fa2ed0>) 
Actual call: Cluster(['192.168.1.1'], load_balancing_policy=<cassandra.policies.DCAwareRoundRobinPolicy object at 0x106fa2cd0>) 

ich dies, weil beide Objekt unterschiedlich sind, ist es eine Möglichkeit, called_with in mock zu überprüfen?

Antwort

1
mock_cluster.assert_called_with(
    config.CASS_CLUSTER, 
    load_balancing_policy=mock.ANY)