cowyo/vendor/github.com/schollz/versionedtext/versionedtext_test.go

package versionedtext

import (
	"encoding/json"
	"fmt"
	"strconv"
	"testing"
	"time"
)

func ExampleNewVersionedText() {
	d := NewVersionedText("The dog jumped over the fence.")
	d.Update("The cat jumped over the fence.")
	fmt.Println(d.GetPreviousByIndex(0))
	fmt.Println(d.GetPreviousByIndex(1))
	fmt.Println(d.NumEdits())
	fmt.Println(d.GetChangeSums())
	// Output: The dog jumped over the fence. <nil>
	// The cat jumped over the fence. <nil>
	// 2
	// [30 0]
}

func BenchmarkUpdate(b *testing.B) {
	d := NewVersionedText("A word")
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		d.Update(strconv.Itoa(i))
	}
}

func BenchmarkRebuild500thOf1000(b *testing.B) {
	d := NewVersionedText("A word")
	for i := 0; i < 1001; i++ {
		d.Update(strconv.Itoa(i))
	}
	snapshots := d.GetSnapshots()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		d.GetPreviousByTimestamp(snapshots[500])
	}
}

func BenchmarkRebuild1000thOf1000(b *testing.B) {
	d := NewVersionedText("A word")
	for i := 0; i < 1001; i++ {
		d.Update(strconv.Itoa(i))
	}
	snapshots := d.GetSnapshots()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		d.GetPreviousByTimestamp(snapshots[1000])
	}
}

func BenchmarkRebuild10000thof10000(b *testing.B) {
	d := NewVersionedText("A word")
	for i := 0; i < 10001; i++ {
		d.Update(strconv.Itoa(i))
	}
	snapshots := d.GetSnapshots()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		d.GetPreviousByTimestamp(snapshots[10000])
	}
}

func BenchmarkMajorChangeSums(b *testing.B) {
	d := NewVersionedText("A word")
	time.Sleep(100 * time.Millisecond)
	d.Update("A word and adding something at the end")
	time.Sleep(2 * time.Second)
	d.Update("A word and at the end")
	time.Sleep(100 * time.Millisecond)
	d.Update("A at the end of this")
	time.Sleep(2 * time.Second)
	d.Update(" the something here end of this another here")
	time.Sleep(1 * time.Second)
	d.Update("")
	time.Sleep(100 * time.Millisecond)
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		d.GetMajorSnapshotsAndChangeSums(1)
	}
}

func TestChangeSums(t *testing.T) {
	d := NewVersionedText("A word")
	time.Sleep(100 * time.Millisecond)
	d.Update("A word and adding something at the end")
	time.Sleep(2 * time.Second)
	d.Update("A word and at the end")
	time.Sleep(100 * time.Millisecond)
	d.Update("A at the end of this")
	time.Sleep(2 * time.Second)
	d.Update(" the something here end of this another here")
	time.Sleep(1 * time.Second)
	d.Update("")
	time.Sleep(100 * time.Millisecond)

	changeSums := d.GetChangeSums()
	if changeSums[2] != -17 {
		t.Errorf("Changesums: %v", changeSums)
	}
	majorSnapshots, majorChangeSums := d.GetMajorSnapshotsAndChangeSums(1)
	if len(majorSnapshots) != 3 || len(majorChangeSums) != 3 {
		t.Errorf("Major change sums: %v\nMajor snapshots: %v", majorChangeSums, majorSnapshots)
	}
}

func TestGeneral(t *testing.T) {
	d := NewVersionedText("A word")
	time.Sleep(2 * time.Second)
	d.Update("A word and adding something at the end")
	time.Sleep(10 * time.Millisecond)
	d.Update("A (deleted) and adding something at the end, with another addition")
	time.Sleep(100 * time.Millisecond)

	// Test getting a snapshot
	snapshots := d.GetSnapshots()
	if len(snapshots) != 3 {
		t.Errorf("Should have 3 snapshots: %v", snapshots)
	}
	previousText, err := d.GetPreviousByTimestamp(snapshots[1])
	if err != nil {
		t.Error(err)
	}
	if previousText != "A word and adding something at the end" {
		t.Errorf("Did not reconstruct properly")
	}

	majorSnapshots := d.GetMajorSnapshots(1)
	if len(majorSnapshots) != 2 && majorSnapshots[0] != snapshots[1] {
		t.Errorf("Should only have one snapshot: %v", majorSnapshots)
	}

	// Test putting them in a struct and marshaling
	type SomeStruct struct {
		SomeThing string
		SomeDiffs VersionedText
	}
	var d1 SomeStruct
	d1.SomeDiffs = d
	d1.SomeThing = "Some thing"
	bJSON, err := json.Marshal(d1)
	if err != nil {
		t.Error(err)
	}
	var d2 SomeStruct
	err = json.Unmarshal(bJSON, &d2)
	if err != nil {
		t.Error(err)
	}
	if d2.SomeDiffs.GetCurrent() != d.GetCurrent() {
		t.Errorf("Problem reloading from JSON")
	}

}