(new): v1.2 release

drafts_test.go

@@ -29,6 +29,32 @@ func TestDraftFlushRequiresChatID(t *testing.T) {
 	}
 }
 
+func TestDraftFlushEmptyRemovesDraft(t *testing.T) {
+	provider := NewLinearDraftProvider(nil, 0)
+	draft := provider.NewDraft(tgapi.ParseNone)
+
+	if err := draft.Flush(); err != nil {
+		t.Fatalf("Flush returned error: %v", err)
+	}
+	if _, ok := provider.GetDraft(draft.ID); ok {
+		t.Fatal("empty flushed draft remained in provider")
+	}
+}
+
+func TestDraftFlushAllRemovesClearedDrafts(t *testing.T) {
+	provider := NewLinearDraftProvider(nil, 0)
+	draft := provider.NewDraft(tgapi.ParseNone)
+	draft.Message = "discarded"
+	draft.Clear()
+
+	if err := provider.FlushAll(); err != nil {
+		t.Fatalf("FlushAll returned error: %v", err)
+	}
+	if _, ok := provider.GetDraft(draft.ID); ok {
+		t.Fatal("cleared draft remained in provider")
+	}
+}
+
 func TestMsgContextNewDraftWorksWithoutLimiter(t *testing.T) {
 	ctx := &MessageContext{
 		API: &tgapi.API{},