Two extremely low-hanging fruit tests generated by GitHub Workspace (#1172)

pkg/util/daystr/daystr_test.go

@@ -0,0 +1,68 @@
+package daystr
+
+import (
+	"testing"
+	"time"
+)
+
+func TestGetCurDayStr(t *testing.T) {
+	expected := time.Now().Format("2006-01-02")
+	result := GetCurDayStr()
+	if result != expected {
+		t.Errorf("GetCurDayStr() = %v; want %v", result, expected)
+	}
+}
+
+func TestGetRelDayStr(t *testing.T) {
+	expected := time.Now().AddDate(0, 0, 5).Format("2006-01-02")
+	result := GetRelDayStr(5)
+	if result != expected {
+		t.Errorf("GetRelDayStr(5) = %v; want %v", result, expected)
+	}
+
+	expected = time.Now().AddDate(0, 0, -5).Format("2006-01-02")
+	result = GetRelDayStr(-5)
+	if result != expected {
+		t.Errorf("GetRelDayStr(-5) = %v; want %v", result, expected)
+	}
+}
+
+func TestGetCustomDayStr(t *testing.T) {
+	tests := []struct {
+		format   string
+		expected string
+	}{
+		{"today", time.Now().Format("2006-01-02")},
+		{"yesterday", time.Now().AddDate(0, 0, -1).Format("2006-01-02")},
+		{"bom", time.Date(time.Now().Year(), time.Now().Month(), 1, 0, 0, 0, 0, time.Now().Location()).Format("2006-01-02")},
+		{"bow", time.Now().AddDate(0, 0, -int(time.Now().Weekday())).Format("2006-01-02")},
+		{"2024-04-01", "2024-04-01"},
+		{"2024-04-01+1w", "2024-04-08"},
+		{"2024-04-01+1w-1d", "2024-04-07"},
+		{"2024-04-01+1m", "2024-05-01"},
+		{"2024-04-01+1m-1d", "2024-04-30"},
+	}
+
+	for _, test := range tests {
+		result, err := GetCustomDayStr(test.format)
+		if err != nil {
+			t.Errorf("GetCustomDayStr(%v) returned error: %v", test.format, err)
+		}
+		if result != test.expected {
+			t.Errorf("GetCustomDayStr(%v) = %v; want %v", test.format, result, test.expected)
+		}
+	}
+
+	invalidTests := []string{
+		"invalid",
+		"2024-04-01+1x",
+		"2024-04-01+1m-1x",
+	}
+
+	for _, test := range invalidTests {
+		_, err := GetCustomDayStr(test)
+		if err == nil {
+			t.Errorf("GetCustomDayStr(%v) expected error, got nil", test)
+		}
+	}
+}

pkg/util/ds/syncmap_test.go

@@ -0,0 +1,46 @@
+package ds
+
+import (
+	"testing"
+)
+
+func TestSyncMap_Set(t *testing.T) {
+	sm := NewSyncMap[int]()
+	sm.Set("key1", 1)
+	if sm.Get("key1") != 1 {
+		t.Errorf("expected 1, got %d", sm.Get("key1"))
+	}
+}
+
+func TestSyncMap_Get(t *testing.T) {
+	sm := NewSyncMap[int]()
+	sm.Set("key1", 1)
+	if sm.Get("key1") != 1 {
+		t.Errorf("expected 1, got %d", sm.Get("key1"))
+	}
+	if sm.Get("key2") != 0 {
+		t.Errorf("expected 0, got %d", sm.Get("key2"))
+	}
+}
+
+func TestSyncMap_GetEx(t *testing.T) {
+	sm := NewSyncMap[int]()
+	sm.Set("key1", 1)
+	value, ok := sm.GetEx("key1")
+	if !ok || value != 1 {
+		t.Errorf("expected 1, got %d", value)
+	}
+	value, ok = sm.GetEx("key2")
+	if ok || value != 0 {
+		t.Errorf("expected 0, got %d", value)
+	}
+}
+
+func TestSyncMap_Delete(t *testing.T) {
+	sm := NewSyncMap[int]()
+	sm.Set("key1", 1)
+	sm.Delete("key1")
+	if sm.Get("key1") != 0 {
+		t.Errorf("expected 0, got %d", sm.Get("key1"))
+	}
+}