Function TestBundles

pkg/database/collections_test.go:171–212

(t *testing.T)

}

func TestBundles(t *testing.T) {
	// Test TupleBundle
	tuple1 := &base.Tuple{
		Subject: &base.Subject{Type: "user", Id: "u1"},
		Entity:  &base.Entity{Type: "book", Id: "b1"},
	}
	tuple2 := &base.Tuple{
		Subject: &base.Subject{Type: "user", Id: "u2"},
		Entity:  &base.Entity{Type: "book", Id: "b2"},
	}

	tupleBundle := TupleBundle{
		Write:  *NewTupleCollection(tuple1),
		Delete: *NewTupleCollection(tuple2),
	}

	assert.NotNil(t, tupleBundle.Write)
	assert.NotNil(t, tupleBundle.Delete)
	assert.Equal(t, 1, len(tupleBundle.Write.GetTuples()))
	assert.Equal(t, 1, len(tupleBundle.Delete.GetTuples()))
	assert.Equal(t, tuple1, tupleBundle.Write.GetTuples()[0])
	assert.Equal(t, tuple2, tupleBundle.Delete.GetTuples()[0])

	// Test AttributeBundle
	attr1, err := attribute.Attribute("book:b1$status|string:published")
	assert.NoError(t, err)

	attr2, err := attribute.Attribute("book:b2$status|string:draft")
	assert.NoError(t, err)

	attrBundle := AttributeBundle{
		Write:  *NewAttributeCollection(attr1),
		Delete: *NewAttributeCollection(attr2),
	}

	assert.NotNil(t, attrBundle.Write)
	assert.NotNil(t, attrBundle.Delete)
	assert.Equal(t, 1, len(attrBundle.Write.GetAttributes()))
	assert.Equal(t, 1, len(attrBundle.Delete.GetAttributes()))
	assert.Equal(t, attr1, attrBundle.Write.GetAttributes()[0])
	assert.Equal(t, attr2, attrBundle.Delete.GetAttributes()[0])
}