30 lines
751 B
Go
30 lines
751 B
Go
|
package storage
|
||
|
|
|
||
|
|
import (
|
||
|
|
"cwtch.im/cwtch/model"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
const testingDir = "./testing"
|
||
|
|
const filenameBase = "testStream"
|
||
|
|
const password = "asdfqwer"
|
||
|
|
const line1 = "Hello from storage!"
|
||
|
|
|
||
|
|
func TestStreamStoreWriteRead(t *testing.T) {
|
||
|
|
os.RemoveAll(testingDir)
|
||
|
|
os.Mkdir(testingDir, 0777)
|
||
|
|
ss1 := NewStreamStore(testingDir, filenameBase, password)
|
||
|
|
m := model.Message{Message: line1}
|
||
|
|
ss1.Write(m)
|
||
|
|
|
||
|
|
ss2 := NewStreamStore(testingDir, filenameBase, password)
|
||
|
|
messages := ss2.Read()
|
||
|
|
if len(messages) != 1 {
|
||
|
|
t.Errorf("Read messages has wrong length. Expected: 1 Actual: %d\n", len(messages))
|
||
|
|
}
|
||
|
|
if messages[0].Message != line1 {
|
||
|
|
t.Errorf("Read message has wrong content. Expected: '%v' Actual: '%v'\n", line1, messages[0].Message)
|
||
|
|
}
|
||
|
|
}
|