@@ -22,7 +22,6 @@ def test_wo_dataset(self):
22
22
entity_pb = datastore_pb .Entity ()
23
23
entity_pb .key .partition_id .dataset_id = _DATASET_ID
24
24
entity_pb .key .path_element .add (kind = _KIND , id = _ID )
25
- entity_pb .key .partition_id .dataset_id = _DATASET_ID
26
25
prop_pb = entity_pb .property .add ()
27
26
prop_pb .name = 'foo'
28
27
prop_pb .value .string_value = 'Foo'
@@ -32,6 +31,7 @@ def test_wo_dataset(self):
32
31
self .assertEqual (entity ['foo' ], 'Foo' )
33
32
key = entity .key ()
34
33
self .assertEqual (key ._dataset_id , _DATASET_ID )
34
+ self .assertEqual (key .namespace (), None )
35
35
self .assertEqual (key .kind (), _KIND )
36
36
self .assertEqual (key .id (), _ID )
37
37
@@ -45,7 +45,6 @@ def test_w_dataset(self):
45
45
entity_pb = datastore_pb .Entity ()
46
46
entity_pb .key .partition_id .dataset_id = _DATASET_ID
47
47
entity_pb .key .path_element .add (kind = _KIND , id = _ID )
48
- entity_pb .key .partition_id .dataset_id = _DATASET_ID
49
48
prop_pb = entity_pb .property .add ()
50
49
prop_pb .name = 'foo'
51
50
prop_pb .value .string_value = 'Foo'
@@ -56,6 +55,7 @@ def test_w_dataset(self):
56
55
self .assertEqual (entity ['foo' ], 'Foo' )
57
56
key = entity .key ()
58
57
self .assertEqual (key ._dataset_id , _DATASET_ID )
58
+ self .assertEqual (key .namespace (), None )
59
59
self .assertEqual (key .kind (), _KIND )
60
60
self .assertEqual (key .id (), _ID )
61
61
@@ -88,11 +88,13 @@ def test_w_dataset_id_in_pb(self):
88
88
pb = self ._makePB (_DATASET )
89
89
key = self ._callFUT (pb )
90
90
self .assertEqual (key ._dataset_id , _DATASET )
91
+ self .assertEqual (key .namespace (), None )
91
92
92
93
def test_w_namespace_in_pb (self ):
93
94
_NAMESPACE = 'NAMESPACE'
94
95
pb = self ._makePB (namespace = _NAMESPACE )
95
96
key = self ._callFUT (pb )
97
+ self .assertEqual (key ._dataset_id , None )
96
98
self .assertEqual (key .namespace (), _NAMESPACE )
97
99
98
100
def test_w_path_in_pb (self ):
0 commit comments