def test_gen_primary_key_compound_key(self):
# gen_primary_key() returns an iterable of values from the model
# instance's primary key.
storm_object = FakeStormCompoundPrimaryKeyClass()
storm_object.id1 = 1234
storm_object.id2 = 5678
self.assertEqual([1234, 5678], list(gen_primary_key(storm_object)))
def test_gen_primary_key_compound_key(self):
# gen_primary_key() returns an iterable of values from the model
# instance's primary key.
storm_object = FakeStormCompoundPrimaryKeyClass()
storm_object.id1 = 1234
storm_object.id2 = 5678
self.assertEqual([1234, 5678], list(gen_primary_key(storm_object)))
def test_gen_primary_key(self):
# gen_primary_key() returns an iterable of values from the model
# instance's primary key.
storm_object = FakeStormClass()
storm_object.id = 1234
self.assertEqual([1234], list(gen_primary_key(storm_object)))
def test_gen_primary_key(self):
# gen_primary_key() returns an iterable of values from the model
# instance's primary key.
storm_object = FakeStormClass()
storm_object.id = 1234
self.assertEqual([1234], list(gen_primary_key(storm_object)))
