github.com/whtcorpsinc/MilevaDB-Prod@v0.0.0-20211104133533-f57f4be3b597/causetstore/petri/petrictx_test.go (about) 1 // Copyright 2020 WHTCORPS INC, Inc. 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // See the License for the specific language governing permissions and 12 // limitations under the License. 13 14 package petri 15 16 import ( 17 . "github.com/whtcorpsinc/check" 18 "github.com/whtcorpsinc/milevadb/soliton/mock" 19 "github.com/whtcorpsinc/milevadb/soliton/testleak" 20 ) 21 22 var _ = Suite(&testPetriCtxSuite{}) 23 24 type testPetriCtxSuite struct { 25 } 26 27 func (s *testPetriCtxSuite) TestPetri(c *C) { 28 defer testleak.AfterTest(c)() 29 ctx := mock.NewContext() 30 31 c.Assert(petriKey.String(), Not(Equals), "") 32 33 BindPetri(ctx, nil) 34 v := GetPetri(ctx) 35 c.Assert(v, IsNil) 36 37 ctx.ClearValue(petriKey) 38 v = GetPetri(ctx) 39 c.Assert(v, IsNil) 40 }