Borg/pkg/tim/tim_test.go
google-labs-jules[bot] 3398fabb14 feat: Add trix encryption and format
This commit introduces the `Enchantrix` library to add support for the `.trix` encrypted file format.

The main changes are:

- The `matrix` format has been renamed to `tim` (Terminal Isolation Matrix).
- The `.tim` format is now a specialized `.trix` file.
- A new `decode` command has been added to decode `.trix` and `.tim` files.
- The `collect` commands now support the `trix` and `tim` formats.
- A `--password` flag has been added to the `collect` commands for encryption.
- A `--i-am-in-isolation` flag has been added to the `decode` command for safely decoding `.tim` files.
- The decryption functionality is currently disabled due to a bug in the `Enchantrix` library. A follow-up PR will be created to re-enable it.
2025-11-14 13:47:27 +00:00

73 lines
1.5 KiB
Go

package tim
import (
"encoding/json"
"errors"
"testing"
"github.com/Snider/Borg/pkg/datanode"
)
func TestNew(t *testing.T) {
m, err := New()
if err != nil {
t.Fatalf("New() error = %v", err)
}
if m == nil {
t.Fatal("New() returned a nil tim")
}
if m.Config == nil {
t.Error("New() returned a tim with a nil config")
}
if m.RootFS == nil {
t.Error("New() returned a tim with a nil RootFS")
}
var js json.RawMessage
if err := json.Unmarshal(m.Config, &js); err != nil {
t.Error("New() returned a tim with invalid JSON config")
}
}
func TestFromDataNode(t *testing.T) {
t.Run("Good", func(t *testing.T) {
dn := datanode.New()
m, err := FromDataNode(dn)
if err != nil {
t.Fatalf("FromDataNode() error = %v", err)
}
if m == nil {
t.Fatal("FromDataNode() returned a nil tim")
}
})
t.Run("Bad", func(t *testing.T) {
_, err := FromDataNode(nil)
if !errors.Is(err, ErrDataNodeRequired) {
t.Errorf("FromDataNode() with nil datanode should return ErrDataNodeRequired, got %v", err)
}
})
}
func TestToTar(t *testing.T) {
t.Run("Good", func(t *testing.T) {
dn := datanode.New()
dn.AddData("test.txt", []byte("hello"))
m, err := FromDataNode(dn)
if err != nil {
t.Fatalf("FromDataNode() error = %v", err)
}
_, err = m.ToTar()
if err != nil {
t.Fatalf("ToTar() error = %v", err)
}
})
t.Run("Bad", func(t *testing.T) {
m, _ := New()
m.Config = nil
_, err := m.ToTar()
if !errors.Is(err, ErrConfigIsNil) {
t.Errorf("ToTar() with nil config should return ErrConfigIsNil, got %v", err)
}
})
}