package models

import (
  dmp "github.com/sergi/go-diff/diffmatchpatch"
  "html/template"
  "testing"
)

func assertEqual(t *testing.T, s1 string, s2 template.HTML) {
  if s1 != string(s2) {
    t.Errorf("%s should be equal %s", s2, s1)
  }
}

func TestDiffToHtml(t *testing.T) {
  assertEqual(t, "foo <span class=\"added-code\">bar</span> biz", diffToHtml([]dmp.Diff{
    dmp.Diff{dmp.DiffEqual, "foo "},
    dmp.Diff{dmp.DiffInsert, "bar"},
    dmp.Diff{dmp.DiffDelete, " baz"},
    dmp.Diff{dmp.DiffEqual, " biz"},
  }, DIFF_LINE_ADD))

  assertEqual(t, "foo <span class=\"removed-code\">bar</span> biz", diffToHtml([]dmp.Diff{
    dmp.Diff{dmp.DiffEqual, "foo "},
    dmp.Diff{dmp.DiffDelete, "bar"},
    dmp.Diff{dmp.DiffInsert, " baz"},
    dmp.Diff{dmp.DiffEqual, " biz"},
  }, DIFF_LINE_DEL))
}