mdtoc/main_test.go

package main

import (
	"log"
	"testing"
)

func TestGet_heading_level(t *testing.T) {
	level := Get_heading_level("# foo")

	if level != 1 {
		log.Fatal("Level should be 1.")
	}

	level = Get_heading_level("not a header")

	if level != 0 {
		log.Fatal("A non header should be level 0!")
	}

	level = Get_heading_level("#")

	if level != 1 {
		log.Fatal("Level should be 1.")
	}
}