Do not assume every client module has clientSpec

uber · Mar 8, 2018 · 30969a3 · 30969a3
1 parent ad7c719
commit 30969a3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/codegen/module_system.go b/codegen/module_system.go
@@ -1276,9 +1276,8 @@ func GenerateMockInitializer(
 ) ([]byte, error) {
 	leafWithFixture := map[string]string{}
 	for _, leaf := range instance.RecursiveDependencies["client"] {
-		// cast is guaranteed to succeed at this point
-		spec := leaf.genSpec.(*ClientSpec)
-		if spec.Fixture != nil {
+		spec, ok := leaf.genSpec.(*ClientSpec)
+		if ok && spec.Fixture != nil {
 			leafWithFixture[leaf.InstanceName] = spec.Fixture.ImportPath
 		}
 	}