add more expects to FK generation test