Skip to content
Snippets Groups Projects
Commit eb752e21 authored by Will Carter's avatar Will Carter
Browse files

Added a new LiteNode type

parent cc116cc4
No related branches found
No related tags found
No related merge requests found
...@@ -19,6 +19,7 @@ const ( ...@@ -19,6 +19,7 @@ const (
Generic = Type(iota) Generic = Type(iota)
Gateway Gateway
Node Node
LiteNode
User User
NumTypes // Gives number of ID types NumTypes // Gives number of ID types
) )
...@@ -32,6 +33,8 @@ func (t Type) String() string { ...@@ -32,6 +33,8 @@ func (t Type) String() string {
return "gateway" return "gateway"
case Node: case Node:
return "node" return "node"
case LiteNode:
return "litenode"
case User: case User:
return "user" return "user"
case NumTypes: case NumTypes:
......
...@@ -16,8 +16,9 @@ func TestType_String(t *testing.T) { ...@@ -16,8 +16,9 @@ func TestType_String(t *testing.T) {
expectedGeneric := "generic" expectedGeneric := "generic"
expectedGateway := "gateway" expectedGateway := "gateway"
expectedNode := "node" expectedNode := "node"
expectedLiteNode := "litenode"
expectedUser := "user" expectedUser := "user"
expectedNumTypes := "4" expectedNumTypes := "5"
// Test Generic stringer // Test Generic stringer
testVal := Generic.String() testVal := Generic.String()
...@@ -40,6 +41,13 @@ func TestType_String(t *testing.T) { ...@@ -40,6 +41,13 @@ func TestType_String(t *testing.T) {
"\n\texpected: %s\n\treceived: %s", expectedNode, testVal) "\n\texpected: %s\n\treceived: %s", expectedNode, testVal)
} }
// Test LiteNode stringer
testVal = LiteNode.String()
if expectedLiteNode != testVal {
t.Errorf("String() returned incorrect string for LiteNode type."+
"\n\texpected: %s\n\treceived: %s", expectedLiteNode, testVal)
}
// Test User stringer // Test User stringer
testVal = User.String() testVal = User.String()
if expectedUser != testVal { if expectedUser != testVal {
...@@ -59,8 +67,8 @@ func TestType_String(t *testing.T) { ...@@ -59,8 +67,8 @@ func TestType_String(t *testing.T) {
// defined. // defined.
func TestType_String_Error(t *testing.T) { func TestType_String_Error(t *testing.T) {
// Expected/test values // Expected/test values
expectedError := "UNKNOWN ID TYPE: 5" expectedError := "UNKNOWN ID TYPE: 6"
testType := Type(5) testType := Type(6)
// Test stringer error // Test stringer error
testVal := testType.String() testVal := testType.String()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment