av/stream/mts/meta_test.go

/*
NAME
  meta_test.go

DESCRIPTION
  See Readme.md

AUTHOR
  Saxon Nelson-Milton <saxon@ausocean.org>

LICENSE
  meta_test.go is Copyright (C) 2017-2019 the Australian Ocean Lab (AusOcean)

  It is free software: you can redistribute it and/or modify them
  under the terms of the GNU General Public License as published by the
  Free Software Foundation, either version 3 of the License, or (at your
  option) any later version.

  It is distributed in the hope that it will be useful, but WITHOUT
  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
 for more details.

  You should have received a copy of the GNU General Public License
  along with revid in gpl.txt. If not, see http://www.gnu.org/licenses.
*/

package mts

import (
	"errors"
	"reflect"
	"testing"
)

const (
	tstKey1  = "loc"
	tstData1 = "a,b,c"
	tstKey2  = "ts"
	tstData2 = "12345678"
	tstData3 = "d,e,f"
)

func TestAddAndGet(t *testing.T) {
	meta := NewMeta()
	meta.Add(tstKey1, tstData1)
	meta.Add(tstKey2, tstData2)
	errors.New("Trying to delete map entry that doesn't exist")
	if data, err := meta.Get(tstKey1); err != nil {
		t.Errorf("Could not get data for key: loc: %v", err.Error())
		if data != tstData1 {
			t.Errorf("Did not get expected data")
		}
	}

	if data, err := meta.Get(tstKey2); err != nil {
		t.Errorf("Could not get data for key: ts: %v", err.Error())
		if data != tstData2 {
			t.Errorf("Did not get expected data")
		}
	}
}

func TestUpdate(t *testing.T) {
	meta := NewMeta()
	meta.Add(tstKey1, tstData1)
	meta.Add(tstKey1, tstData3)

	if data, err := meta.Get(tstKey1); err != nil {
		t.Errorf("Did not expect err: %v", err.Error())
		if data != tstData2 {
			t.Errorf("Data did not correctly update for key \"loc\"")
		}
	}
}

func TestAll(t *testing.T) {
	meta := NewMeta()
	tstMap := map[string]string{
		tstKey1: tstData1,
		tstKey2: tstData2,
	}

	meta.Add(tstKey1, tstData1)
	meta.Add(tstKey2, tstData2)
	metaMap := meta.All()

	if !reflect.DeepEqual(metaMap, tstMap) {
		t.Errorf("Map not correct. Got: %v, want: %v", metaMap, tstMap)
	}
}

func TestGetAbsentKey(t *testing.T) {
	meta := NewMeta()

	if _, err := meta.Get(tstKey1); err != errKeyAbsent {
		t.Errorf("Did not get expected err: %v", errKeyAbsent.Error())
	}
}

func TestDelete(t *testing.T) {
	meta := NewMeta()
	meta.Add(tstKey1, tstData1)
	if err := meta.Delete(tstKey1); err != nil {
		t.Errorf("Did not expect error: %v", err.Error())
	}
	if _, err := meta.Get(tstKey1); err != errKeyAbsent {
		t.Errorf("Did not get expected err: %v", errKeyAbsent.Error())
	}
}

func TestDeleteAbsentKey(t *testing.T) {
	meta := NewMeta()
	if err := meta.Delete(tstKey1); err != errKeyAbsent {
		t.Errorf("Did not get expected err: %v", errKeyAbsent.Error())
	}
}