mirror of
https://github.com/ralsina/tartrazine.git
synced 2024-09-20 15:31:21 +00:00
61 lines
1.7 KiB
Go
61 lines
1.7 KiB
Go
|
package slinguist
|
||
|
|
|
||
|
|
import . "gopkg.in/check.v1"
|
||
|
|
|
||
|
|
const (
|
||
|
|
multilineExecHack = `#!/bin/sh
|
||
|
|
# Next line is comment in Tcl, but not in sh... \
|
||
|
|
exec tclsh "$0" ${1+"$@"}`
|
||
|
|
|
||
|
|
multilineNoExecHack = `#!/bin/sh
|
||
|
|
#<<<#
|
||
|
|
echo "A shell script in a zkl program ($0)"
|
||
|
|
echo "Now run zkl <this file> with Hello World as args"
|
||
|
|
zkl $0 Hello World!
|
||
|
|
exit
|
||
|
|
#<<<#
|
||
|
|
println("The shell script says ",vm.arglist.concat(" "));`
|
||
|
|
)
|
||
|
|
|
||
|
|
func (s *TSuite) TestGetLanguageByShebang(c *C) {
|
||
|
|
lang, safe := GetLanguageByShebang([]byte(`#!/unknown/interpreter`))
|
||
|
|
c.Assert(lang, Equals, OtherLanguage)
|
||
|
|
c.Assert(safe, Equals, false)
|
||
|
|
|
||
|
|
lang, safe = GetLanguageByShebang([]byte(`no shebang`))
|
||
|
|
c.Assert(lang, Equals, OtherLanguage)
|
||
|
|
c.Assert(safe, Equals, false)
|
||
|
|
|
||
|
|
lang, safe = GetLanguageByShebang([]byte(`#!/usr/bin/env`))
|
||
|
|
c.Assert(lang, Equals, OtherLanguage)
|
||
|
|
c.Assert(safe, Equals, false)
|
||
|
|
|
||
|
|
lang, safe = GetLanguageByShebang([]byte(`#!/usr/bin/python -tt`))
|
||
|
|
c.Assert(lang, Equals, "Python")
|
||
|
|
c.Assert(safe, Equals, true)
|
||
|
|
|
||
|
|
lang, safe = GetLanguageByShebang([]byte(`#!/usr/bin/env python2.6`))
|
||
|
|
c.Assert(lang, Equals, "Python")
|
||
|
|
c.Assert(safe, Equals, true)
|
||
|
|
|
||
|
|
lang, safe = GetLanguageByShebang([]byte(`#!/usr/bin/env perl`))
|
||
|
|
c.Assert(lang, Equals, "Perl")
|
||
|
|
c.Assert(safe, Equals, true)
|
||
|
|
|
||
|
|
lang, safe = GetLanguageByShebang([]byte(`#! /bin/sh`))
|
||
|
|
c.Assert(lang, Equals, "Shell")
|
||
|
|
c.Assert(safe, Equals, true)
|
||
|
|
|
||
|
|
lang, safe = GetLanguageByShebang([]byte(`#!bash`))
|
||
|
|
c.Assert(lang, Equals, "Shell")
|
||
|
|
c.Assert(safe, Equals, true)
|
||
|
|
|
||
|
|
lang, safe = GetLanguageByShebang([]byte(multilineExecHack))
|
||
|
|
c.Assert(lang, Equals, "Tcl")
|
||
|
|
c.Assert(safe, Equals, true)
|
||
|
|
|
||
|
|
lang, safe = GetLanguageByShebang([]byte(multilineNoExecHack))
|
||
|
|
c.Assert(lang, Equals, "Shell")
|
||
|
|
c.Assert(safe, Equals, true)
|
||
|
|
}
|