Updated TestUIntPrimaryKey to no longer require incrementing primary keys

This commit is contained in:
Tristan Rice 2016-01-29 14:16:15 -08:00
parent 039a759a53
commit 597b31e969

View File

@ -52,15 +52,19 @@ func TestFirstAndLastWithNoStdPrimaryKey(t *testing.T) {
} }
func TestUIntPrimaryKey(t *testing.T) { func TestUIntPrimaryKey(t *testing.T) {
insertedAnimal := &Animal{Name: "animalUint1"}
insertedAnimal2 := &Animal{Name: "animalUint2"}
DB.Save(insertedAnimal)
DB.Save(insertedAnimal2)
var animal Animal var animal Animal
DB.First(&animal, uint64(1)) DB.First(&animal, insertedAnimal.Counter)
if animal.Counter != 1 { if animal.Counter != insertedAnimal.Counter || animal.Counter <= 0 {
t.Errorf("Fetch a record from with a non-int primary key should work, but failed") t.Errorf("Fetch a record from with a non-int primary key should work, but failed; got %d", animal.Counter)
} }
DB.Model(Animal{}).Where(Animal{Counter: uint64(2)}).Scan(&animal) DB.Model(Animal{}).Where(Animal{Counter: insertedAnimal2.Counter}).Scan(&animal)
if animal.Counter != 2 { if animal.Counter != insertedAnimal2.Counter || animal.Counter <= 0 {
t.Errorf("Fetch a record from with a non-int primary key should work, but failed") t.Errorf("Fetch a record from with a non-int primary key should work, but failed; got %d", animal.Counter)
} }
} }