fixup! fixup! fixup! fixup! Blah

Author: shreyasbharath

lib/cpp_dependency_graph/include_file_dependency_graph.rb

@@ -4,6 +4,8 @@
 require_relative 'link'
 require_relative 'cycle_detector'
 
+require 'pp'
+
 # Returns a hash of individual file include links
 class IncludeFileDependencyGraph
   def initialize(project)
@@ -23,10 +25,10 @@ def links(file_name)
     components = @project.source_components
     all_source_files = components.values.flat_map(&:source_files)
     files = all_source_files.select do |file|
-      file.basename == file_name
+      file.includes.include?(file_name)
     end
     files.map do |file|
-      links = file.includes.map { |inc| Link.new(file.basename, inc, false) }
+      links = [Link.new(file.basename, file_name, false)]
       [file.basename, links]
     end.to_h
   end

spec/test/include_file_dependency_graph_spec.rb

@@ -13,9 +13,9 @@
 
   it 'returns include links for a specified file' do
     expected_links = {}
-    expected_links['Engine.h'] = [Link.new('Engine.h', 'Engine.cpp', false),
-                                  Link.new('Engine.h', 'Display.cpp', false),
-                                  Link.new('Engine.h', 'Display.h', false)]
+    expected_links['Display.cpp'] = [Link.new('Display.cpp', 'Engine.h', false)]
+    # expected_links['Display.h'] = [Link.new('Display.h', 'Engine.h', false)]
+    expected_links['Engine.cpp'] = [Link.new('Engine.cpp', 'Engine.h', false)]
     expect(include_dependency_graph.links('Engine.h')).to eq(expected_links)
   end
 end