lib/nyanko/test.rb in nyanko-0.0.8 vs lib/nyanko/test.rb in nyanko-0.0.9

@@ -1,8 +1,14 @@
 module Nyanko
   module Test
-    def self.activations
-      @activations ||= {}
+    class << self
+      def activations
+        @activations ||= {}
+      end
+
+      def included(base)
+        base.send :include, UnitProxyProvider
+      end
     end
 
     def enable_unit(unit_name)
       Test.activations[unit_name] = true
     end