mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2024-11-14 06:54:05 -08:00
47 lines
1.4 KiB
Go
47 lines
1.4 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestWriteAndRemoveText(t *testing.T) {
|
||
|
|
writer := &ColorWriter{
|
||
|
|
Buffer: new(bytes.Buffer),
|
||
|
|
}
|
||
|
|
inputText := "This is white, <#ff5733>this is orange</>, white again"
|
||
|
|
text := writer.writeAndRemoveText("#193549", "#fff", "This is white, ", "This is white, ", inputText)
|
||
|
|
assert.Equal(t, "This is white, ", writer.string())
|
||
|
|
assert.Equal(t, "<#ff5733>this is orange</>, white again", text)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWriteAndRemoveTextColored(t *testing.T) {
|
||
|
|
writer := &ColorWriter{
|
||
|
|
Buffer: new(bytes.Buffer),
|
||
|
|
}
|
||
|
|
inputText := "This is white, <#ff5733>this is orange</>, white again"
|
||
|
|
text := writer.writeAndRemoveText("#193549", "#ff5733", "this is orange", "<#ff5733>this is orange</>", inputText)
|
||
|
|
assert.Equal(t, "this is orange", writer.string())
|
||
|
|
assert.Equal(t, "This is white, , white again", text)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWritePlain(t *testing.T) {
|
||
|
|
writer := &ColorWriter{
|
||
|
|
Buffer: new(bytes.Buffer),
|
||
|
|
}
|
||
|
|
text := "This is white, this is orange, white again"
|
||
|
|
writer.write("#193549", "#ff5733", text)
|
||
|
|
assert.Equal(t, "This is white, this is orange, white again", writer.string())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWriteColorOverride(t *testing.T) {
|
||
|
|
writer := &ColorWriter{
|
||
|
|
Buffer: new(bytes.Buffer),
|
||
|
|
}
|
||
|
|
text := "This is white, <#ff5733>this is orange</>, white again"
|
||
|
|
writer.write("#193549", "#ff5733", text)
|
||
|
|
assert.Equal(t, "This is white, this is orange, white again", writer.string())
|
||
|
|
}
|