Add keywords and test keyword identification

Gamelan currently playing: Lancaran Penghijauan co-authored-by: Connor Walsh <[email protected]> co-authored-by: Octavia <[email protected]>
asmr-hex · Apr 7, 2018 · 6f14d3d · 6f14d3d
1 parent eff5c13
commit 6f14d3d
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/lexer_test.go b/lexer_test.go
@@ -124,6 +124,25 @@ func (s *LexerSuite) TestTokenizeIdentifier() {
 	}
 }
 
+func (s *LexerSuite) TestTokenizeIdentifier_Keyword() {
+	conf := &Config{SrcFileName: "test/test_identifiers_keyword.doc"}
+	compt := NewCompterpreter(conf)
+
+	err := compt.LoadSourceCode()
+	s.NoError(err)
+
+	compt.Advance()
+	// advance ptr to first character
+	for _, op := range []string{"if"} {
+		compt.CurrentToken = Token{}
+		compt.TokenizeIdentifier(compt.CurrentChar)
+		if string(compt.CurrentChar) == "EOF" {
+			break
+		}
+		s.EqualValues(compt.CurrentToken.Value, op)
+	}
+}
+
 func (s *LexerSuite) TestLex() {
 	conf := &Config{SrcFileName: "test/test_tokenize.doc"}
 	compt := NewCompterpreter(conf)

diff --git a/symbols.go b/symbols.go
@@ -18,6 +18,9 @@ const (
 
 	VARIABLE_IDENTIFIER = "VARIABLE_IDENTIFIER"
 	FUNCTION_IDENTIFIER = "FUNCTION_IDENTIFIER"
+
+	IF_KEYWORD       = "if"
+	FUNCTION_KEYWORD = "❦"
 )
 
 var (
@@ -55,7 +58,10 @@ func PopulateSymbols() *Symbols {
 			EXIT_OPERATOR,
 			NOOP,
 		},
-		Keywords: []string{},
+		Keywords: []string{
+			IF_KEYWORD,
+			FUNCTION_KEYWORD,
+		},
 		Punctuation: []string{
 			R_PAREN_PUNCTION,
 			L_PAREN_PUNCUTATION,