@@ -419,24 +419,25 @@ func TestDiagnosticsHandlerWithEDOT(t *testing.T) {
419419 err := os .MkdirAll (path .Join (tempAgentRoot , "data" ), 0755 )
420420 require .NoError (t , err )
421421 called := false
422- s := mockhandlers . NewMockServer (t , paths .DiagnosticsExtensionSocket (), & called )
422+ s := NewMockServer (t , paths .DiagnosticsExtensionSocket (), & called )
423423 defer func () {
424424 require .NoError (t , s .Shutdown (context .Background ()))
425425 }()
426- mockDiagProvider := mockhandlers . NewDiagnosticsProvider (t )
426+ mockDiagProvider := newMockDiagnosticsProvider (t )
427427 mockDiagProvider .EXPECT ().DiagnosticHooks ().Return ([]diagnostics.Hook {hook1 })
428428 mockDiagProvider .EXPECT ().PerformDiagnostics (mock .Anything , mock .Anything ).Return ([]runtime.ComponentUnitDiagnostic {mockUnitDiagnostic })
429429 mockDiagProvider .EXPECT ().PerformComponentDiagnostics (mock .Anything , mock .Anything ).Return ([]runtime.ComponentDiagnostic {mockComponentDiagnostic }, nil )
430430
431- mockAcker := mockackers .NewAcker (t )
431+ mockAcker := acker .NewMockAcker (t )
432+
432433 mockAcker .EXPECT ().Ack (mock .Anything , mock .Anything ).RunAndReturn (func (ctx context.Context , a fleetapi.Action ) error {
433434 require .IsType (t , new (fleetapi.ActionDiagnostics ), a )
434435 assert .NoError (t , a .(* fleetapi.ActionDiagnostics ).Err )
435436 return nil
436437 })
437438 mockAcker .EXPECT ().Commit (mock .Anything ).Return (nil )
438439
439- mockUploader := mockhandlers . NewUploader (t )
440+ mockUploader := NewMockUploader (t )
440441 mockUploader .EXPECT ().UploadDiagnostics (mock .Anything , mock .Anything , mock .Anything , mock .Anything , mock .Anything ).RunAndReturn (func (ctx context.Context , s1 , s2 string , i int64 , r io.Reader ) (string , error ) {
441442 expectedContent := map [string ][]byte {
442443 "components/ComponentID/mock_component_diag_file.yaml" : []byte ("hello: component\n " ),
0 commit comments