← Back to branch summary

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

« back to all changes in this revision

Viewing changes to src/metadataserver/models/tests/test_nodekey.py

merged upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/metadataserver/models/tests/test_nodekey.py
23
23
    """Test NodeKeyManager."""
24
24
 
25
25
    def test_get_token_for_node_registers_node_key(self):
26
 
        node = factory.make_node()
 
26
        node = factory.make_Node()
27
27
        token = NodeKey.objects.get_token_for_node(node)
28
28
        nodekey = NodeKey.objects.get(node=node, key=token.key)
29
29
        self.assertNotEqual(None, nodekey)
30
30
        self.assertEqual(token, nodekey.token)
31
31
 
32
32
    def test_get_node_for_key_finds_node(self):
33
 
        node = factory.make_node()
 
33
        node = factory.make_Node()
34
34
        token = NodeKey.objects.get_token_for_node(node)
35
35
        self.assertEqual(node, NodeKey.objects.get_node_for_key(token.key))
36
36
 
40
40
            NodeKey.DoesNotExist, NodeKey.objects.get_node_for_key, non_key)
41
41
 
42
42
    def test_get_token_for_node_creates_token(self):
43
 
        node = factory.make_node()
 
43
        node = factory.make_Node()
44
44
        token = NodeKey.objects.get_token_for_node(node)
45
45
        self.assertEqual(node, NodeKey.objects.get_node_for_key(token.key))
46
46
 
47
47
    def test_get_token_for_node_returns_existing_token(self):
48
 
        node = factory.make_node()
 
48
        node = factory.make_Node()
49
49
        original_token = NodeKey.objects.get_token_for_node(node)
50
50
        repeated_token = NodeKey.objects.get_token_for_node(node)
51
51
        self.assertEqual(original_token, repeated_token)
52
52
 
53
53
    def test_get_token_for_node_inverts_get_node_for_key(self):
54
 
        node = factory.make_node()
 
54
        node = factory.make_Node()
55
55
        self.assertEqual(
56
56
            node,
57
57
            NodeKey.objects.get_node_for_key(
58
58
                NodeKey.objects.get_token_for_node(node).key))
59
59
 
60
60
    def test_get_node_for_key_inverts_get_token_for_node(self):
61
 
        key = NodeKey.objects.get_token_for_node(factory.make_node()).key
 
61
        key = NodeKey.objects.get_token_for_node(factory.make_Node()).key
62
62
        self.assertEqual(
63
63
            key,
64
64
            NodeKey.objects.get_token_for_node(