~rvb/maas/transaction-1.7-bug-1409852

« back to all changes in this revision

Viewing changes to src/maasserver/clusterrpc/tests/test_power.py

merged upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
61
61
    def test__powers_single_node(self):
62
62
        rpc_fixture = self.prepare_rpc()
63
63
 
64
 
        node = factory.make_node()
 
64
        node = factory.make_Node()
65
65
        cluster, io = rpc_fixture.makeCluster(node.nodegroup, self.command)
66
66
 
67
67
        nodes_for_call = [
89
89
            ))
90
90
 
91
91
    def test__powers_multiple_nodes(self):
92
 
        nodegroup = factory.make_node_group()
93
 
        nodes = [factory.make_node(nodegroup=nodegroup) for _ in xrange(3)]
 
92
        nodegroup = factory.make_NodeGroup()
 
93
        nodes = [factory.make_Node(nodegroup=nodegroup) for _ in xrange(3)]
94
94
        nodes_for_call = list(  # Use list() to avoid namespace leaks.
95
95
            (node.system_id, node.hostname, node.nodegroup.uuid,
96
96
             node.get_effective_power_info())
130
130
 
131
131
        nodes = {}
132
132
        for _ in xrange(3):
133
 
            node = factory.make_node()
 
133
            node = factory.make_Node()
134
134
            cluster, io = rpc_fixture.makeCluster(node.nodegroup, self.command)
135
135
            nodes[node] = cluster, io
136
136