shared/flavor/flay/recipes/cookbook.rb in chef-flavor-flay-1.0.1 vs shared/flavor/flay/recipes/cookbook.rb in chef-flavor-flay-1.1.0

@@ -1,6 +1,6 @@
-context      = ChefDK::Generator.context
-cookbook_dir = File.join(context.cookbook_root, context.cookbook_name)
+context      = Flay::Context.current
+cookbook_dir = File.join(context.cookbook_root, context.directory_name)
 kitchen_dir  = File.join(cookbook_dir, "test", "integration")
 
 # Common Cookbook Things
 directory cookbook_dir
 directory "#{cookbook_dir}/.bundle"