Update Factory To Use New Reader Wrapper

Signed-off-by: Mahad Zaryab <[email protected]>

Author: mahadzaryab1

storage_v2/factoryadapter/factory.go

@@ -55,5 +55,9 @@ func (f *Factory) CreateTraceWriter() (tracestore.Writer, error) {
 
 // CreateDependencyReader implements depstore.Factory.
 func (f *Factory) CreateDependencyReader() (depstore.Reader, error) {
-	return f.ss.CreateDependencyReader()
+	dr, err := f.ss.CreateDependencyReader()
+	if err != nil {
+		return nil, err
+	}
+	return NewDependencyReader(dr), nil
 }

storage_v2/factoryadapter/factory_test.go

@@ -87,6 +87,7 @@ func TestAdapterCreateDependencyReaderError(t *testing.T) {
 	f1.On("CreateDependencyReader").Return(nil, testErr)
 
 	f := NewFactory(f1)
-	_, err := f.CreateDependencyReader()
+	r, err := f.CreateDependencyReader()
 	require.ErrorIs(t, err, testErr)
+	require.Nil(t, r)
 }