unioffice/spreadsheet/cell_test.go

// Copyright 2017 Baliance. All rights reserved.
//
// Use of this source code is governed by the terms of the Affero GNU General
// Public License version 3.0 as published by the Free Software Foundation and
// appearing in the file LICENSE included in the packaging of this file. A
// commercial license can be purchased by contacting sales@baliance.com.

package spreadsheet_test

import (
	"testing"
	"time"

	sml "baliance.com/gooxml/schema/schemas.openxmlformats.org/spreadsheetml"
	"baliance.com/gooxml/spreadsheet"
)

func TestCell(t *testing.T) {
	wb := spreadsheet.New()
	sheet := wb.AddSheet()
	row := sheet.AddRow()
	cell := row.AddCell()

	cell.SetBool(true)
	if cell.X().TAttr != sml.ST_CellTypeB {
		t.Errorf("expected boolean cell type, got %s", cell.X().TAttr)
	}
	if *cell.X().V != "1" {
		t.Errorf("expected 1, got %s", *cell.X().V)
	}
	cell.SetBool(false)
	if *cell.X().V != "0" {
		t.Errorf("expected 0, got %s", *cell.X().V)
	}

	cell.SetInlineString("test123")
	if cell.X().TAttr != sml.ST_CellTypeInlineStr {
		t.Errorf("expected boolean cell type, got %s", cell.X().TAttr)
	}
	if *cell.X().Is.T != "test123" {
		t.Errorf("expected test123, got %s", *cell.X().V)
	}

	cell.SetNumber(1.23)
	if cell.X().TAttr != sml.ST_CellTypeN {
		t.Errorf("expected number cell type, got %s", cell.X().TAttr)
	}
	if *cell.X().V != "1.23" {
		t.Errorf("expected 1.23, got %s", *cell.X().V)
	}

}

func TestCellGetNumber(t *testing.T) {
	wb := spreadsheet.New()
	sheet := wb.AddSheet()
	row := sheet.AddRow()
	cell := row.AddCell()

	cell.SetNumber(1.234)
	f, err := cell.GetValueAsNumber()
	if err != nil {
		t.Errorf("expected no error")
	}
	if f != 1.234 {
		t.Errorf("expected f = 1.234, got %f", f)
	}
}

func TestCellGetNumberFromText(t *testing.T) {
	wb := spreadsheet.New()
	sheet := wb.AddSheet()
	row := sheet.AddRow()
	cell := row.AddCell()

	cell.SetString("foo")
	_, err := cell.GetValueAsNumber()
	if err == nil {
		t.Errorf("expected an error")
	}
}

func TestCellGetBool(t *testing.T) {
	wb := spreadsheet.New()
	sheet := wb.AddSheet()
	row := sheet.AddRow()
	cell := row.AddCell()

	cell.SetBool(true)
	b, err := cell.GetValueAsBool()
	if err != nil {
		t.Errorf("expected no error")
	}
	if !b {
		t.Errorf("expected b = true, got false")
	}
}

func TestCellGetDate(t *testing.T) {
	wb := spreadsheet.New()
	sheet := wb.AddSheet()
	row := sheet.AddRow()
	cell := row.AddCell()

	tm := time.Date(1991, time.April, 8, 1, 2, 3, 0, time.Local)

	cell.SetDate(tm)
	f, err := cell.GetValueAsTime()
	if err != nil {
		t.Errorf("expected no error")
	}

	// SetDate truncates time
	exp := time.Date(1991, time.April, 8, 0, 0, 0, 0, time.Local)
	if !f.Equal(exp) {
		t.Errorf("expected f = %s, got %s", exp, f)
	}
}

func TestCellGetTime(t *testing.T) {
	wb := spreadsheet.New()
	sheet := wb.AddSheet()
	row := sheet.AddRow()
	cell := row.AddCell()

	tm := time.Date(1991, time.April, 8, 1, 2, 3, 0, time.Local)
	cell.SetTime(tm)
	f, err := cell.GetValueAsTime()
	if err != nil {
		t.Errorf("expected no error")
	}
	if !f.Equal(tm) {
		t.Errorf("expected f = %s, got %s", tm, f)
	}
}