@@ -221,13 +221,11 @@ func TestProvisionerd(t *testing.T) {
221221closer := createProvisionerd (t ,func (ctx context.Context ) (proto.DRPCProvisionerDaemonClient ,error ) {
222222return createProvisionerDaemonClient (t ,provisionerDaemonTestServer {
223223acquireJob :func (ctx context.Context ,_ * proto.Empty ) (* proto.AcquiredJob ,error ) {
224- if didAcquireJob .Load () {
225- completeOnce .Do (func () {
226- close (completeChan )
227- })
224+ if ! didAcquireJob .CAS (false ,true ) {
225+ completeOnce .Do (func () {close (completeChan ) })
228226return & proto.AcquiredJob {},nil
229227}
230- didAcquireJob . Store ( true )
228+
231229return & proto.AcquiredJob {
232230JobId :"test" ,
233231Provisioner :"someprovisioner" ,
@@ -325,11 +323,11 @@ func TestProvisionerd(t *testing.T) {
325323closer := createProvisionerd (t ,func (ctx context.Context ) (proto.DRPCProvisionerDaemonClient ,error ) {
326324return createProvisionerDaemonClient (t ,provisionerDaemonTestServer {
327325acquireJob :func (ctx context.Context ,_ * proto.Empty ) (* proto.AcquiredJob ,error ) {
328- if didAcquireJob .Load ( ) {
326+ if ! didAcquireJob .CAS ( false , true ) {
329327completeOnce .Do (func () {close (completeChan ) })
330328return & proto.AcquiredJob {},nil
331329}
332- didAcquireJob . Store ( true )
330+
333331return & proto.AcquiredJob {
334332JobId :"test" ,
335333Provisioner :"someprovisioner" ,
@@ -395,11 +393,11 @@ func TestProvisionerd(t *testing.T) {
395393closer := createProvisionerd (t ,func (ctx context.Context ) (proto.DRPCProvisionerDaemonClient ,error ) {
396394return createProvisionerDaemonClient (t ,provisionerDaemonTestServer {
397395acquireJob :func (ctx context.Context ,_ * proto.Empty ) (* proto.AcquiredJob ,error ) {
398- if didAcquireJob .Load ( ) {
396+ if ! didAcquireJob .CAS ( false , true ) {
399397completeOnce .Do (func () {close (completeChan ) })
400398return & proto.AcquiredJob {},nil
401399}
402- didAcquireJob . Store ( true )
400+
403401return & proto.AcquiredJob {
404402JobId :"test" ,
405403Provisioner :"someprovisioner" ,