lib/avromatic/model_registry.rb in avromatic-2.2.1 vs lib/avromatic/model_registry.rb in avromatic-2.2.2

@@ -27,10 +27,11 @@
       name = model_fullname(model)
       raise "'#{name}' has already been registered" if registered?(name)
       @hash[name] = model
     end
 
-    def registered?(fullname)
+    def registered?(fullname_or_model)
+      fullname = fullname_or_model.is_a?(String) ? fullname_or_model : model_fullname(fullname_or_model)
       @hash.key?(fullname)
     end
 
     def model_fullname(model)
       name = model.avro_schema.fullname