Function TestHandler_GetObject

common/nodes/encryption/handler-encryption_test.go:54–103

(t *testing.T)

var testcases = test.TemplateSQL(srv_dao.NewKeyDAO)

func TestHandler_GetObject(t *testing.T) {

	handler := &Handler{
		// 		UserTools: &EncryptionClientMock{},
	}
	mock := nodes.NewHandlerMock()
	mock.Nodes["test"] = &tree.Node{Path: "test"}
	handler.SetNextHandler(mock)

	ctx := context.Background()
	branchInfo := nodes.BranchInfo{}
	branchInfo.DataSource = &object.DataSource{}
	// 	branchInfo.Encrypted = true
	ctx = nodes.WithBranchInfo(ctx, "in", branchInfo)

	Convey("Test Get Object without enc", t, func() {

		reqData := &models.GetRequestData{}
		reader, e := handler.GetObject(ctx, &tree.Node{Path: "test"}, reqData)
		So(reader, ShouldNotBeNil)
		So(e, ShouldBeNil)

	})

	Convey("Test Put Object without enc", t, func() {

		reqData := &models.PutRequestData{}
		_, e := handler.PutObject(ctx, &tree.Node{Path: "test2"}, strings.NewReader(""), reqData)
		// Wait for the stub to finish - TODO we should use actual service instead of stubs
		<-time.After(time.Second * 1)

		So(e, ShouldBeNil)
		So(mock.Nodes["in"], ShouldNotBeNil)
		So(mock.Nodes["in"].Path, ShouldEqual, "test2")

	})

	Convey("Test Copy Object without Enc", t, func() {

		ctx = nodes.WithBranchInfo(ctx, "from", branchInfo)
		ctx = nodes.WithBranchInfo(ctx, "to", branchInfo)
		reqData := &models.CopyRequestData{}
		_, e := handler.CopyObject(ctx, &tree.Node{Path: "test2"}, &tree.Node{Path: "test2"}, reqData)
		So(e, ShouldBeNil)
		So(mock.Nodes["from"], ShouldNotBeNil)
		So(mock.Nodes["to"], ShouldNotBeNil)

	})

}

func TestHandler_GetPut_Encrypted(t *testing.T) {
	sql.TestPrintQueries = false