p2p/discover/sim_testmain_test.go

   1 // Copyright 2016 The go-ethereum Authors
   2 // This file is part of the go-ethereum library.
   3 //
   4 // The go-ethereum library is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU Lesser General Public License as published by
   6 // the Free Software Foundation, either version 3 of the License, or
   7 // (at your option) any later version.
   8 //
   9 // The go-ethereum library is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12 // GNU Lesser General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Lesser General Public License
  15 // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
  16
  17 // +build go1.4,nacl,faketime_simulation
  18
  19 package discover
  20
  21 import (
  22         "os"
  23         "runtime"
  24         "testing"
  25         "unsafe"
  26 )
  27
  28 // Enable fake time mode in the runtime, like on the go playground.
  29 // There is a slight chance that this won't work because some go code
  30 // might have executed before the variable is set.
  31
  32 //go:linkname faketime runtime.faketime
  33 var faketime = 1
  34
  35 func TestMain(m *testing.M) {
  36         // We need to use unsafe somehow in order to get access to go:linkname.
  37         _ = unsafe.Sizeof(0)
  38
  39         // Run the actual test. runWithPlaygroundTime ensures that the only test
  40         // that runs is the one calling it.
  41         runtime.GOMAXPROCS(8)
  42         os.Exit(m.Run())
  43 }