From 4880f4be64dd6dfc2cf053753886161ab4a56d6b Mon Sep 17 00:00:00 2001
From: weihuanhuan <weihuanhuan@kylinos.cn>
Date: Tue, 9 Jul 2024 10:24:53 +0800
Subject: [PATCH] add tftpserver test file

---
 pkg/tftpserver/tftpserver_test.go | 74 ++++++++++++++++++++-----------
 1 file changed, 49 insertions(+), 25 deletions(-)

diff --git a/pkg/tftpserver/tftpserver_test.go b/pkg/tftpserver/tftpserver_test.go
index 1acd0d8..ee9bf3a 100644
--- a/pkg/tftpserver/tftpserver_test.go
+++ b/pkg/tftpserver/tftpserver_test.go
@@ -18,6 +18,7 @@ package tftpserver
 
 import (
 	"io"
+	"os"
 	"testing"
 )
 
@@ -29,43 +30,66 @@ func (rf *RFWT) ReadFrom(r io.Reader) (n int64, err error) {
 	return 0, nil
 }
 
-func (wt *RFWT) WriteTo(w io.Writer) (n int64, err error) {
+func (rf *RFWT) WriteTo(w io.Writer) (n int64, err error) {
 	return 0, nil
 }
 
-func TestTFTPServer(t *testing.T) {
-	service := NewTFTPService("127.0.0.1", "69", "testDir")
+func TestTFTPHandler(t *testing.T) {
+	tftpHandler := TFTPHandler{
+		RootDir: "",
+	}
+	sf := &RFWT{
+		RootDir: "./",
+	}
 
-	t.Run("TestStartStop", func(t *testing.T) {
-		go func() {
-			if err := service.Start(); err != nil {
-				t.Error("test fail", err)
-				return
-			}
-		}()
-		if err := service.Stop(); err != nil {
-			t.Error("test fail", err)
+	t.Run("ReadHandler_file_fail", func(t *testing.T) {
+		err := tftpHandler.ReadHandler("tftpserver.ssss", sf)
+		if err != nil {
+			t.Error(err)
 			return
 		}
+		t.Log("success")
 	})
-
-	th := &TFTPHandler{
-		RootDir: "testDir",
-	}
-	t.Run("TestReadHandler", func(t *testing.T) {
-		rf := &RFWT{}
-
-		if err := th.ReadHandler("test", rf); err != nil {
-			t.Error("test fail", err)
+	t.Run("ReadHandler", func(t *testing.T) {
+		err := tftpHandler.ReadHandler("tftpserver.go", sf)
+		if err != nil {
+			t.Error(err)
 			return
 		}
+		t.Log("success")
 	})
-	t.Run("TestWriteHandler", func(t *testing.T) {
-		wt := &RFWT{}
+	p := "tftpservers"
+	t.Run("WriteHandler", func(t *testing.T) {
+		err := tftpHandler.WriteHandler(p, sf)
+		if err != nil {
+			t.Error(err)
+			return
+		}
+		t.Log("success")
+		err = os.RemoveAll(p)
+		if err != nil {
+			t.Errorf("Error removing directory: %s\n", err)
+		} else {
+			t.Log("Directory removed successfully")
+		}
+	})
+
+}
 
-		if err := th.WriteHandler("test", wt); err != nil {
-			t.Error("test fail", err)
+func TestTFTPServer(t *testing.T) {
+	service := NewTFTPService("127.0.0.1", "69", "testDir")
+	t.Run("start", func(t *testing.T) {
+		service.Start()
+		t.Log("sssssss")
+	})
+	t.Run("stop", func(t *testing.T) {
+		service.server = nil
+		err := service.Stop()
+		if err != nil {
+			t.Error(err)
 			return
 		}
+		t.Log("stop success")
 	})
+
 }
-- 
Gitee