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)))